plugin: get cpu_count from sched_getaffinity

Signed-off-by: Yoan Blanc <yoan.blanc@exoscale.ch>
2018-07-18 08:02:53 +02:00
parent 06593498cd
commit 04882074e1
3 changed files with 12 additions and 3 deletions
--- a/xdist/plugin.py
+++ b/xdist/plugin.py
@@ -5,9 +5,15 @@ import pytest
 def parse_numprocesses(s):
    if s == 'auto':
        try:
-            from os import cpu_count
+            from os import sched_getaffinity
+            def cpu_count():
+                return len(sched_getaffinity(0))
        except ImportError:
-            from multiprocessing import cpu_count
+            try:
+                from os import cpu_count
+            except ImportError:
+                from multiprocessing import cpu_count
+
        try:
            n = cpu_count()
        except NotImplementedError: