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

Add actions and complete tests

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2022-12-13 10:46:28 +01:00
parent f6fdc96c6d
commit 139d080df1
5 changed files with 110 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -74,6 +74,18 @@ class FImdlpTest(unittest.TestCase):
clf.fit([[1, 2], [3, 4]], [1, 2], features=["a", "b", "c"])
with self.assertRaises(ValueError):
clf.fit([[1, 2], [3, 4]], [1, 2], unexpected="class_name")
with self.assertRaises(ValueError):
clf.fit([[1, 2], [3, 4]], [1, 2], features="01")
with self.assertRaises(ValueError):
clf.fit([[1, 2], [3, 4]], [1, 2], features=[0, 0])
with self.assertRaises(ValueError):
clf.fit([[1, 2], [3, 4]], [1, 2], features=[0, 2])
def test_fit_features(self):
clf = FImdlp()
clf.fit([[1, 2], [3, 4]], [1, 2], features=[0])
res = clf.transform([[1, 2], [3, 4]])
self.assertListEqual(res.tolist(), [[0, 2], [0, 4]])
def test_transform_original(self):
clf = FImdlp(proposal=False)