| Current Path : /proc/thread-self/root/usr/local/lib/python3.8/dist-packages/sklearn/utils/tests/ |
Linux ift1.ift-informatik.de 5.4.0-216-generic #236-Ubuntu SMP Fri Apr 11 19:53:21 UTC 2025 x86_64 |
| Current File : //proc/thread-self/root/usr/local/lib/python3.8/dist-packages/sklearn/utils/tests/test_parallel.py |
from distutils.version import LooseVersion
import pytest
from joblib import Parallel
import joblib
from numpy.testing import assert_array_equal
from sklearn._config import config_context, get_config
from sklearn.utils.fixes import delayed
def get_working_memory():
return get_config()["working_memory"]
@pytest.mark.parametrize("n_jobs", [1, 2])
@pytest.mark.parametrize("backend", ["loky", "threading",
"multiprocessing"])
def test_configuration_passes_through_to_joblib(n_jobs, backend):
# Tests that the global global configuration is passed to joblib jobs
if joblib.__version__ < LooseVersion('0.12') and backend == 'loky':
pytest.skip('loky backend does not exist in joblib <0.12')
with config_context(working_memory=123):
results = Parallel(n_jobs=n_jobs, backend=backend)(
delayed(get_working_memory)() for _ in range(2))
assert_array_equal(results, [123] * 2)