fix e2e test

Parsl · Oct 12, 2024 · e44262f · e44262f
1 parent 6ebcd5b
commit e44262f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/parsl/tests/test_htex/test_manager_selector_by_block.py b/parsl/tests/test_htex/test_manager_selector_by_block.py
@@ -41,14 +41,13 @@ def get_worker_pid():
     import os
     return os.environ.get('PARSL_WORKER_BLOCK_ID')
 
-
 @pytest.mark.local
-def test_block_id_selection():
+def test_block_id_selection(try_assert):
     blockids = []
+    try_assert(lambda: (time.sleep(1) or get_worker_pid().result() == "1"), timeout_ms=20000)
     for i in range(10):
         future = get_worker_pid()
         blockids.append(future.result())
         print(future.result())
-        time.sleep(1)
 
     assert all(blockid == "1" for blockid in blockids)