From 84cbb099e0d0f1e1947d51d957f0b65e9b0635dd Mon Sep 17 00:00:00 2001 From: crusaderky Date: Tue, 26 Apr 2022 14:54:03 +0100 Subject: [PATCH] Harmonize #6161 and #6173 (#6207) --- distributed/tests/test_worker.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/distributed/tests/test_worker.py b/distributed/tests/test_worker.py index 06d71a03e29..6bea4a3fdf5 100644 --- a/distributed/tests/test_worker.py +++ b/distributed/tests/test_worker.py @@ -1819,7 +1819,7 @@ async def test_gather_dep_local_workers_first(c, s, a, lw, *rws): )["f"] g = c.submit(inc, f, key="g", workers=[a.address]) assert await g == 2 - assert_worker_story(a.story("f"), [("receive-dep", lw.address, {"f"})]) + assert_story(a.story("f"), [("receive-dep", lw.address, {"f"})]) @pytest.mark.skipif(not LINUX, reason="Need 127.0.0.2 to mean localhost") @@ -1846,7 +1846,7 @@ async def test_gather_dep_from_remote_workers_if_all_local_workers_are_busy( assert sorted(ev[1] for ev in a.story("busy-gather")) == sorted( w.address for w in lws ) - assert_worker_story(a.story("receive-dep"), [("receive-dep", rw.address, {"f"})]) + assert_story(a.story("receive-dep"), [("receive-dep", rw.address, {"f"})]) @gen_cluster(client=True, nthreads=[("127.0.0.1", 0)])