Prevent test timeout on a single CPU (#108)

dask · Oct 2, 2023 · b142286 · b142286
1 parent 910c63e
commit b142286
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/zict/tests/utils_test.py b/zict/tests/utils_test.py
@@ -169,7 +169,9 @@ def check_different_keys_threadsafe(
 
     def worker(idx, key, value):
         barrier.wait()
-        while any(c < 10 for c in counters):
+        # When running on a single CPU (`taskset -c 0 pytest`), multitasking can
+        # misbehave and almost completely starve one of the two threads
+        while any(c < 10 for c in counters) and all(c < 1000 for c in counters):
             z[key] = value
             try:
                 assert z[key] == value