Doctorado-ML/fimdlp
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/FImdlp.git synced 2025-08-18 00:45:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix tests because stable_sort in c++

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-12-13 17:16:23 +01:00
parent 0ca507c692
commit 3ae0d67884
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/fimdlp/tests/FImdlp_test.py
View File

@@ -60,11 +60,12 @@ class FImdlpTest(unittest.TestCase):
self.assertEqual(clf.n_features_, 4) self.assertEqual(clf.n_features_, 4)
self.assertTrue(np.array_equal(X, clf.X_)) self.assertTrue(np.array_equal(X, clf.X_))
self.assertTrue(np.array_equal(y, clf.y_)) self.assertTrue(np.array_equal(y, clf.y_))
expected = [ expected = [
[5.5, 5.800000190734863], [5.5, 5.800000190734863],
[3.0999999046325684], [2.9000000953674316, 3.3499999046325684],
[2.450000047683716, 4.800000190734863, 5.099999904632568], [2.450000047683716, 4.800000190734863],
[0.800000011920929, 1.7000000476837158], [0.800000011920929, 1.7999999523162842],
] ]
self.assertListEqual(expected, clf.get_cut_points()) self.assertListEqual(expected, clf.get_cut_points())
self.assertListEqual([0, 1, 2, 3], clf.features_) self.assertListEqual([0, 1, 2, 3], clf.features_)
@@ -108,11 +109,11 @@ class FImdlpTest(unittest.TestCase):
) )
expected = [ expected = [
[0, 0, 1, 1], [0, 0, 1, 1],
[2, 0, 1, 1], [2, 1, 1, 1],
[1, 0, 1, 1], [1, 0, 1, 1],
[0, 0, 1, 1], [0, 0, 1, 1],
[1, 0, 1, 1], [1, 0, 1, 1],
[1, 0, 1, 1], [1, 1, 1, 1],
[1, 0, 1, 1], [1, 0, 1, 1],
] ]
self.assertTrue(np.array_equal(clf.transform(X[90:97]), expected)) self.assertTrue(np.array_equal(clf.transform(X[90:97]), expected))