diff --git a/conftest.py b/conftest.py index 18b59a9ed22..c98d558023b 100644 --- a/conftest.py +++ b/conftest.py @@ -35,5 +35,8 @@ def pytest_collection_modifyitems(config, items): if "slow" in item.keywords: item.add_marker(skip_slow) + if "ws" in item.fixturenames: + item.add_marker(pytest.mark.workerstate) + pytest_plugins = ["distributed.pytest_resourceleaks"] diff --git a/setup.cfg b/setup.cfg index bc76731dde2..5e36d379991 100644 --- a/setup.cfg +++ b/setup.cfg @@ -96,6 +96,7 @@ markers = ipython: marks tests as exercising IPython gpu: marks tests we want to run on GPUs leaking: ignore leaked resources; see pytest_resourceleaks.py for usage + workerstate: deterministic test for the worker state machine. Automatically applied to all tests that use the 'ws' fixture. # pytest-timeout settings # 'thread' kills off the whole test suite. 'signal' only kills the offending test.