Added tests and fixed flake8

Sentient07 · Sentient07 · commit 47f0c20bdd67 · 2017-06-21T01:04:25.000+05:30
diff --git a/sklearn/linear_model/__init__.py b/sklearn/linear_model/__init__.py
@@ -8,7 +8,7 @@
 # See http://scikit-learn.sourceforge.net/modules/sgd.html and
 # http://scikit-learn.sourceforge.net/modules/linear_model.html for
 # complete documentation.
-import warnings
+
 from .base import LinearRegression
 
 from .bayes import BayesianRidge, ARDRegression
@@ -31,11 +31,6 @@
 from .passive_aggressive import PassiveAggressiveRegressor
 from .perceptron import Perceptron
 
-with warnings.catch_warnings():
-    warnings.simplefilter('ignore')
-    from .randomized_l1 import (RandomizedLasso, RandomizedLogisticRegression,
-                                lasso_stability_path)
-
 from .ransac import RANSACRegressor
 from .theil_sen import TheilSenRegressor
 
diff --git a/sklearn/linear_model/tests/test_randomized_l1.py b/sklearn/linear_model/tests/test_randomized_l1.py
@@ -12,8 +12,8 @@
 from sklearn.utils.testing import assert_raises_regex
 from sklearn.utils.testing import assert_allclose
 from sklearn.utils.testing import ignore_warnings
+from sklearn.utils.testing import assert_warns_message
 
-import warnings
 from sklearn.linear_model.randomized_l1 import(lasso_stability_path,
                                                 RandomizedLasso,
                                                 RandomizedLogisticRegression)
@@ -32,6 +32,7 @@
 # test that the feature score of the best features
 F, _ = f_regression(X, y)
 
+
 @ignore_warnings(category=DeprecationWarning)
 def test_lasso_stability_path():
     # Check lasso stability path
@@ -186,3 +187,31 @@ def test_randomized_logistic_sparse():
                                        tol=1e-3)
     feature_scores_sp = clf.fit(X_sp, y).scores_
     assert_array_equal(feature_scores, feature_scores_sp)
+
+
+def test_warning_raised():
+
+    scaling = 0.3
+    selection_threshold = 0.5
+    tempdir = 5
+    assert_warns_message(DeprecationWarning, "The function"
+                         " lasso_stability_path is "
+                         "deprecated in 0.19 and will be removed in 0.21.",
+                         lasso_stability_path, X, y, scaling=scaling,
+                         random_state=42, n_resampling=30)
+
+    assert_warns_message(DeprecationWarning, "The class"
+                         " RandomizedLasso is "
+                         "deprecated in 0.19 and will be removed in 0.21.",
+                         RandomizedLasso, verbose=False, alpha=[1, 0.8],
+                         random_state=42, scaling=scaling,
+                         selection_threshold=selection_threshold,
+                         memory=tempdir)
+
+    assert_warns_message(DeprecationWarning, "The function"
+                         " log_multivariate_normal_density is "
+                         "deprecated in 0.19 and will be removed in 0.21.",
+                         RandomizedLogisticRegression,
+                         verbose=False, C=1., random_state=42,
+                         scaling=scaling, n_resampling=50,
+                         tol=1e-3)
