Add an hardcoded test

MRXLT · Oct 21, 2015 · be21b6a · be21b6a
1 parent bc02bd5
commit be21b6a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/sklearn/tree/tests/test_tree.py b/sklearn/tree/tests/test_tree.py
@@ -1337,6 +1337,14 @@ def test_presort_sparse():
         yield check_presort_sparse, est, sparse_matrix(X), y
 
 
+def test_decision_path_hardcoded():
+    X = iris.data
+    y = iris.target
+    est = DecisionTreeClassifier(random_state=0, max_depth=1).fit(X, y)
+    node_indicator = est.decision_path(X[:2]).toarray()
+    assert_array_equal(node_indicator, [[1, 1, 0], [1, 0, 1]])
+
+
 def check_decision_path(name):
     X = iris.data
     y = iris.target