adding simplefilter for warning

Atthemoment · Dec 10, 2014 · 5a019b7 · 5a019b7
1 parent b99824f
commit 5a019b7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sklearn/linear_model/tests/test_sgd.py b/sklearn/linear_model/tests/test_sgd.py
@@ -609,6 +609,7 @@ def test_class_weight_warning(self):
                            "the constructor instead.")
         import warnings
         with warnings.catch_warnings(record=True) as w:
+            warnings.simplefilter("always", DeprecationWarning)
             clf.fit(X4, Y4, class_weight=1)
             assert_true(warning_message == str(w[0].message))
             assert_true(issubclass(w[0].category, DeprecationWarning))