Replace use or assert_no_warnings

zitorelova · zitorelova · commit 9450c332304f · 2021-02-11T09:26:20.000-08:00
diff --git a/sklearn/feature_extraction/tests/test_text.py b/sklearn/feature_extraction/tests/test_text.py
@@ -30,7 +30,6 @@
 from sklearn.utils import IS_PYPY
 from sklearn.utils._testing import (assert_almost_equal,
                                     assert_raise_message,
-                                    assert_no_warnings,
                                     fails_if_pypy,
                                     assert_allclose_dense_sparse,
                                     skip_if_32bit)
@@ -1215,7 +1214,9 @@ def test_vectorizer_stop_words_inconsistent():
         assert _check_stop_words_consistency(vec) is False
 
     # Only one warning per stop list
-    assert_no_warnings(vec.fit_transform, ['hello world'])
+    with pytest.warns(None) as record:
+        vec.fit_transform(['hello world'])
+    assert not len(record)
     assert _check_stop_words_consistency(vec) is None
 
     # Test caching of inconsistency assessment
