[MRG+1] Fix test of SingleInheritanceEstimator to not raise DeprecationWarning (scikit-learn#8526)

zxcvbnius · Joan Massich · commit 577ff967b7a6 · 2017-04-26T10:57:19.000+02:00
* Fix test of SingleInheritanceEstimator to not raise DeprecationWarning * [MRG+1] Fix test of SingleInheritanceEstimator to not raise DeprecationWarning (scikit-learn#8526) Fix: test of SingleInheritanceEstimator to not raise DeprecationWarning (scikit-learn#8526) - Ignore a DeprecationWarning about unpickling an estimator from a different version * [MRG+2] Fix test of SingleInheritanceEstimator to not raise DeprecationWarning (scikit-learn#8526) Fix: test of SingleInheritanceEstimator to not raise DeprecationWarning (scikit-learn#8526) - Test of SingleInheritanceEstimator would raise UserWarning, not DeprecationWarning - Ignore a UserWarning about unpickling an estimator from a different version
diff --git a/sklearn/tests/test_base.py b/sklearn/tests/test_base.py
@@ -14,6 +14,7 @@
 from sklearn.utils.testing import assert_no_warnings
 from sklearn.utils.testing import assert_warns_message
 from sklearn.utils.testing import assert_dict_equal
+from sklearn.utils.testing import ignore_warnings
 
 from sklearn.base import BaseEstimator, clone, is_classifier
 from sklearn.svm import SVC
@@ -440,6 +441,7 @@ def __getstate__(self):
         return data
 
 
+@ignore_warnings(category=(UserWarning))
 def test_pickling_works_when_getstate_is_overwritten_in_the_child_class():
     estimator = SingleInheritanceEstimator()
     estimator._attribute_not_pickled = "this attribute should not be pickled"
