FIX AdaBoost ZeroDivisionError in proba #7501 - tests corrected

Dominik Krzeminski · Dominik Krzeminski · commit 42f478feb3d2 · 2017-02-16T12:00:21.000+01:00
diff --git a/sklearn/ensemble/tests/test_weight_boosting.py b/sklearn/ensemble/tests/test_weight_boosting.py
@@ -73,11 +73,13 @@ def predict_proba(self, X):
     assert_array_equal(np.argmin(samme_proba, axis=1), [2, 0, 0, 2])
     assert_array_equal(np.argmax(samme_proba, axis=1), [0, 1, 1, 1])
 
+
 def test_oneclass_proba():
     # Test `predict_proba` robustness for one class label input.
-    y_t = np.ones((X.shape[0],))
+    y_t = np.ones((len(X),))
     clf = AdaBoostClassifier().fit(X, y_t)
-    assert_array_equal(clf.predict_proba(X), np.ones((X.shape[0],1)))
+    assert_array_equal(clf.predict_proba(X), np.ones((X.shape[0], 1)))
+
 
 def test_classification_toy():
     # Check classification on a toy dataset.
