improve dask distributed test

mllam · Oct 30, 2024 · c951797 · c951797
1 parent 0eeca64
commit c951797
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/test_distributed.py b/tests/test_distributed.py
@@ -25,10 +25,15 @@ def distributed():
 
 
 @pytest.mark.parametrize(
-    "args", [["example.danra.yaml", "--dask-distributed-local-core-fraction", "0.1"]]
+    "args", [["example.danra.yaml", "--dask-distributed-local-core-fraction", "0.1"], ["example.danra.yaml"]]
 )
 def test_run_distributed(args):
-    if distributed():
+    if distributed() and "--dask-distributed-local-core-fraction" in args:
+        pytest.raises(
+            ZeroDivisionError,
+            call_wrapper(args),
+            match="division by zero",
+    elif distributed():
         call_wrapper(args)
     else:
         pytest.raises(