from helper import unittest, PillowTestCase, hopper from PIL import Image from PIL import ImageChops BLACK = (0, 0, 0) BROWN = (127, 64, 0) CYAN = (0, 255, 255) DARK_GREEN = (0, 128, 0) GREEN = (0, 255, 0) ORANGE = (255, 128, 0) WHITE = (255, 255, 255) GREY = 128 class TestImageChops(PillowTestCase): def test_sanity(self): im = hopper("L") ImageChops.constant(im, 128) ImageChops.duplicate(im) ImageChops.invert(im) ImageChops.lighter(im, im) ImageChops.darker(im, im) ImageChops.difference(im, im) ImageChops.multiply(im, im) ImageChops.screen(im, im) ImageChops.add(im, im) ImageChops.add(im, im, 2.0) ImageChops.add(im, im, 2.0, 128) ImageChops.subtract(im, im) ImageChops.subtract(im, im, 2.0) ImageChops.subtract(im, im, 2.0, 128) ImageChops.add_modulo(im, im) ImageChops.subtract_modulo(im, im) ImageChops.blend(im, im, 0.5) ImageChops.composite(im, im, im) ImageChops.offset(im, 10) ImageChops.offset(im, 10, 20) def test_add(self): # Arrange im1 = Image.open("Tests/images/imagedraw_ellipse_RGB.png") im2 = Image.open("Tests/images/imagedraw_floodfill.png") # Act new = ImageChops.add(im1, im2) # Assert self.assertEqual(new.getbbox(), (25, 25, 76, 76)) self.assertEqual(new.getpixel((50, 50)), ORANGE) def test_add_scale_offset(self): # Arrange im1 = Image.open("Tests/images/imagedraw_ellipse_RGB.png") im2 = Image.open("Tests/images/imagedraw_floodfill.png") # Act new = ImageChops.add(im1, im2, scale=2.5, offset=100) # Assert self.assertEqual(new.getbbox(), (0, 0, 100, 100)) self.assertEqual(new.getpixel((50, 50)), (202, 151, 100)) def test_add_clip(self): # Arrange im = hopper() # Act new = ImageChops.add(im, im) # Assert self.assertEqual(new.getpixel((50, 50)), (255, 255, 254)) def test_add_modulo(self): # Arrange im1 = Image.open("Tests/images/imagedraw_ellipse_RGB.png") im2 = Image.open("Tests/images/imagedraw_floodfill.png") # Act new = ImageChops.add_modulo(im1, im2) # Assert self.assertEqual(new.getbbox(), (25, 25, 76, 76)) self.assertEqual(new.getpixel((50, 50)), ORANGE) def test_add_modulo_no_clip(self): # Arrange im = hopper() # Act new = ImageChops.add_modulo(im, im) # Assert self.assertEqual(new.getpixel((50, 50)), (224, 76, 254)) def test_blend(self): # Arrange im1 = Image.open("Tests/images/imagedraw_ellipse_RGB.png") im2 = Image.open("Tests/images/imagedraw_floodfill.png") # Act new = ImageChops.blend(im1, im2, 0.5) # Assert self.assertEqual(new.getbbox(), (25, 25, 76, 76)) self.assertEqual(new.getpixel((50, 50)), BROWN) def test_constant(self): # Arrange im = Image.new("RGB", (20, 10)) # Act new = ImageChops.constant(im, GREY) # Assert self.assertEqual(new.size, im.size) self.assertEqual(new.getpixel((0, 0)), GREY) self.assertEqual(new.getpixel((19, 9)), GREY) def test_darker_image(self): # Arrange im1 = Image.open("Tests/images/imagedraw_chord_RGB.png") im2 = Image.open("Tests/images/imagedraw_outline_chord_RGB.png") # Act new = ImageChops.darker(im1, im2) # Assert self.assert_image_equal(new, im2) def test_darker_pixel(self): # Arrange im1 = hopper() im2 = Image.open("Tests/images/imagedraw_chord_RGB.png") # Act new = ImageChops.darker(im1, im2) # Assert self.assertEqual(new.getpixel((50, 50)), (240, 166, 0)) def test_difference(self): # Arrange im1 = Image.open("Tests/images/imagedraw_arc_end_le_start.png") im2 = Image.open("Tests/images/imagedraw_arc_no_loops.png") # Act new = ImageChops.difference(im1, im2) # Assert self.assertEqual(new.getbbox(), (25, 25, 76, 76)) def test_difference_pixel(self): # Arrange im1 = hopper() im2 = Image.open("Tests/images/imagedraw_polygon_kite_RGB.png") # Act new = ImageChops.difference(im1, im2) # Assert self.assertEqual(new.getpixel((50, 50)), (240, 166, 128)) def test_duplicate(self): # Arrange im = hopper() # Act new = ImageChops.duplicate(im) # Assert self.assert_image_equal(new, im) def test_invert(self): # Arrange im = Image.open("Tests/images/imagedraw_floodfill.png") # Act new = ImageChops.invert(im) # Assert self.assertEqual(new.getbbox(), (0, 0, 100, 100)) self.assertEqual(new.getpixel((0, 0)), WHITE) self.assertEqual(new.getpixel((50, 50)), CYAN) def test_lighter_image(self): # Arrange im1 = Image.open("Tests/images/imagedraw_chord_RGB.png") im2 = Image.open("Tests/images/imagedraw_outline_chord_RGB.png") # Act new = ImageChops.lighter(im1, im2) # Assert self.assert_image_equal(new, im1) def test_lighter_pixel(self): # Arrange im1 = hopper() im2 = Image.open("Tests/images/imagedraw_chord_RGB.png") # Act new = ImageChops.lighter(im1, im2) # Assert self.assertEqual(new.getpixel((50, 50)), (255, 255, 127)) def test_multiply_black(self): """If you multiply an image with a solid black image, the result is black.""" # Arrange im1 = hopper() black = Image.new("RGB", im1.size, "black") # Act new = ImageChops.multiply(im1, black) # Assert self.assert_image_equal(new, black) def test_multiply_green(self): # Arrange im = Image.open("Tests/images/imagedraw_floodfill.png") green = Image.new("RGB", im.size, "green") # Act new = ImageChops.multiply(im, green) # Assert self.assertEqual(new.getbbox(), (25, 25, 76, 76)) self.assertEqual(new.getpixel((25, 25)), DARK_GREEN) self.assertEqual(new.getpixel((50, 50)), BLACK) def test_multiply_white(self): """If you multiply with a solid white image, the image is unaffected.""" # Arrange im1 = hopper() white = Image.new("RGB", im1.size, "white") # Act new = ImageChops.multiply(im1, white) # Assert self.assert_image_equal(new, im1) def test_offset(self): # Arrange im = Image.open("Tests/images/imagedraw_ellipse_RGB.png") xoffset = 45 yoffset = 20 # Act new = ImageChops.offset(im, xoffset, yoffset) # Assert self.assertEqual(new.getbbox(), (0, 45, 100, 96)) self.assertEqual(new.getpixel((50, 50)), BLACK) self.assertEqual(new.getpixel((50+xoffset, 50+yoffset)), DARK_GREEN) # Test no yoffset self.assertEqual(ImageChops.offset(im, xoffset), ImageChops.offset(im, xoffset, xoffset)) def test_screen(self): # Arrange im1 = Image.open("Tests/images/imagedraw_ellipse_RGB.png") im2 = Image.open("Tests/images/imagedraw_floodfill.png") # Act new = ImageChops.screen(im1, im2) # Assert self.assertEqual(new.getbbox(), (25, 25, 76, 76)) self.assertEqual(new.getpixel((50, 50)), ORANGE) def test_subtract(self): # Arrange im1 = Image.open("Tests/images/imagedraw_chord_RGB.png") im2 = Image.open("Tests/images/imagedraw_outline_chord_RGB.png") # Act new = ImageChops.subtract(im1, im2) # Assert self.assertEqual(new.getbbox(), (25, 50, 76, 76)) self.assertEqual(new.getpixel((50, 50)), GREEN) self.assertEqual(new.getpixel((50, 51)), BLACK) def test_subtract_scale_offset(self): # Arrange im1 = Image.open("Tests/images/imagedraw_chord_RGB.png") im2 = Image.open("Tests/images/imagedraw_outline_chord_RGB.png") # Act new = ImageChops.subtract(im1, im2, scale=2.5, offset=100) # Assert self.assertEqual(new.getbbox(), (0, 0, 100, 100)) self.assertEqual(new.getpixel((50, 50)), (100, 202, 100)) def test_subtract_clip(self): # Arrange im1 = hopper() im2 = Image.open("Tests/images/imagedraw_chord_RGB.png") # Act new = ImageChops.subtract(im1, im2) # Assert self.assertEqual(new.getpixel((50, 50)), (0, 0, 127)) def test_subtract_modulo(self): # Arrange im1 = Image.open("Tests/images/imagedraw_chord_RGB.png") im2 = Image.open("Tests/images/imagedraw_outline_chord_RGB.png") # Act new = ImageChops.subtract_modulo(im1, im2) # Assert self.assertEqual(new.getbbox(), (25, 50, 76, 76)) self.assertEqual(new.getpixel((50, 50)), GREEN) self.assertEqual(new.getpixel((50, 51)), BLACK) def test_subtract_modulo_no_clip(self): # Arrange im1 = hopper() im2 = Image.open("Tests/images/imagedraw_chord_RGB.png") # Act new = ImageChops.subtract_modulo(im1, im2) # Assert self.assertEqual(new.getpixel((50, 50)), (241, 167, 127)) def test_logical(self): def table(op, a, b): out = [] for x in (a, b): imx = Image.new("1", (1, 1), x) for y in (a, b): imy = Image.new("1", (1, 1), y) out.append(op(imx, imy).getpixel((0, 0))) return tuple(out) self.assertEqual( table(ImageChops.logical_and, 0, 1), (0, 0, 0, 255)) self.assertEqual( table(ImageChops.logical_or, 0, 1), (0, 255, 255, 255)) self.assertEqual( table(ImageChops.logical_xor, 0, 1), (0, 255, 255, 0)) self.assertEqual( table(ImageChops.logical_and, 0, 128), (0, 0, 0, 255)) self.assertEqual( table(ImageChops.logical_or, 0, 128), (0, 255, 255, 255)) self.assertEqual( table(ImageChops.logical_xor, 0, 128), (0, 255, 255, 0)) self.assertEqual( table(ImageChops.logical_and, 0, 255), (0, 0, 0, 255)) self.assertEqual( table(ImageChops.logical_or, 0, 255), (0, 255, 255, 255)) self.assertEqual( table(ImageChops.logical_xor, 0, 255), (0, 255, 255, 0)) if __name__ == '__main__': unittest.main()