MAINT Make test_iterative_imputer_catch_warning failures more informative (#21709)

ogrisel · glemaitre · commit 87f94b965c91 · 2021-12-25T12:26:49.000+01:00
diff --git a/sklearn/impute/tests/test_impute.py b/sklearn/impute/tests/test_impute.py
@@ -936,7 +936,7 @@ def test_iterative_imputer_catch_warning():
     imputer = IterativeImputer(n_nearest_features=5, sample_posterior=True)
     with pytest.warns(None) as record:
         X_fill = imputer.fit_transform(X, y)
-    assert not record.list
+    assert not [w.message for w in record.list]
     assert not np.any(np.isnan(X_fill))
 
 
