75 lines
2.7 KiB
Python
75 lines
2.7 KiB
Python
|
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)
|