add test nan

alexpantyukhin · alexpantyukhin · commit e209dfacc1d8 · 2018-04-28T10:31:56.000+04:00
diff --git a/sklearn/feature_selection/tests/test_from_model.py b/sklearn/feature_selection/tests/test_from_model.py
@@ -1,5 +1,6 @@
 import numpy as np
 
+from sklearn.impute import SimpleImputer
 from sklearn.utils.testing import assert_true
 from sklearn.utils.testing import assert_false
 from sklearn.utils.testing import assert_equal
@@ -202,8 +203,9 @@ def test_threshold_without_refitting():
 
 def test_transform_accepts_infinite_data():
     # Test that transform doesn't check for np.inf and np.nan values.
-    est = RandomForestClassifier()
+    est = SimpleImputer(strategy='mean', missing_values='NaN')
     model = SelectFromModel(estimator=est)
+    data.fill(np.nan)
     model.fit(data, y)
 
     X_len = len(model.get_support())
