def test_get_worker_name(client): def f(): get_client().submit(inc, 1).result() client.run(f) def func(dask_scheduler): return list(dask_scheduler.clients) start = time() while not any("worker" in n for n in client.run_on_scheduler(func)): sleep(0.1) assert time() < start + 10
def test_get_worker_name(client): def f(): get_client().submit(inc, 1).result() client.run(f) def func(dask_scheduler): return list(dask_scheduler.clients) start = time() while not any('worker' in n for n in client.run_on_scheduler(func)): sleep(0.1) assert time() < start + 10
def test_scheduler_equals_client(c): x = delayed(lambda: 1)() assert x.compute(scheduler=c) == 1 assert c.run_on_scheduler(lambda dask_scheduler: dask_scheduler.story(x.key))