| Current Path : /proc/thread-self/root/usr/local/lib/python3.8/dist-packages/sklearn/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/tests/test_config.py |
from sklearn import get_config, set_config, config_context
from sklearn.utils._testing import assert_raises
def test_config_context():
assert get_config() == {'assume_finite': False, 'working_memory': 1024,
'print_changed_only': True,
'display': 'text'}
# Not using as a context manager affects nothing
config_context(assume_finite=True)
assert get_config()['assume_finite'] is False
with config_context(assume_finite=True):
assert get_config() == {'assume_finite': True, 'working_memory': 1024,
'print_changed_only': True,
'display': 'text'}
assert get_config()['assume_finite'] is False
with config_context(assume_finite=True):
with config_context(assume_finite=None):
assert get_config()['assume_finite'] is True
assert get_config()['assume_finite'] is True
with config_context(assume_finite=False):
assert get_config()['assume_finite'] is False
with config_context(assume_finite=None):
assert get_config()['assume_finite'] is False
# global setting will not be retained outside of context that
# did not modify this setting
set_config(assume_finite=True)
assert get_config()['assume_finite'] is True
assert get_config()['assume_finite'] is False
assert get_config()['assume_finite'] is True
assert get_config() == {'assume_finite': False, 'working_memory': 1024,
'print_changed_only': True,
'display': 'text'}
# No positional arguments
assert_raises(TypeError, config_context, True)
# No unknown arguments
assert_raises(TypeError, config_context(do_something_else=True).__enter__)
def test_config_context_exception():
assert get_config()['assume_finite'] is False
try:
with config_context(assume_finite=True):
assert get_config()['assume_finite'] is True
raise ValueError()
except ValueError:
pass
assert get_config()['assume_finite'] is False
def test_set_config():
assert get_config()['assume_finite'] is False
set_config(assume_finite=None)
assert get_config()['assume_finite'] is False
set_config(assume_finite=True)
assert get_config()['assume_finite'] is True
set_config(assume_finite=None)
assert get_config()['assume_finite'] is True
set_config(assume_finite=False)
assert get_config()['assume_finite'] is False
# No unknown arguments
assert_raises(TypeError, set_config, do_something_else=True)