From 425ebcb6a5ebe5cea00ce6fb96a742ab09e03567 Mon Sep 17 00:00:00 2001 From: NickleDave Date: Sat, 15 Jul 2017 17:34:13 -0500 Subject: [PATCH] use ignore_warning to catch FutueWarning --- sklearn/utils/tests/test_estimator_checks.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/sklearn/utils/tests/test_estimator_checks.py b/sklearn/utils/tests/test_estimator_checks.py index 8ac31764e89ad..1b3a1ea7e597a 100644 --- a/sklearn/utils/tests/test_estimator_checks.py +++ b/sklearn/utils/tests/test_estimator_checks.py @@ -6,7 +6,7 @@ from sklearn.base import BaseEstimator, ClassifierMixin from sklearn.utils.testing import (assert_raises_regex, assert_true, - assert_equal) + assert_equal, ignore_warnings) from sklearn.utils.estimator_checks import check_estimator from sklearn.utils.estimator_checks import set_random_state from sklearn.utils.estimator_checks import set_checking_parameters @@ -203,7 +203,9 @@ def test_check_estimator_clones(): for Estimator in [GaussianMixture, LinearRegression, RandomForestClassifier, NMF, SGDClassifier, MiniBatchKMeans]: - est = Estimator() + with ignore_warnings(category=FutureWarning): + # when 'est = SGDClassifier()' + est = Estimator() set_checking_parameters(est) set_random_state(est) # without fitting @@ -211,7 +213,9 @@ def test_check_estimator_clones(): check_estimator(est) assert_equal(old_hash, joblib.hash(est)) - est = Estimator() + with ignore_warnings(category=FutureWarning): + # when 'est = SGDClassifier()' + est = Estimator() set_checking_parameters(est) set_random_state(est) # with fitting