From e9f100f5a13c58e2a13caa3a61901833e973d634 Mon Sep 17 00:00:00 2001 From: Thomas Fan Date: Sat, 29 Jun 2019 14:23:50 -0400 Subject: [PATCH] TST refactor test_numeric_stability --- sklearn/preprocessing/tests/test_discretization.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/sklearn/preprocessing/tests/test_discretization.py b/sklearn/preprocessing/tests/test_discretization.py index 102b789eb093d..6dd0abdb99e9f 100644 --- a/sklearn/preprocessing/tests/test_discretization.py +++ b/sklearn/preprocessing/tests/test_discretization.py @@ -135,15 +135,15 @@ def test_transform_1d_behavior(): assert_raises(ValueError, est.transform, X) -def test_numeric_stability(): +@pytest.mark.parametrize('i', range(1, 9)) +def test_numeric_stability(i): X_init = np.array([2., 4., 6., 8., 10.]).reshape(-1, 1) Xt_expected = np.array([0, 0, 1, 1, 1]).reshape(-1, 1) # Test up to discretizing nano units - for i in range(1, 9): - X = X_init / 10**i - Xt = KBinsDiscretizer(n_bins=2, encode='ordinal').fit_transform(X) - assert_array_equal(Xt_expected, Xt) + X = X_init / 10**i + Xt = KBinsDiscretizer(n_bins=2, encode='ordinal').fit_transform(X) + assert_array_equal(Xt_expected, Xt) def test_invalid_encode_option():