Check deprecation message in SparsePCA test

naoyak · naoyak · commit c60db808284d · 2017-01-03T12:05:27.000-08:00
diff --git a/sklearn/decomposition/tests/test_sparse_pca.py b/sklearn/decomposition/tests/test_sparse_pca.py
@@ -11,7 +11,7 @@
 from sklearn.utils.testing import SkipTest
 from sklearn.utils.testing import assert_true
 from sklearn.utils.testing import assert_false
-from sklearn.utils.testing import assert_warns
+from sklearn.utils.testing import assert_warns_message
 from sklearn.utils.testing import if_safe_multiprocessing_with_blas
 
 from sklearn.decomposition import SparsePCA, MiniBatchSparsePCA
@@ -72,8 +72,9 @@ def test_fit_transform():
     assert_array_almost_equal(spca_lasso.components_, spca_lars.components_)
 
     # Test that deprecated ridge_alpha parameter throws warning
-    assert_warns(DeprecationWarning, spca_lars.transform, Y, ridge_alpha=0.01)
-    assert_warns(DeprecationWarning, spca_lars.transform, Y, ridge_alpha=None)
+    warning_msg = "The ridge_alpha parameter on transform()"
+    assert_warns_message(DeprecationWarning, warning_msg, spca_lars.transform, Y, ridge_alpha=0.01)
+    assert_warns_message(DeprecationWarning, warning_msg, spca_lars.transform, Y, ridge_alpha=None)
 
 
 @if_safe_multiprocessing_with_blas
