Fix test_auto_detect_cpus

2016-08-06 17:06:37 -03:00
parent d8d9f3684b
commit a2881fd8a9
1 changed files with 6 additions and 2 deletions
--- a/testing/test_plugin.py
+++ b/testing/test_plugin.py
@@ -26,8 +26,12 @@ def test_dist_options(testdir):


 def test_auto_detect_cpus(testdir, monkeypatch):
-    import multiprocessing
-    monkeypatch.setattr(multiprocessing, 'cpu_count', lambda: 99)
+    import os
+    if hasattr(os, 'cpu_count'):
+        monkeypatch.setattr(os, 'cpu_count', lambda: 99)
+    else:
+        import multiprocessing
+        monkeypatch.setattr(multiprocessing, 'cpu_count', lambda: 99)

    config = testdir.parseconfigure("-n2")
    assert config.getoption('numprocesses') == 2