Skip to content
Snippets Groups Projects
Commit dee8b55c authored by Baptiste Bauvin's avatar Baptiste Bauvin
Browse files

Correcting tests

parent 91e292f1
No related branches found
No related tags found
No related merge requests found
...@@ -8,7 +8,7 @@ class Test_genName(unittest.TestCase): ...@@ -8,7 +8,7 @@ class Test_genName(unittest.TestCase):
def test_late(self): def test_late(self):
self.config = {"fusionType": "LateFusion", self.config = {"fusionType": "LateFusion",
"fusionMethod": "chicken_is_heaven", "fusionMethod": "chicken_is_heaven",
"classifierNames": ["cheese", "is", "no", "disease"]} "classifiersNames": ["cheese", "is", "no", "disease"]}
res = FusionModule.genName(self.config) res = FusionModule.genName(self.config)
self.assertEqual(res, "Late-chic-chee-is-no-dise") self.assertEqual(res, "Late-chic-chee-is-no-dise")
...@@ -119,3 +119,15 @@ class Test_genDirecortiesNames(unittest.TestCase): ...@@ -119,3 +119,15 @@ class Test_genDirecortiesNames(unittest.TestCase):
directories = execution.genDirecortiesNames(cls.directory, cls.stats_iter) directories = execution.genDirecortiesNames(cls.directory, cls.stats_iter)
cls.assertEqual(len(directories), 1) cls.assertEqual(len(directories), 1)
cls.assertEqual(directories[0], "../chicken_is_heaven/") cls.assertEqual(directories[0], "../chicken_is_heaven/")
class Test_genArgumentDictionaries(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.labelsDictionary = {0:"yes", 1:"No", 2:"Maybe"}
cls.direcories = ["Res/iter_1", "Res/iter_2"]
cls.multiclassLabels = [np.array([0, 1, -100, 1, 0]),
np.array([1, 0, -100, 1, 0]),
np.array([0, 1, -100, 0, 1])]
cls.labelsCombinations = [[0,1], [0,2], [1,2]]
cls.indicesMulticlass = [[[[], []], [[], []], [[], []]], [[], [], []]]
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment