Move test_get_adapter_from_container to test_set_output.py

patrickkwang · patrickkwang · commit b66a38d57a02 · 2024-04-09T15:29:21.000-04:00
diff --git a/sklearn/preprocessing/tests/test_function_transformer.py b/sklearn/preprocessing/tests/test_function_transformer.py
@@ -5,7 +5,6 @@
 
 from sklearn.pipeline import make_pipeline
 from sklearn.preprocessing import FunctionTransformer, StandardScaler
-from sklearn.preprocessing._function_transformer import _get_adapter_from_container
 from sklearn.utils._testing import (
     _convert_container,
     assert_allclose_dense_sparse,
@@ -14,17 +13,6 @@
 from sklearn.utils.fixes import CSC_CONTAINERS, CSR_CONTAINERS
 
 
-def test_get_adapter_from_container():
-    """Check the behavior fo `_get_adapter_from_container`."""
-    pd = pytest.importorskip("pandas")
-    X = pd.DataFrame({"a": [1, 2, 3], "b": [10, 20, 100]})
-    adapter = _get_adapter_from_container(X)
-    assert adapter.container_lib == "pandas"
-    err_msg = "The container does not have a registered adapter in scikit-learn."
-    with pytest.raises(ValueError, match=err_msg):
-        _get_adapter_from_container(X.to_numpy())
-
-
 def _make_func(args_store, kwargs_store, func=lambda X, *a, **k: X):
     def _func(X, *args, **kwargs):
         args_store.append(X)
diff --git a/sklearn/utils/tests/test_set_output.py b/sklearn/utils/tests/test_set_output.py
@@ -10,6 +10,7 @@
 from sklearn.utils._set_output import (
     ADAPTERS_MANAGER,
     ContainerAdapterProtocol,
+    _get_adapter_from_container,
     _get_output_config,
     _safe_set_output,
     _SetOutputMixin,
@@ -450,3 +451,14 @@ def patched_import_module(name):
     msg = "Setting output container to 'pandas' requires"
     with pytest.raises(ImportError, match=msg):
         check_library_installed("pandas")
+
+
+def test_get_adapter_from_container():
+    """Check the behavior fo `_get_adapter_from_container`."""
+    pd = pytest.importorskip("pandas")
+    X = pd.DataFrame({"a": [1, 2, 3], "b": [10, 20, 100]})
+    adapter = _get_adapter_from_container(X)
+    assert adapter.container_lib == "pandas"
+    err_msg = "The container does not have a registered adapter in scikit-learn."
+    with pytest.raises(ValueError, match=err_msg):
+        _get_adapter_from_container(X.to_numpy())
