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]