TST remove test_check_feature_names_in_pandas (scikit-learn#33954)

Author: lorentzenchr

sklearn/utils/tests/test_validation.py

@@ -2113,36 +2113,34 @@ def get_feature_names_out(self, input_features=None):
         return _check_feature_names_in(self, input_features)
 
 
-def test_check_feature_names_in():
+@pytest.mark.parametrize(
+    ["constructor_name", "feature_names", "msg"],
+    [
+        ("array", ["x0", "x1", "x2"], "input_features should have length equal to"),
+        ("pandas", ["a", "b", "c"], "input_features is not equal to"),
+        ("polars", ["a", "b", "c"], "input_features is not equal to"),
+    ],
+)
+def test_check_feature_names_in(constructor_name, feature_names, msg):
     """Check behavior of check_feature_names_in for arrays."""
     X = np.array([[0.0, 1.0, 2.0]])
+    X = _convert_container(X, constructor_name, column_names=["a", "b", "c"])
     est = PassthroughTransformer().fit(X)
 
     names = est.get_feature_names_out()
-    assert_array_equal(names, ["x0", "x1", "x2"])
+    assert_array_equal(names, feature_names)
 
-    incorrect_len_names = ["x10", "x1"]
-    with pytest.raises(ValueError, match="input_features should have length equal to"):
+    incorrect_len_names = feature_names[:2]
+    with pytest.raises(ValueError, match=msg):
         est.get_feature_names_out(incorrect_len_names)
 
     # remove n_feature_in_
     del est.n_features_in_
-    with pytest.raises(ValueError, match="Unable to generate feature names"):
-        est.get_feature_names_out()
-
-
-def test_check_feature_names_in_pandas():
-    """Check behavior of check_feature_names_in for pandas dataframes."""
-    pd = pytest.importorskip("pandas")
-    names = ["a", "b", "c"]
-    df = pd.DataFrame([[0.0, 1.0, 2.0]], columns=names)
-    est = PassthroughTransformer().fit(df)
-
-    names = est.get_feature_names_out()
-    assert_array_equal(names, ["a", "b", "c"])
-
-    with pytest.raises(ValueError, match="input_features is not equal to"):
-        est.get_feature_names_out(["x1", "x2", "x3"])
+    if constructor_name == "array":
+        with pytest.raises(ValueError, match="Unable to generate feature names"):
+            est.get_feature_names_out()
+    else:
+        assert_array_equal(est.get_feature_names_out(), feature_names)
 
 
 def test_check_response_method_unknown_method():