Check in-flight task count in steal test

dask · Mar 4, 2021 · 47ada8e · 47ada8e
1 parent 94d184a
commit 47ada8e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/distributed/tests/test_steal.py b/distributed/tests/test_steal.py
@@ -660,6 +660,9 @@ async def test_steal_twice(c, s, a, b):
         )
     assert max(map(len, has_what.values())) < 30
 
+    assert a.in_flight_tasks == 0
+    assert b.in_flight_tasks == 0
+
     await c._close()
     await asyncio.gather(*[w.close() for w in workers])