Use psutil.cpu_count(logical=False) in auto_detect_cpus()
This commit is contained in:
@@ -35,17 +35,10 @@ def test_dist_options(testdir):
|
||||
|
||||
|
||||
def test_auto_detect_cpus(testdir, monkeypatch):
|
||||
import os
|
||||
import psutil
|
||||
from xdist.plugin import pytest_cmdline_main as check_options
|
||||
|
||||
if hasattr(os, "sched_getaffinity"):
|
||||
monkeypatch.setattr(os, "sched_getaffinity", lambda _pid: set(range(99)))
|
||||
elif hasattr(os, "cpu_count"):
|
||||
monkeypatch.setattr(os, "cpu_count", lambda: 99)
|
||||
else:
|
||||
import multiprocessing
|
||||
|
||||
monkeypatch.setattr(multiprocessing, "cpu_count", lambda: 99)
|
||||
monkeypatch.setattr(psutil, "cpu_count", lambda logical=True: 99)
|
||||
|
||||
config = testdir.parseconfigure("-n2")
|
||||
assert config.getoption("numprocesses") == 2
|
||||
@@ -58,10 +51,12 @@ def test_auto_detect_cpus(testdir, monkeypatch):
|
||||
assert config.getoption("usepdb")
|
||||
assert config.getoption("numprocesses") == 0
|
||||
|
||||
monkeypatch.delattr(os, "sched_getaffinity", raising=False)
|
||||
monkeypatch.setenv("TRAVIS", "true")
|
||||
def cpu_count_not_implemented(logical=True):
|
||||
raise NotImplementedError
|
||||
|
||||
monkeypatch.setattr(psutil, "cpu_count", cpu_count_not_implemented)
|
||||
config = testdir.parseconfigure("-nauto")
|
||||
assert config.getoption("numprocesses") == 2
|
||||
assert config.getoption("numprocesses") == 1
|
||||
|
||||
|
||||
def test_boxed_with_collect_only(testdir):
|
||||
|
||||
Reference in New Issue
Block a user