From 2e6a2d7a2c8c06d0440b4edc3764dbd5c4d4501f Mon Sep 17 00:00:00 2001 From: joaopfonseca Date: Sun, 19 Dec 2021 23:17:50 +0000 Subject: [PATCH] FIX incorrect test (#881) --- imblearn/over_sampling/_smote/tests/test_geometric_smote.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/imblearn/over_sampling/_smote/tests/test_geometric_smote.py b/imblearn/over_sampling/_smote/tests/test_geometric_smote.py index 24e3d809c..3c14bdfec 100644 --- a/imblearn/over_sampling/_smote/tests/test_geometric_smote.py +++ b/imblearn/over_sampling/_smote/tests/test_geometric_smote.py @@ -157,7 +157,7 @@ def test_gsmote_fit_resample_binary( n_maj, n_min, step, min_coor, max_coor = 12, 5, 0.5, 0.0, 8.5 X = np.repeat(np.arange(min_coor, max_coor, step), 2).reshape(-1, 2) y = np.concatenate([np.repeat(0, n_maj), np.repeat(1, n_min)]) - radius = np.sqrt(0.5) * step + max_radius = np.sqrt(np.sum((X[-1] - X[n_maj - 1]) ** 2)) k_neighbors = 1 gsmote = GeometricSMOTE( "auto", @@ -170,7 +170,7 @@ def test_gsmote_fit_resample_binary( X_resampled, y_resampled = gsmote.fit_resample(X, y) assert gsmote.sampling_strategy_ == {1: (n_maj - n_min)} assert y_resampled.sum() == n_maj - np.testing.assert_array_less(X[n_maj - 1] - radius, X_resampled[n_maj + n_min]) + np.testing.assert_array_less(X[-1] - max_radius, X_resampled[n_maj + n_min]) @pytest.mark.parametrize(