Move assert_warns to imblearn.utils.test.warns

Author: Joan Massich

imblearn/under_sampling/prototype_selection/tests/test_nearmiss.py

@@ -6,7 +6,8 @@
 from __future__ import print_function
 
 import numpy as np
-from sklearn.utils.testing import assert_array_equal, assert_warns
+from sklearn.utils.testing import assert_array_equal
+from imblearn.utils.testing import warns
 from sklearn.neighbors import NearestNeighbors
 from pytest import raises
 
@@ -36,7 +37,8 @@
 # FIXME remove at the end of the deprecation 0.4
 def test_nearmiss_deprecation():
     nm = NearMiss(ver3_samp_ngh=3, version=3)
-    assert_warns(DeprecationWarning, nm.fit_sample, X, Y)
+    with warns(DeprecationWarning, match="deprecated from 0.2"):
+        nm.fit_sample(X, Y)
 
 
 def test_nearmiss_wrong_version():

imblearn/utils/estimator_checks.py

@@ -18,8 +18,8 @@
     as sklearn_yield_all_checks, check_estimator \
     as sklearn_check_estimator, check_parameters_default_constructible
 from sklearn.exceptions import NotFittedError
-from sklearn.utils.testing import assert_warns
 from pytest import raises
+from imblearn.utils.testing import warns
 
 from sklearn.utils.testing import set_random_state
 
@@ -77,15 +77,8 @@ def check_target_type(name, Estimator):
     y = np.linspace(0, 1, 20)
     estimator = Estimator()
     set_random_state(estimator)
-    assert_warns(UserWarning, estimator.fit, X, y)
-
-
-def check_multiclass_warning(name, Estimator):
-    X = np.random.random((20, 2))
-    y = np.array([0] * 3 + [1] * 2 + [2] * 15)
-    estimator = Estimator()
-    set_random_state(estimator)
-    assert_warns(UserWarning, estimator.fit, X, y)
+    with warns(UserWarning, match='should be of types'):
+        estimator.fit(X, y)
 
 
 def multioutput_estimator_convert_y_2d(name, y):