Uploaded Test files

venv/Lib/site-packages/sklearn/tests/test_config.py

@@ -0,0 +1,74 @@
+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)