class TestImage(unittest.TestCase):
def setUp(self):
self.im1 = Image(random.random((1300, 1200)))
self.im3 = Image(random.random((1300, 1200, 3)))
self.im4 = Image(random.random((1300, 1200, 4)))
self.images = [self.im1, self.im3, self.im4]
def test_size(self):
for im in self.images:
self.assertEqual(im.width, im.shape[1], 1200)
self.assertEqual(im.height, im.shape[0], 1300)
def test_resize(self):
for im in self.images:
smaller = im.resize(0, 400)
bigger = im.resize(0, 2400)
self.assertEqual(smaller.height, 400)
self.assertEqual(bigger.height, 2400)
def test_rotate(self):
pass
def test_save(self):
self.im4.save("tmp.png")
im = imread("tmp.png")
self.assertEqual(im.shape, self.im4.shape)
def test_at(self):
self.assertEqual(self.im4.at(70, 50), self.im4[50, 70])
def test_to_gray(self):
for i in self.images:
g = i.to_gray()
self.assertEqual(g.channels, 1)
def test_percent(self):
self.assertEqual((600, 650), self.im3.percent(50, 50))
def test_op(self):
for i in self.images:
self.assertTrue((i.op('self + 1') == 1).all())
def test_clip_add(self):
self.assertTrue((self.im3.clip_add(10) >= self.im3).all())
def setUp(self):
self.im1 = Image(random.random((1300, 1200)))
self.im3 = Image(random.random((1300, 1200, 3)))
self.im4 = Image(random.random((1300, 1200, 4)))
self.images = [self.im1, self.im3, self.im4]
