diff --git a/tests/test_server.py b/tests/test_server.py index 184a391a..1fa345eb 100644 --- a/tests/test_server.py +++ b/tests/test_server.py @@ -26,4 +26,5 @@ def test_basic(server): timeout=None, ) - server.run_blockwise([task]) + task_state = server.run_blockwise([task]) + assert task_state[task.task_id].is_done(), task_state[task.task_id]