async def test_worker_stealing_interval(c, s, a, b):
from distributed.scheduler import WorkStealing
ws = WorkStealing(s)
assert ws._pc.callback_time == 100
with dask.config.set({"distributed.scheduler.work-stealing-interval": "500ms"}):
ws = WorkStealing(s)
assert ws._pc.callback_time == 500
# Default unit is `ms`
with dask.config.set({"distributed.scheduler.work-stealing-interval": 2}):
ws = WorkStealing(s)
assert ws._pc.callback_time == 2
async def test_parse_stealing_interval(s, interval, expected):
from distributed.scheduler import WorkStealing
if interval:
ctx = dask.config.set(
{"distributed.scheduler.work-stealing-interval": "500ms"})
else:
ctx = contextlib.nullcontext()
with ctx:
ws = WorkStealing(s)
s.periodic_callbacks["stealing"].callback_time == expected