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)