Traktor/myenv/Lib/site-packages/joblib/test/test_module.py

54 lines
1.9 KiB
Python
Raw Permalink Normal View History

2024-05-26 05:12:46 +02:00
import sys
import joblib
from joblib.testing import check_subprocess_call
from joblib.test.common import with_multiprocessing
def test_version():
assert hasattr(joblib, '__version__'), (
"There are no __version__ argument on the joblib module")
@with_multiprocessing
def test_no_start_method_side_effect_on_import():
# check that importing joblib does not implicitly set the global
# start_method for multiprocessing.
code = """if True:
import joblib
import multiprocessing as mp
# The following line would raise RuntimeError if the
# start_method is already set.
mp.set_start_method("loky")
"""
check_subprocess_call([sys.executable, '-c', code])
@with_multiprocessing
def test_no_semaphore_tracker_on_import():
# check that importing joblib does not implicitly spawn a resource tracker
# or a semaphore tracker
code = """if True:
import joblib
from multiprocessing import semaphore_tracker
# The following line would raise RuntimeError if the
# start_method is already set.
msg = "multiprocessing.semaphore_tracker has been spawned on import"
assert semaphore_tracker._semaphore_tracker._fd is None, msg"""
if sys.version_info >= (3, 8):
# semaphore_tracker was renamed in Python 3.8:
code = code.replace("semaphore_tracker", "resource_tracker")
check_subprocess_call([sys.executable, '-c', code])
@with_multiprocessing
def test_no_resource_tracker_on_import():
code = """if True:
import joblib
from joblib.externals.loky.backend import resource_tracker
# The following line would raise RuntimeError if the
# start_method is already set.
msg = "loky.resource_tracker has been spawned on import"
assert resource_tracker._resource_tracker._fd is None, msg
"""
check_subprocess_call([sys.executable, '-c', code])