From 5a019b7202fb887506ed6ae2ef1c1716ca9a6f7a Mon Sep 17 00:00:00 2001 From: dsullivan7 Date: Wed, 10 Dec 2014 16:07:46 +0100 Subject: [PATCH] adding simplefilter for warning --- sklearn/linear_model/tests/test_sgd.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sklearn/linear_model/tests/test_sgd.py b/sklearn/linear_model/tests/test_sgd.py index 56357bab6b675..1845b1418fdb1 100644 --- 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))