def testCategorie_tag(self): tag = [1, 0] dataset_path = self.get_test_dataset_path() folder = dataset_path + "crosswalks/" cat = Categorie.from_default(folder, tag) cat.read_folder() self.assertEquals(cat.samples[0].tag, tag)
def testCategorie_read_folder(self): dataset_path = self.get_test_dataset_path() folder = dataset_path + "crosswalks/" cat = Categorie.from_default(folder, [1, 0]) cat.read_folder() self.assertEquals(len(cat.samples), 4)