Fix up use of threading events

numba · Oct 7, 2020 · a2cc68a · a2cc68a
1 parent 3c997f0
commit a2cc68a
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/numba/tests/test_parallel_backend.py b/numba/tests/test_parallel_backend.py
@@ -873,18 +873,26 @@ def test_fork_from_non_main_thread(self):
             from numba import njit, prange, objmode
             import os
 
+            e_running = threading.Event()
+            e_proceed = threading.Event()
+
+            def indirect():
+                e_running.set()
+                # wait for forker() to have forked
+                while not e_proceed.isSet():
+                    pass
+
             def runner():
                 @njit(parallel=True, nogil=True)
                 def work():
                     acc = 0
                     for x in prange(10):
                         acc += x
                     with objmode():
-                        e_running.set()
-                        # wait for forker() to have forked
-                        while not e_proceed.isSet():
-                            pass
+                        indirect()
+
                     return acc
+
                 work()
 
             def forker():
@@ -896,8 +904,6 @@ def forker():
                 # now fork is done signal the runner to proceed to exit
                 e_proceed.set()
 
-            e_running = threading.Event()
-            e_proceed = threading.Event()
             numba_runner = threading.Thread(target=runner,)
             fork_runner =  threading.Thread(target=forker,)