def setUp(self): self.ImageDataset = ImageDataset() self.path = pynopticon.tests.__path__[0] self.setInvalid = [ './testimgs/nonexisting.jpg', './testimgs/test1.jpg' ] self.setValid = ['./testimgs/test1.jpg', './testimgs/test2.jpg']
def setUp(self): self.ImageDataset = ImageDataset() self.path = pynopticon.tests.__path__[0] self.setInvalid = ['./testimgs/nonexisting.jpg', './testimgs/test1.jpg'] self.setValid = ['./testimgs/test1.jpg', './testimgs/test2.jpg']
class TestImageDataset(unittest.TestCase): def setUp(self): self.ImageDataset = ImageDataset() self.path = pynopticon.tests.__path__[0] self.setInvalid = [ './testimgs/nonexisting.jpg', './testimgs/test1.jpg' ] self.setValid = ['./testimgs/test1.jpg', './testimgs/test2.jpg'] def addCategories(self, set=None): if not set: set = self.setInvalid self.ImageDataset.addCategory() self.ImageDataset.addCategory('test1') self.ImageDataset.addCategory('test2', set) def testAddCategories(self): self.addCategories() # self.ImageDataset.saveToXML('test.xml') self.checkConsistency() def checkConsistency(self): self.assertEqual(self.ImageDataset.categories[0].name, '') self.assertEqual(self.ImageDataset.categories[1].name, 'test1') self.assertEqual(self.ImageDataset.categories[2].name, 'test2') self.assertEqual(self.ImageDataset.categories[2].fnames, self.setInvalid) def testDelCategories(self): self.addCategories() self.ImageDataset.delCategory(0) self.assertEqual(len(self.ImageDataset.categories), 2) self.assertEqual(self.ImageDataset.categories[0].name, 'test1') self.assertRaises(IndexError, self.ImageDataset.delCategory, 2) def testLoadFromXML(self, fname=None): if not fname: self.testSaveToXML() fname = os.path.join(self.path, 'test_saved.xml') self.ImageDataset.loadFromXML(fname) self.checkConsistency() def testSaveToXML(self): self.addCategories() self.ImageDataset.saveToXML(os.path.join(self.path, 'test_saved.xml')) del self.ImageDataset self.setUp() self.ImageDataset.loadFromXML(os.path.join(self.path, 'test_saved.xml')) self.checkConsistency() def testAddAndDelFnames(self): self.addCategories() self.ImageDataset.categories[1].addDir( os.path.join(pynopticon.__path__[0], 'datasets', 'testimgs')) self.assertEqual(self.ImageDataset.categories[1].fnames, self.setValid) self.ImageDataset.categories[1].delFile('./testimgs/test1.jpg') self.assertEqual(self.ImageDataset.categories[1].fnames, ['./testimgs/test2.jpg']) self.ImageDataset.categories[1].delID(0) self.assertEqual(len(self.ImageDataset.categories[1].fnames), 0) def testAccessNoContainer(self): self.addCategories() self.assertRaises(AttributeError, self.ImageDataset.__iter__) def testIterateFnames(self): self.addCategories() for x, y in zip( self.ImageDataset.categories[2], ['./testimgs/nonexisting.jpg', './testimgs/test1.jpg']): self.assertEqual(x, y) def testSeqContainerValid(self): self.addCategories(set=self.setValid) self.ImageDataset.prepare() seqContainer = self.ImageDataset.outputSlotTrain def testSeqContainerValidIter(self): self.addCategories(set=self.setValid) self.ImageDataset.prepare() seqContainer = self.ImageDataset.outputSlotTrain def testSeqContainerInvalid(self): self.addCategories()
class TestImageDataset(unittest.TestCase): def setUp(self): self.ImageDataset = ImageDataset() self.path = pynopticon.tests.__path__[0] self.setInvalid = ['./testimgs/nonexisting.jpg', './testimgs/test1.jpg'] self.setValid = ['./testimgs/test1.jpg', './testimgs/test2.jpg'] def addCategories(self, set = None): if not set: set = self.setInvalid self.ImageDataset.addCategory() self.ImageDataset.addCategory('test1') self.ImageDataset.addCategory('test2', set) def testAddCategories(self): self.addCategories() # self.ImageDataset.saveToXML('test.xml') self.checkConsistency() def checkConsistency(self): self.assertEqual(self.ImageDataset.categories[0].name, '') self.assertEqual(self.ImageDataset.categories[1].name, 'test1') self.assertEqual(self.ImageDataset.categories[2].name, 'test2') self.assertEqual(self.ImageDataset.categories[2].fnames, self.setInvalid) def testDelCategories(self): self.addCategories() self.ImageDataset.delCategory(0) self.assertEqual(len(self.ImageDataset.categories), 2) self.assertEqual(self.ImageDataset.categories[0].name, 'test1') self.assertRaises(IndexError, self.ImageDataset.delCategory, 2) def testLoadFromXML(self, fname=None): if not fname: self.testSaveToXML() fname = os.path.join(self.path, 'test_saved.xml') self.ImageDataset.loadFromXML(fname) self.checkConsistency() def testSaveToXML(self): self.addCategories() self.ImageDataset.saveToXML(os.path.join(self.path, 'test_saved.xml')) del self.ImageDataset self.setUp() self.ImageDataset.loadFromXML(os.path.join(self.path, 'test_saved.xml')) self.checkConsistency() def testAddAndDelFnames(self): self.addCategories() self.ImageDataset.categories[1].addDir(os.path.join(pynopticon.__path__[0], 'datasets', 'testimgs')) self.assertEqual(self.ImageDataset.categories[1].fnames, self.setValid) self.ImageDataset.categories[1].delFile('./testimgs/test1.jpg') self.assertEqual(self.ImageDataset.categories[1].fnames, ['./testimgs/test2.jpg']) self.ImageDataset.categories[1].delID(0) self.assertEqual(len(self.ImageDataset.categories[1].fnames), 0) def testAccessNoContainer(self): self.addCategories() self.assertRaises(AttributeError, self.ImageDataset.__iter__) def testIterateFnames(self): self.addCategories() for x,y in zip(self.ImageDataset.categories[2], ['./testimgs/nonexisting.jpg', './testimgs/test1.jpg']): self.assertEqual(x,y) def testSeqContainerValid(self): self.addCategories(set = self.setValid) self.ImageDataset.prepare() seqContainer = self.ImageDataset.outputSlotTrain def testSeqContainerValidIter(self): self.addCategories(set = self.setValid) self.ImageDataset.prepare() seqContainer = self.ImageDataset.outputSlotTrain def testSeqContainerInvalid(self): self.addCategories()