TST check that ensure_min_features is enforced with allow_nd=True

ogrisel · ogrisel · commit 7d7b95a307f9 · 2015-02-25T16:06:56.000+01:00
diff --git a/sklearn/utils/tests/test_validation.py b/sklearn/utils/tests/test_validation.py
@@ -250,6 +250,11 @@ def test_check_array_min_samples_and_features_messages():
     assert_raise_message(ValueError, msg, check_X_y, X, y,
                          ensure_min_features=3)
 
+    # Only the feature check is enabled whenever the number of dimensions is 2
+    # even if allow_nd is enabled:
+    assert_raise_message(ValueError, msg, check_X_y, X, y,
+                         ensure_min_features=3, allow_nd=True)
+
     # Simulate a case where a pipeline stage as trimmed all the features of a
     # 2D dataset.
     X = np.empty(0).reshape(10, 0)
