def test_076(self): """ Image - load() attr type and size """ image = Image("files/0_100.jpg", 1) image = Image() image.load('0_100.h5') self.assertEquals(image.type, 'jpg') self.assertEquals(image.size, 3643) os.remove('0_100.h5')
def test_031(self): """ config - load """ image = Image("files/0_100.jpg", config=['flat']) self.assertTrue(os.path.isfile("0_100.h5")) image = None image = Image() image.load("files/0_100.jpg") self.assertEqual(image.name, "0_100") self.assertEqual(image.type, "jpg") self.assertEqual(image.dir, "./") self.assertEqual(image.size, 3643) self.assertEqual(image.shape, (30000, )) self.assertEqual(image.label, 0) os.remove("0_100.h5")