Update tests/unit/run_modes/test_skip_units.py

Co-authored-by: Ronnie Dutta <[email protected]>
cylc · Nov 14, 2024 · c5e84a8 · c5e84a8
1 parent 6085a90
commit c5e84a8
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/tests/unit/run_modes/test_skip_units.py b/tests/unit/run_modes/test_skip_units.py
@@ -129,8 +129,12 @@ def test_skip_mode_validate(caplog, log_filter):
 
     skip_mode_validate(taskdefs)
 
-    message = caplog.messages[0]
-
-    assert 'skip mode:\n    * skip_task' in message
-    assert ' live mode' not in message   # Avoid matching "non-live mode"
-    assert 'workflow mode' not in message
+    assert len(caplog.records) == 1
+    assert log_filter(
+        level=logging.WARNING,
+        exact_match=(
+            "The following tasks are set to run in skip mode:\n"
+            "    * skip_task"
+        ),
+        log=caplog
+    )