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

Correcting tests

parent 91e292f1
Branches
Tags
No related merge requests found
......@@ -8,7 +8,7 @@ class Test_genName(unittest.TestCase):
def test_late(self):
self.config = {"fusionType": "LateFusion",
"fusionMethod": "chicken_is_heaven",
"classifierNames": ["cheese", "is", "no", "disease"]}
"classifiersNames": ["cheese", "is", "no", "disease"]}
res = FusionModule.genName(self.config)
self.assertEqual(res, "Late-chic-chee-is-no-dise")
......@@ -119,3 +119,15 @@ class Test_genDirecortiesNames(unittest.TestCase):
directories = execution.genDirecortiesNames(cls.directory, cls.stats_iter)
cls.assertEqual(len(directories), 1)
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