Refine case

mars-project · Aug 23, 2021 · 20ad748 · 20ad748
1 parent ebc7033
commit 20ad748
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/mars/services/task/tests/test_service.py b/mars/services/task/tests/test_service.py
@@ -366,6 +366,11 @@ def _get_fields(details, field, wrapper=None):
     details = await task_api.get_tileable_details(task_id)
     assert details[r5.key]['progress'] == details[r6.key]['progress'] == 0.25
 
+    await ref.set()
+    await asyncio.sleep(0.1)
+    await ref.set()
+    await task_api.wait_task(task_id)
+
     # test raises
     r7 = mr.spawn(f, kwargs={'raises': 1})