Merge pull request #5242 from arjoly/stable-test-2

glouppe · glouppe · commit 7e2453ac48f6 · 2015-09-11T09:16:08.000+02:00
FIX unstable test due to bootstrap and unset random state
diff --git a/sklearn/ensemble/tests/test_forest.py b/sklearn/ensemble/tests/test_forest.py
@@ -980,11 +980,12 @@ def test_warm_start_oob():
         yield check_warm_start_oob, name
 
 
-def test_dtype_convert():
-    classifier = RandomForestClassifier()
-    CLASSES = 15
-    X = np.eye(CLASSES)
-    y = [ch for ch in 'ABCDEFGHIJKLMNOPQRSTU'[:CLASSES]]
+def test_dtype_convert(n_classes=15):
+    classifier = RandomForestClassifier(random_state=0, bootstrap=False)
+
+    X = np.eye(n_classes)
+    y = [ch for ch in 'ABCDEFGHIJKLMNOPQRSTU'[:n_classes]]
 
     result = classifier.fit(X, y).predict(X)
+    assert_array_equal(classifier.classes_, y)
     assert_array_equal(result, y)
