summaryrefslogtreecommitdiff
path: root/tests/util/test_cpu.py
blob: 3850ac35e4c7198356e403056b1bf722f0ed0105 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from __future__ import annotations

import multiprocessing

from pytest_mock import MockerFixture

from tox.util.cpu import auto_detect_cpus


def test_auto_detect_cpus() -> None:
    num_cpus_actual = multiprocessing.cpu_count()
    assert auto_detect_cpus() == num_cpus_actual


def test_auto_detect_cpus_returns_one_when_cpu_count_throws(mocker: MockerFixture) -> None:
    mocker.patch.object(multiprocessing, "cpu_count", side_effect=NotImplementedError)
    assert auto_detect_cpus() == 1