simpler test

TomDLT · TomDLT · commit b5b6083ad947 · 2018-07-09T17:45:16.000+02:00
diff --git a/sklearn/preprocessing/tests/test_discretization.py b/sklearn/preprocessing/tests/test_discretization.py
@@ -144,14 +144,13 @@ def test_ignored_invalid():
 
 
 @pytest.mark.parametrize('strategy', ['uniform', 'kmeans', 'quantile'])
-@pytest.mark.parametrize('n_bins', [2, 3])
-def test_same_min_max(strategy, n_bins):
+def test_same_min_max(strategy):
     warnings.simplefilter("always")
     X = np.array([[1, -2],
                   [1, -1],
                   [1, 0],
                   [1, 1]])
-    est = KBinsDiscretizer(strategy=strategy, n_bins=n_bins, encode='ordinal')
+    est = KBinsDiscretizer(strategy=strategy, n_bins=3, encode='ordinal')
     assert_warns_message(UserWarning,
                          "Feature 0 is constant and will be replaced "
                          "with 0.", est.fit, X)
