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')
