diff --git a/tests/test_distributed.py b/tests/test_distributed.py index facc692..ca420b8 100644 --- a/tests/test_distributed.py +++ b/tests/test_distributed.py @@ -32,14 +32,14 @@ def distributed(): ], ) def test_run_distributed(args): - if distributed() and "--dask-distributed-local-core-fraction" in args: + if distributed(): + call_wrapper(args) + elif not 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( ModuleNotFoundError,