def test_021(self): """ config resize is wrong format """ with pytest.raises(AttributeError): image = Image("files/0_100.jpg", dir='tmp', config=['resize=1']) with pytest.raises(AttributeError): image = Image("files/0_100.jpg", dir='tmp', config=['resize=a,2']) with pytest.raises(AttributeError): image = Image("files/0_100.jpg", dir='tmp', config=['resize=2,b'])
def test_033(self): """ thumbnail invalid """ with pytest.raises(AttributeError): image = Image("files/0_100.jpg", dir='tmp', config=['thumb=1']) with pytest.raises(AttributeError): image = Image("files/0_100.jpg", dir='tmp', config=['thumb=a,2']) with pytest.raises(AttributeError): image = Image("files/0_100.jpg", dir='tmp', config=['thumb=2,b'])
def test_032(self): """ thumbnail """ image = Image("files/0_100.jpg", config=['thumb=(32,32)']) self.assertTrue(os.path.isfile("0_100.h5")) self.assertEqual(image.thumb.shape, (32, 32, 3)) os.remove("0_100.h5") image = Image("files/0_100.jpg", config=['thumb=(16,16)']) self.assertTrue(os.path.isfile("0_100.h5")) self.assertEqual(image.thumb.shape, (16, 16, 3)) os.remove("0_100.h5")
def test_083(self): """ Image - rotate - rgb,negative degree """ image = Image("files/1_100.jpg", 1, config=['resize=(64,64)']) rotated = image.rotate(-30) self.assertTrue(rotated.shape, (64, 64)) rotated = image.rotate(-45) self.assertTrue(rotated.shape, (64, 64)) rotated = image.rotate(-60) self.assertTrue(rotated.shape, (64, 64)) os.remove('1_100.h5')
def test_029(self): """ tif file """ image = Image("files/6page.tif") self.assertEqual(image.name, "6page") self.assertEqual(image.type, "tif") os.remove("6page.h5") image = Image("files/text.tiff") self.assertEqual(image.name, "text") self.assertEqual(image.type, "tiff") os.remove("text.h5")
def test_081(self): """ Image - rotate - rgb """ image = Image("files/1_100.jpg", 1, config=['resize=(64,64)']) rotated = image.rotate(90) self.assertTrue(rotated.shape, (64, 64)) rotated = image.rotate(180) self.assertTrue(rotated.shape, (64, 64)) rotated = image.rotate(270) self.assertTrue(rotated.shape, (64, 64)) os.remove('1_100.h5')
def test_080(self): """ Image - rotate - grayscale, non-90 degree """ image = Image("files/1_100.jpg", 1, config=['resize=(64,64)', 'grayscale']) rotated = image.rotate(30) self.assertTrue(rotated.shape, (64, 64)) rotated = image.rotate(45) self.assertTrue(rotated.shape, (64, 64)) rotated = image.rotate(60) self.assertTrue(rotated.shape, (64, 64)) os.remove('1_100.h5')
def test_079(self): """ Image - gif file """ image = Image("files/gray.gif", 1) self.assertEquals(image.shape, (415, 506, 3)) image = Image("files/gray.gif", 1, config=['grayscale']) self.assertEquals(image.shape, (415, 506)) image = Image("files/rgb.gif", 1) self.assertEquals(image.shape, (561, 748, 3)) image = Image("files/rgb.gif", 1, config=['grayscale']) self.assertEquals(image.shape, (561, 748)) os.remove("gray.h5") os.remove("rgb.h5")
def test_091(self): """ Image - raw pixel input - gray to color """ pixels = cv2.imread('files/1_100.jpg', cv2.IMREAD_GRAYSCALE) image = Image(pixels, 1) self.assertEquals(image.name, 'untitled') self.assertEquals(image.type, 'raw') self.assertEquals(image.size, 10000) self.assertEquals(image.shape, (100, 100, 3)) image = Image(pixels, 1, config=['gray']) self.assertEquals(image.name, 'untitled') self.assertEquals(image.type, 'raw') self.assertEquals(image.size, 10000) self.assertEquals(image.shape, (100, 100)) os.remove('untitled.h5')
def test_090(self): """ Image - raw pixel input """ pixels = cv2.imread('files/1_100.jpg') image = Image(pixels, 1) self.assertEquals(image.name, 'untitled') self.assertEquals(image.type, 'raw') self.assertEquals(image.size, 30000) self.assertEquals(image.shape, (100, 100, 3)) image = Image(pixels, 1, config=['gray']) self.assertEquals(image.name, 'untitled') self.assertEquals(image.type, 'raw') self.assertEquals(image.size, 30000) self.assertEquals(image.shape, (100, 100)) os.remove('untitled.h5')
def test_069(self): """ Image - remote image """ image = Image( 'https://cdn.cnn.com/cnnnext/dam/assets/180727161452-trump-speech-economy-072718-exlarge-tease.jpg', 2) self.assertEqual(image.type, "jpg") self.assertEqual(image.size, 38302) self.assertEqual(image.shape, (438, 780, 3)) image = Image( 'https://cdn.cnn.com/cnnnext/dam/assets/180727161452-trump-speech-economy-072718-exlarge-tease.jpg', 2, config=['grayscale']) self.assertEqual(image.type, "jpg") self.assertEqual(image.size, 38302) self.assertEqual(image.shape, (438, 780)) os.remove('180727161452-trump-speech-economy-072718-exlarge-tease.h5')
def test_027(self): """ async processing """ image = Image("files/0_100.jpg", ehandler=self.done) time.sleep(3) self.assertTrue(self.isdone) os.remove("0_100.h5") self._isdone = False
def test_011(self): """ image dir is not-None - nonexist """ image = Image("files/0_100.jpg", dir='tmp') self.assertTrue(os.path.isfile("tmp/0_100.h5")) self.assertTrue(len(image.raw) > 0) self.assertEqual(image.thumb, None) os.remove("tmp/0_100.h5")
def test_014(self): """ label is valid """ image = Image("files/0_100.jpg", dir='tmp', label=16) self.assertEqual(image.label, 16) self.assertTrue(len(image.raw) > 0) self.assertEqual(image.thumb, None) os.remove("tmp/0_100.h5") os.rmdir("tmp")
def test_094(self): """ Image - ehandler with arguments """ image = Image('files/1_100.jpg', 1, ehandler=(self.done2, 6)) time.sleep(3) self.assertTrue(self.isdone) self.assertTrue(self.args, 6) os.remove("1_100.h5") self._isdone = False
def test_010(self): """ image dir is None """ image = Image("files/0_100.jpg", dir=None) self.assertEqual(image.image, "files/0_100.jpg") self.assertTrue(os.path.isfile("0_100.h5")) self.assertTrue(len(image.raw) > 0) self.assertEqual(image.thumb, None) os.remove("0_100.h5")
def test_071(self): """ Image - bad image """ f = open("tmp.jpg", "w") f.write("foobar") f.close() image = Image('tmp.jpg', 2) self.assertEqual(image.data, None) os.remove('tmp.jpg')
def test_026(self): """ config - nostore """ image = Image("files/0_100.jpg", config=['flat', 'nostore']) 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) self.assertFalse(os.path.isfile("0_100.h5"))
def test_025(self): """ config - flatten """ image = Image("files/0_100.jpg", config=['flat']) 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) self.assertTrue(os.path.isfile("0_100.h5")) os.remove("0_100.h5") image = Image("files/0_100.jpg", config=['flatten']) 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) self.assertTrue(os.path.isfile("0_100.h5")) os.remove("0_100.h5")
def test_024(self): """ config - resize """ image = Image("files/0_100.jpg", config=['resize=50,50']) self.assertEqual(image.name, "0_100") self.assertEqual(image.type, "jpg") self.assertEqual(image.dir, "./") self.assertEqual(image.size, 3643) self.assertEqual(image.shape, (50, 50, 3)) self.assertEqual(image.label, 0) self.assertTrue(os.path.isfile("0_100.h5")) os.remove("0_100.h5") image = Image("files/0_100.jpg", config=['resize=(200,200)']) self.assertEqual(image.name, "0_100") self.assertEqual(image.type, "jpg") self.assertEqual(image.dir, "./") self.assertEqual(image.size, 3643) self.assertEqual(image.shape, (200, 200, 3)) self.assertEqual(image.label, 0) self.assertTrue(os.path.isfile("0_100.h5")) os.remove("0_100.h5")
def test_099(self): """ Images - += Image """ images = Images(['files/1_100.jpg'], 1) image = Image('files/2_100.jpg', 2, config=['nostore']) images += image self.assertEquals(len(images), 2) self.assertEquals(images[0].name, '1_100') self.assertEquals(images[1].name, '2_100') self.assertEquals(images[0].label, 1) self.assertEquals(images[1].label, 2) os.remove("collection.1_100.h5")
def test_023(self): """ config - grayscale to RGB """ image = Image("files/0_100g.jpg") self.assertEqual(image.image, "files/0_100g.jpg") self.assertEqual(image.name, "0_100g") self.assertEqual(image.type, "jpg") self.assertEqual(image.dir, "./") self.assertEqual(image.size, 4253) self.assertEqual(image.shape, (100, 100, 3)) self.assertEqual(image.label, 0) self.assertTrue(os.path.isfile("0_100g.h5")) os.remove("0_100g.h5")
def test_022(self): """ config - RGB to grayscale """ image = Image("files/0_100.jpg", config=['gray']) self.assertEqual(image.image, "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, (100, 100)) self.assertEqual(image.label, 0) self.assertTrue(os.path.isfile("0_100.h5")) os.remove("0_100.h5") image = Image("files/0_100.jpg", config=['grayscale']) self.assertEqual(image.image, "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, (100, 100)) self.assertEqual(image.label, 0) self.assertTrue(os.path.isfile("0_100.h5")) os.remove("0_100.h5")
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_084(self): """ Image - rotate invalid """ image = Image("files/1_100.jpg", 1, config=['resize=(64,64)']) with pytest.raises(ValueError): image.rotate(-360) with pytest.raises(ValueError): image.rotate(360) os.remove('1_100.h5')
def test_009(self): """ image properties """ image = Image("files/0_100.jpg") self.assertEqual(image.image, "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, (100, 100, 3)) self.assertEqual(image.label, 0) self.assertTrue(len(image.raw) > 0) self.assertEqual(image.thumb, None) self.assertTrue(os.path.isfile("0_100.h5")) os.remove("0_100.h5")
def test_016(self): """ config is empty """ image = Image("files/0_100.jpg", config=[]) self.assertEqual(image.image, "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, (100, 100, 3)) self.assertEqual(image.label, 0) self.assertTrue(os.path.isfile("0_100.h5")) self.assertTrue(len(image.raw) > 0) self.assertEqual(image.thumb, None) os.remove("0_100.h5")
def test_070(self): """ Image - nonexistent remote image """ image = Image( 'https://cdn.cnn.com/cnnnext/dam/assets/18ch-economy-072718-exlarge-tease.jpg', 2) self.assertEqual(image.data, None)
def test_008(self): """ Image constructor - config is not a list """ with pytest.raises(TypeError): image = Image(label='c')
def test_007(self): """ Image constructor - label is not an int """ with pytest.raises(TypeError): image = Image(label='c')