From 3ae0d678845aa891adf7b05bebef40c4356eaa91 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= Date: Tue, 13 Dec 2022 17:16:23 +0100 Subject: [PATCH] Fix tests because stable_sort in c++ --- src/fimdlp/tests/FImdlp_test.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/fimdlp/tests/FImdlp_test.py b/src/fimdlp/tests/FImdlp_test.py index 7eb282b..c6d01f0 100644 --- a/src/fimdlp/tests/FImdlp_test.py +++ b/src/fimdlp/tests/FImdlp_test.py @@ -60,11 +60,12 @@ class FImdlpTest(unittest.TestCase): self.assertEqual(clf.n_features_, 4) self.assertTrue(np.array_equal(X, clf.X_)) self.assertTrue(np.array_equal(y, clf.y_)) + expected = [ [5.5, 5.800000190734863], - [3.0999999046325684], - [2.450000047683716, 4.800000190734863, 5.099999904632568], - [0.800000011920929, 1.7000000476837158], + [2.9000000953674316, 3.3499999046325684], + [2.450000047683716, 4.800000190734863], + [0.800000011920929, 1.7999999523162842], ] self.assertListEqual(expected, clf.get_cut_points()) self.assertListEqual([0, 1, 2, 3], clf.features_) @@ -108,11 +109,11 @@ class FImdlpTest(unittest.TestCase): ) expected = [ [0, 0, 1, 1], - [2, 0, 1, 1], + [2, 1, 1, 1], [1, 0, 1, 1], [0, 0, 1, 1], [1, 0, 1, 1], - [1, 0, 1, 1], + [1, 1, 1, 1], [1, 0, 1, 1], ] self.assertTrue(np.array_equal(clf.transform(X[90:97]), expected))