Merge pull request #54 from ngoldbaum/fix-32bit

attempt to fix 32 bit architectures

Author: andfoy

src/pytest_run_parallel/plugin.py

@@ -65,52 +65,53 @@ def pytest_configure(config):
 
 
 def wrap_function_parallel(fn, n_workers, n_iterations):
-    barrier = threading.Barrier(n_workers)
-    original_switch = sys.getswitchinterval()
-    sys.setswitchinterval(0.000001)
-
     @functools.wraps(fn)
     def inner(*args, **kwargs):
         errors = []
         skip = None
         failed = None
-
-        def closure(*args, **kwargs):
-            for _ in range(n_iterations):
-                barrier.wait()
-                try:
-                    fn(*args, **kwargs)
-                except Warning:
-                    pass
-                except Exception as e:
-                    errors.append(e)
-                except _pytest.outcomes.Skipped as s:
-                    nonlocal skip
-                    skip = s.msg
-                except _pytest.outcomes.Failed as f:
-                    nonlocal failed
-                    failed = f
-
-        workers = []
-        for _ in range(0, n_workers):
-            worker_kwargs = kwargs
-            workers.append(
-                threading.Thread(target=closure, args=args, kwargs=worker_kwargs)
-            )
-
-        num_completed = 0
+        barrier = threading.Barrier(n_workers)
+        original_switch = sys.getswitchinterval()
         try:
-            for worker in workers:
-                worker.start()
-                num_completed += 1
-        finally:
-            if num_completed < len(workers):
-                barrier.abort()
+            sys.setswitchinterval(0.000001)
+
+            def closure(*args, **kwargs):
+                for _ in range(n_iterations):
+                    barrier.wait()
+                    try:
+                        fn(*args, **kwargs)
+                    except Warning:
+                        pass
+                    except Exception as e:
+                        errors.append(e)
+                    except _pytest.outcomes.Skipped as s:
+                        nonlocal skip
+                        skip = s.msg
+                    except _pytest.outcomes.Failed as f:
+                        nonlocal failed
+                        failed = f
+
+            workers = []
+            for _ in range(0, n_workers):
+                worker_kwargs = kwargs
+                workers.append(
+                    threading.Thread(target=closure, args=args, kwargs=worker_kwargs)
+                )
+
+            num_completed = 0
+            try:
+                for worker in workers:
+                    worker.start()
+                    num_completed += 1
+            finally:
+                if num_completed < len(workers):
+                    barrier.abort()
 
-        for worker in workers:
-            worker.join()
+            for worker in workers:
+                worker.join()
 
-        sys.setswitchinterval(original_switch)
+        finally:
+            sys.setswitchinterval(original_switch)
 
         if skip is not None:
             pytest.skip(skip)