FIX/TST remove ExtraTree from test

Guillaume Lemaitre · Guillaume Lemaitre · commit 225995f35b88 · 2017-03-13T20:55:47.000+01:00
diff --git a/sklearn/tree/tests/test_tree.py b/sklearn/tree/tests/test_tree.py
@@ -1191,10 +1191,12 @@ def test_behaviour_constant_feature_after_splits():
                                np.zeros((4, 11)))))
     y = [0, 0, 0, 1, 1, 2, 2, 2, 3, 3, 3]
     for name, TreeEstimator in ALL_TREES.items():
-        est = TreeEstimator(random_state=0, max_features=1)
-        est.fit(X, y)
-        assert_equal(est.tree_.max_depth, 2)
-        assert_equal(est.tree_.node_count, 5)
+        # do not check extra random trees
+        if not "ExtraTree" in name:
+            est = TreeEstimator(random_state=0, max_features=1)
+            est.fit(X, y)
+            assert_equal(est.tree_.max_depth, 2)
+            assert_equal(est.tree_.node_count, 5)
 
 
 def test_with_only_one_non_constant_features():
