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)