test results & fix single predict_proba

2025-08-16 07:56:06 +00:00 · 2020-05-18 17:08:05 +02:00
parent 382ae921ab
commit 68512b3d75
4 changed files with 113 additions and 49 deletions
--- a/tests/Stree_test.py
+++ b/tests/Stree_test.py
@@ -148,7 +148,7 @@ class Stree_test(unittest.TestCase):
        yp = self._clf.predict_proba(X[28, :].reshape(-1, X.shape[1]))
        self.assertEqual(0, yp[0:, 0])
        self.assertEqual(1, y[28])
-        self.assertEqual(0.29026400765649235, yp[0, 1])
+        self.assertEqual(0.29026400766, round(yp[0, 1], 11))

    def test_multiple_predict_proba(self):
        # First 27 elements the predictions are the same as the truth