TST Replaces pytest.warns(None) in test_feature_agglomeration (scikit…

…-learn#22871) Co-authored-by: Shan-Deng <[email protected]>
aaghamohammadi · Mar 17, 2022 · 0f38376 · 0f38376
1 parent e397f76
commit 0f38376
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/sklearn/cluster/tests/test_feature_agglomeration.py b/sklearn/cluster/tests/test_feature_agglomeration.py
@@ -3,7 +3,6 @@
 """
 # Authors: Sergul Aydore 2017
 import numpy as np
-import pytest
 
 from numpy.testing import assert_array_equal
 from sklearn.cluster import FeatureAgglomeration
@@ -17,12 +16,8 @@ def test_feature_agglomeration():
 
     agglo_mean = FeatureAgglomeration(n_clusters=n_clusters, pooling_func=np.mean)
     agglo_median = FeatureAgglomeration(n_clusters=n_clusters, pooling_func=np.median)
-    with pytest.warns(None) as record:
-        agglo_mean.fit(X)
-    assert not [w.message for w in record]
-    with pytest.warns(None) as record:
-        agglo_median.fit(X)
-    assert not [w.message for w in record]
+    agglo_mean.fit(X)
+    agglo_median.fit(X)
 
     assert np.size(np.unique(agglo_mean.labels_)) == n_clusters
     assert np.size(np.unique(agglo_median.labels_)) == n_clusters