def test_getImages(self): imageDatabase.wipeDatabase() imageDatabase.addImage("test1") imageDatabase.addImage("test2") imageDatabase.addImage("test3") numImages = imageDatabase.getImages() self.assertEqual(len(numImages), 3)
def test_wipeDatabase(self): imageDatabase.addImage("test1") imageDatabase.addImage("test2") imageDatabase.addImage("test3") imageDatabase.wipeDatabase() images = imageDatabase.getImages() numImages = len(images) self.assertEqual(numImages, 0)