multithread correctly on macs

Author: sblunt

tests/test_nested_sampler.py

@@ -6,6 +6,7 @@
 import numpy as np
 import pytest
 from orbitize.system import generate_synthetic_data
+import sys
 
 
 def test_nested_sampler():
@@ -24,29 +25,34 @@ def test_nested_sampler():
     ecc = 0.5
 
     # initialize orbitize `System` object
-    sys = system.System(1, data_table, mtot, plx)
-    lab = sys.param_idx
+    mySys = system.System(1, data_table, mtot, plx)
+    lab = mySys.param_idx
 
     ecc = 0.5  # eccentricity
 
     # set all parameters except eccentricity to fixed values (same as used to generate data)
-    sys.sys_priors[lab["inc1"]] = np.pi / 4
-    sys.sys_priors[lab["sma1"]] = sma
-    sys.sys_priors[lab["aop1"]] = np.pi / 4
-    sys.sys_priors[lab["pan1"]] = np.pi / 4
-    sys.sys_priors[lab["tau1"]] = 0.8
-    sys.sys_priors[lab["plx"]] = plx
-    sys.sys_priors[lab["mtot"]] = mtot
+    mySys.sys_priors[lab["inc1"]] = np.pi / 4
+    mySys.sys_priors[lab["sma1"]] = sma
+    mySys.sys_priors[lab["aop1"]] = np.pi / 4
+    mySys.sys_priors[lab["pan1"]] = np.pi / 4
+    mySys.sys_priors[lab["tau1"]] = 0.8
+    mySys.sys_priors[lab["plx"]] = plx
+    mySys.sys_priors[lab["mtot"]] = mtot
+
+    start_method="fork"
+    if sys.platform == "darwin":
+        start_method="spawn"
 
     # run both static & dynamic nested samplers
-    mysampler = sampler.NestedSampler(sys)
-    _ = mysampler.run_sampler(bound="multi", pfrac=0.95, static=False, num_threads=8)
+    mysampler = sampler.NestedSampler(mySys)
+    _ = mysampler.run_sampler(bound="multi", pfrac=0.95, static=False, start_method=start_method, num_threads=8)
     print("Finished first run!")
 
     dynamic_eccentricities = mysampler.results.post[:, lab["ecc1"]]
     assert np.median(dynamic_eccentricities) == pytest.approx(ecc, abs=0.1)
 
-    _ = mysampler.run_sampler(bound="multi", static=True, num_threads=8)
+
+    _ = mysampler.run_sampler(bound="multi", static=True, start_method=start_method, num_threads=8)
     print("Finished second run!")
 
     static_eccentricities = mysampler.results.post[:, lab["ecc1"]]