Skip to content

Commit f183d26

Browse files
committed
Test that implicit multithreading env var is set correctly.
1 parent 5ca6529 commit f183d26

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

tests/test_simple.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,22 @@ def testEnvironment(self):
2121
self.assertIn(envVar, os.environ)
2222
self.assertTrue(os.path.exists(os.environ[envVar]), f"Check path {os.environ[envVar]}")
2323

24+
def testNoImplicitMultithreading(self):
25+
"""Test that the environment has turned off implicit
26+
multithreading.
27+
"""
28+
envVar = "OMP_NUM_THREADS"
29+
self.assertIn(envVar, os.environ)
30+
self.assertEqual(os.environ[envVar], "1")
31+
32+
try:
33+
import numexpr
34+
except ImportError:
35+
numexpr = None
36+
37+
if numexpr:
38+
self.assertEqual(numexpr.utils.get_num_threads(), 1)
39+
2440

2541
if __name__ == "__main__":
2642
unittest.main()

0 commit comments

Comments
 (0)