From 3cd004350dd9b4b7ce1cc3d08ea92b816349eb85 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= Date: Sun, 17 May 2020 18:55:33 +0200 Subject: [PATCH] Fix failing tests --- tests/Stree_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/Stree_test.py b/tests/Stree_test.py index 432ea6c..8fe5785 100644 --- a/tests/Stree_test.py +++ b/tests/Stree_test.py @@ -133,7 +133,7 @@ class Stree_test(unittest.TestCase): def test_score(self): X, y = self._get_Xy() - accuracy_score = self._clf.score(X, y, print_out=False) + accuracy_score = self._clf.score(X, y) yp = self._clf.predict(X) right = (yp == y).astype(int) accuracy_computed = sum(right) / len(y) @@ -178,8 +178,8 @@ class Stree_test(unittest.TestCase): model_predictions.predict(X).tolist(), model_hyperplane.predict(X).tolist() ) - a = model_predictions.score(X, y, print_out=False), - b = model_hyperplane.score(X, y, print_out=False) + a = model_predictions.score(X, y), + b = model_hyperplane.score(X, y) self.assertEqual(a, b) self.assertGreater(b, .95)