Pillow/Tests/test_imagedraw.py

410 lines
13 KiB
Python
Raw Normal View History

2014-09-05 13:36:24 +04:00
from helper import unittest, PillowTestCase, hopper
from PIL import Image
2014-05-13 00:27:02 +04:00
from PIL import ImageColor
from PIL import ImageDraw
import os.path
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
GRAY = (190, 190, 190)
DEFAULT_MODE = 'RGB'
IMAGES_PATH = os.path.join('Tests', 'images', 'imagedraw')
2014-06-10 13:10:47 +04:00
import sys
2014-05-12 18:32:04 +04:00
# Image size
2014-06-23 23:51:31 +04:00
W, H = 100, 100
2014-05-12 18:32:04 +04:00
# Bounding box points
2014-06-23 23:51:31 +04:00
X0 = int(W / 4)
X1 = int(X0 * 3)
Y0 = int(H / 4)
Y1 = int(X0 * 3)
2014-05-12 18:32:04 +04:00
# Two kinds of bounding box
2014-06-23 23:51:31 +04:00
BBOX1 = [(X0, Y0), (X1, Y1)]
BBOX2 = [X0, Y0, X1, Y1]
2014-05-12 18:32:04 +04:00
# Two kinds of coordinate sequences
2014-06-23 23:51:31 +04:00
POINTS1 = [(10, 10), (20, 40), (30, 30)]
POINTS2 = [10, 10, 20, 40, 30, 30]
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
class TestImageDraw(PillowTestCase):
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
def test_sanity(self):
2014-09-05 13:36:24 +04:00
im = hopper("RGB").copy()
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
draw = ImageDraw.ImageDraw(im)
draw = ImageDraw.Draw(im)
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
draw.ellipse(list(range(4)))
draw.line(list(range(10)))
draw.polygon(list(range(100)))
draw.rectangle(list(range(4)))
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
def test_deprecated(self):
2014-09-05 13:36:24 +04:00
im = hopper().copy()
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
draw = ImageDraw.Draw(im)
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
self.assert_warning(DeprecationWarning, lambda: draw.setink(0))
self.assert_warning(DeprecationWarning, lambda: draw.setfill(0))
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
def helper_arc(self, bbox):
# Arrange
2014-06-23 23:51:31 +04:00
im = Image.new("RGB", (W, H))
2014-06-10 13:10:47 +04:00
draw = ImageDraw.Draw(im)
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
# Act
# FIXME Fill param should be named outline.
draw.arc(bbox, 0, 180)
del draw
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
# Assert
self.assert_image_similar(
2015-04-24 02:26:52 +03:00
im, Image.open("Tests/images/imagedraw_arc.png"), 1)
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
def test_arc1(self):
2014-06-23 23:51:31 +04:00
self.helper_arc(BBOX1)
2014-06-10 13:10:47 +04:00
def test_arc2(self):
2014-06-23 23:51:31 +04:00
self.helper_arc(BBOX2)
2014-06-10 13:10:47 +04:00
def test_bitmap(self):
# Arrange
small = Image.open("Tests/images/pil123rgba.png").resize((50, 50))
2014-06-23 23:51:31 +04:00
im = Image.new("RGB", (W, H))
2014-06-10 13:10:47 +04:00
draw = ImageDraw.Draw(im)
2014-06-10 13:10:47 +04:00
# Act
draw.bitmap((10, 10), small)
del draw
2014-06-10 13:10:47 +04:00
# Assert
self.assert_image_equal(
im, Image.open("Tests/images/imagedraw_bitmap.png"))
2014-06-10 13:10:47 +04:00
def helper_chord(self, bbox):
# Arrange
2014-06-23 23:51:31 +04:00
im = Image.new("RGB", (W, H))
2014-06-10 13:10:47 +04:00
draw = ImageDraw.Draw(im)
2014-06-10 13:10:47 +04:00
# Act
draw.chord(bbox, 0, 180, fill="red", outline="yellow")
del draw
2014-06-10 13:10:47 +04:00
# Assert
self.assert_image_similar(
2015-04-24 02:26:52 +03:00
im, Image.open("Tests/images/imagedraw_chord.png"), 1)
2014-06-10 13:10:47 +04:00
def test_chord1(self):
2014-06-23 23:51:31 +04:00
self.helper_chord(BBOX1)
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
def test_chord2(self):
2014-06-23 23:51:31 +04:00
self.helper_chord(BBOX2)
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
def helper_ellipse(self, bbox):
# Arrange
2014-06-23 23:51:31 +04:00
im = Image.new("RGB", (W, H))
2014-06-10 13:10:47 +04:00
draw = ImageDraw.Draw(im)
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
# Act
draw.ellipse(bbox, fill="green", outline="blue")
del draw
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
# Assert
self.assert_image_similar(
2015-04-24 02:26:52 +03:00
im, Image.open("Tests/images/imagedraw_ellipse.png"), 1)
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
def test_ellipse1(self):
2014-06-23 23:51:31 +04:00
self.helper_ellipse(BBOX1)
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
def test_ellipse2(self):
2014-06-23 23:51:31 +04:00
self.helper_ellipse(BBOX2)
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
def helper_line(self, points):
# Arrange
2014-06-23 23:51:31 +04:00
im = Image.new("RGB", (W, H))
2014-06-10 13:10:47 +04:00
draw = ImageDraw.Draw(im)
2014-05-12 18:32:04 +04:00
2014-06-10 13:10:47 +04:00
# Act
draw.line(points, fill="yellow", width=2)
2014-06-10 13:10:47 +04:00
del draw
2014-06-10 13:10:47 +04:00
# Assert
self.assert_image_equal(
im, Image.open("Tests/images/imagedraw_line.png"))
2014-06-10 13:10:47 +04:00
def test_line1(self):
2014-06-23 23:51:31 +04:00
self.helper_line(POINTS1)
2014-06-10 13:10:47 +04:00
def test_line2(self):
2014-06-23 23:51:31 +04:00
self.helper_line(POINTS2)
2014-06-10 13:10:47 +04:00
def helper_pieslice(self, bbox):
# Arrange
2014-06-23 23:51:31 +04:00
im = Image.new("RGB", (W, H))
2014-06-10 13:10:47 +04:00
draw = ImageDraw.Draw(im)
2014-06-10 13:10:47 +04:00
# Act
draw.pieslice(bbox, -90, 45, fill="white", outline="blue")
del draw
2014-06-10 13:10:47 +04:00
# Assert
self.assert_image_similar(
2015-04-24 02:26:52 +03:00
im, Image.open("Tests/images/imagedraw_pieslice.png"), 1)
2014-06-10 13:10:47 +04:00
def test_pieslice1(self):
2014-06-23 23:51:31 +04:00
self.helper_pieslice(BBOX1)
2014-06-10 13:10:47 +04:00
def test_pieslice2(self):
2014-06-23 23:51:31 +04:00
self.helper_pieslice(BBOX2)
2014-06-10 13:10:47 +04:00
def helper_point(self, points):
# Arrange
2014-06-23 23:51:31 +04:00
im = Image.new("RGB", (W, H))
2014-06-10 13:10:47 +04:00
draw = ImageDraw.Draw(im)
2014-06-10 13:10:47 +04:00
# Act
draw.point(points, fill="yellow")
2014-06-10 13:10:47 +04:00
del draw
2014-06-10 13:10:47 +04:00
# Assert
self.assert_image_equal(
im, Image.open("Tests/images/imagedraw_point.png"))
2014-06-10 13:10:47 +04:00
def test_point1(self):
2014-06-23 23:51:31 +04:00
self.helper_point(POINTS1)
2014-06-10 13:10:47 +04:00
def test_point2(self):
2014-06-23 23:51:31 +04:00
self.helper_point(POINTS2)
2014-06-10 13:10:47 +04:00
def helper_polygon(self, points):
# Arrange
2014-06-23 23:51:31 +04:00
im = Image.new("RGB", (W, H))
2014-06-10 13:10:47 +04:00
draw = ImageDraw.Draw(im)
2014-06-10 13:10:47 +04:00
# Act
draw.polygon(points, fill="red", outline="blue")
2014-06-10 13:10:47 +04:00
del draw
2014-06-10 13:10:47 +04:00
# Assert
self.assert_image_equal(
im, Image.open("Tests/images/imagedraw_polygon.png"))
2014-06-10 13:10:47 +04:00
def test_polygon1(self):
2014-06-23 23:51:31 +04:00
self.helper_polygon(POINTS1)
2014-06-10 13:10:47 +04:00
def test_polygon2(self):
2014-06-23 23:51:31 +04:00
self.helper_polygon(POINTS2)
2014-06-10 13:10:47 +04:00
def helper_rectangle(self, bbox):
# Arrange
2014-06-23 23:51:31 +04:00
im = Image.new("RGB", (W, H))
2014-06-10 13:10:47 +04:00
draw = ImageDraw.Draw(im)
2014-06-10 13:10:47 +04:00
# Act
draw.rectangle(bbox, fill="black", outline="green")
del draw
2014-06-10 13:10:47 +04:00
# Assert
self.assert_image_equal(
im, Image.open("Tests/images/imagedraw_rectangle.png"))
2014-06-10 13:10:47 +04:00
def test_rectangle1(self):
2014-06-23 23:51:31 +04:00
self.helper_rectangle(BBOX1)
2014-06-10 13:10:47 +04:00
def test_rectangle2(self):
2014-06-23 23:51:31 +04:00
self.helper_rectangle(BBOX2)
2014-06-10 13:10:47 +04:00
def test_floodfill(self):
# Arrange
2014-06-23 23:51:31 +04:00
im = Image.new("RGB", (W, H))
2014-06-10 13:10:47 +04:00
draw = ImageDraw.Draw(im)
2014-06-23 23:51:31 +04:00
draw.rectangle(BBOX2, outline="yellow", fill="green")
centre_point = (int(W/2), int(H/2))
2014-05-13 00:27:02 +04:00
2014-06-10 13:10:47 +04:00
# Act
ImageDraw.floodfill(im, centre_point, ImageColor.getrgb("red"))
del draw
2014-05-13 00:27:02 +04:00
2014-06-10 13:10:47 +04:00
# Assert
self.assert_image_equal(
im, Image.open("Tests/images/imagedraw_floodfill.png"))
2014-05-13 00:27:02 +04:00
2014-06-10 13:10:47 +04:00
@unittest.skipIf(hasattr(sys, 'pypy_version_info'),
"Causes fatal RPython error on PyPy")
def test_floodfill_border(self):
# floodfill() is experimental
2014-05-13 00:27:02 +04:00
2014-06-10 13:10:47 +04:00
# Arrange
2014-06-23 23:51:31 +04:00
im = Image.new("RGB", (W, H))
2014-06-10 13:10:47 +04:00
draw = ImageDraw.Draw(im)
2014-06-23 23:51:31 +04:00
draw.rectangle(BBOX2, outline="yellow", fill="green")
centre_point = (int(W/2), int(H/2))
2014-06-10 13:10:47 +04:00
# Act
ImageDraw.floodfill(
im, centre_point, ImageColor.getrgb("red"),
border=ImageColor.getrgb("black"))
del draw
2014-06-10 13:10:47 +04:00
# Assert
self.assert_image_equal(
im, Image.open("Tests/images/imagedraw_floodfill2.png"))
2014-06-25 03:48:38 +04:00
def create_base_image_draw(self, size,
mode=DEFAULT_MODE,
background1=WHITE,
background2=GRAY):
img = Image.new(mode, size, background1)
for x in range(0, size[0]):
for y in range(0, size[1]):
if (x + y) % 2 == 0:
img.putpixel((x, y), background2)
return (img, ImageDraw.Draw(img))
def test_square(self):
expected = Image.open(os.path.join(IMAGES_PATH, 'square.png'))
expected.load()
img, draw = self.create_base_image_draw((10, 10))
draw.polygon([(2, 2), (2, 7), (7, 7), (7, 2)], BLACK)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(img, expected,
'square as normal polygon failed')
2014-06-25 03:48:38 +04:00
img, draw = self.create_base_image_draw((10, 10))
draw.polygon([(7, 7), (7, 2), (2, 2), (2, 7)], BLACK)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(img, expected,
'square as inverted polygon failed')
2014-06-25 03:48:38 +04:00
img, draw = self.create_base_image_draw((10, 10))
draw.rectangle((2, 2, 7, 7), BLACK)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(img, expected,
'square as normal rectangle failed')
2014-06-25 03:48:38 +04:00
img, draw = self.create_base_image_draw((10, 10))
draw.rectangle((7, 7, 2, 2), BLACK)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(
img, expected, 'square as inverted rectangle failed')
2014-06-25 03:48:38 +04:00
def test_triangle_right(self):
expected = Image.open(os.path.join(IMAGES_PATH, 'triangle_right.png'))
expected.load()
img, draw = self.create_base_image_draw((20, 20))
draw.polygon([(3, 5), (17, 5), (10, 12)], BLACK)
self.assert_image_equal(img, expected, 'triangle right failed')
def test_line_horizontal(self):
2014-08-28 18:18:54 +04:00
expected = Image.open(os.path.join(IMAGES_PATH,
'line_horizontal_w2px_normal.png'))
2014-06-25 03:48:38 +04:00
expected.load()
img, draw = self.create_base_image_draw((20, 20))
draw.line((5, 5, 14, 5), BLACK, 2)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(
2015-05-29 07:59:54 +03:00
img, expected, 'line straight horizontal normal 2px wide failed')
2014-08-28 18:18:54 +04:00
expected = Image.open(os.path.join(IMAGES_PATH,
'line_horizontal_w2px_inverted.png'))
2014-06-25 03:48:38 +04:00
expected.load()
img, draw = self.create_base_image_draw((20, 20))
draw.line((14, 5, 5, 5), BLACK, 2)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(
2015-05-29 07:59:54 +03:00
img, expected, 'line straight horizontal inverted 2px wide failed')
2014-08-28 18:18:54 +04:00
expected = Image.open(os.path.join(IMAGES_PATH,
'line_horizontal_w3px.png'))
2014-06-25 03:48:38 +04:00
expected.load()
img, draw = self.create_base_image_draw((20, 20))
draw.line((5, 5, 14, 5), BLACK, 3)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(
2015-05-29 07:59:54 +03:00
img, expected, 'line straight horizontal normal 3px wide failed')
2014-06-25 03:48:38 +04:00
img, draw = self.create_base_image_draw((20, 20))
draw.line((14, 5, 5, 5), BLACK, 3)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(
2015-05-29 07:59:54 +03:00
img, expected, 'line straight horizontal inverted 3px wide failed')
2014-08-28 18:18:54 +04:00
expected = Image.open(os.path.join(IMAGES_PATH,
'line_horizontal_w101px.png'))
2014-06-25 03:48:38 +04:00
expected.load()
img, draw = self.create_base_image_draw((200, 110))
draw.line((5, 55, 195, 55), BLACK, 101)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(
2015-05-29 07:59:54 +03:00
img, expected, 'line straight horizontal 101px wide failed')
2014-07-01 20:07:18 +04:00
def test_line_h_s1_w2(self):
self.skipTest('failing')
2014-08-28 18:18:54 +04:00
expected = Image.open(os.path.join(IMAGES_PATH,
'line_horizontal_slope1px_w2px.png'))
2014-06-25 03:48:38 +04:00
expected.load()
img, draw = self.create_base_image_draw((20, 20))
draw.line((5, 5, 14, 6), BLACK, 2)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(
img, expected, 'line horizontal 1px slope 2px wide failed')
2014-06-25 03:48:38 +04:00
def test_line_vertical(self):
2014-08-28 18:18:54 +04:00
expected = Image.open(os.path.join(IMAGES_PATH,
'line_vertical_w2px_normal.png'))
2014-06-25 03:48:38 +04:00
expected.load()
img, draw = self.create_base_image_draw((20, 20))
draw.line((5, 5, 5, 14), BLACK, 2)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(
2015-05-29 07:59:54 +03:00
img, expected, 'line straight vertical normal 2px wide failed')
2014-08-28 18:18:54 +04:00
expected = Image.open(os.path.join(IMAGES_PATH,
'line_vertical_w2px_inverted.png'))
2014-06-25 03:48:38 +04:00
expected.load()
img, draw = self.create_base_image_draw((20, 20))
draw.line((5, 14, 5, 5), BLACK, 2)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(
2015-05-29 07:59:54 +03:00
img, expected, 'line straight vertical inverted 2px wide failed')
2014-08-28 18:18:54 +04:00
expected = Image.open(os.path.join(IMAGES_PATH,
'line_vertical_w3px.png'))
2014-06-25 03:48:38 +04:00
expected.load()
img, draw = self.create_base_image_draw((20, 20))
draw.line((5, 5, 5, 14), BLACK, 3)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(
2015-05-29 07:59:54 +03:00
img, expected, 'line straight vertical normal 3px wide failed')
2014-06-25 03:48:38 +04:00
img, draw = self.create_base_image_draw((20, 20))
draw.line((5, 14, 5, 5), BLACK, 3)
2014-08-28 18:18:54 +04:00
self.assert_image_equal(
2015-05-29 07:59:54 +03:00
img, expected, 'line straight vertical inverted 3px wide failed')
2014-08-28 18:18:54 +04:00
expected = Image.open(os.path.join(IMAGES_PATH,
'line_vertical_w101px.png'))
2014-06-25 03:48:38 +04:00
expected.load()
img, draw = self.create_base_image_draw((110, 200))
draw.line((55, 5, 55, 195), BLACK, 101)
2014-08-28 15:44:19 +04:00
self.assert_image_equal(img, expected,
2015-05-29 07:59:54 +03:00
'line straight vertical 101px wide failed')
2014-08-28 15:44:19 +04:00
expected = Image.open(os.path.join(IMAGES_PATH,
'line_vertical_slope1px_w2px.png'))
2014-06-25 03:48:38 +04:00
expected.load()
img, draw = self.create_base_image_draw((20, 20))
draw.line((5, 5, 6, 14), BLACK, 2)
2014-08-28 15:44:19 +04:00
self.assert_image_equal(img, expected,
'line vertical 1px slope 2px wide failed')
2014-06-25 03:48:38 +04:00
def test_line_oblique_45(self):
2014-08-28 18:18:54 +04:00
expected = Image.open(os.path.join(IMAGES_PATH,
'line_oblique_45_w3px_a.png'))
2014-06-25 03:48:38 +04:00
expected.load()
img, draw = self.create_base_image_draw((20, 20))
draw.line((5, 5, 14, 14), BLACK, 3)
2014-08-28 15:44:19 +04:00
self.assert_image_equal(img, expected,
2014-08-28 18:18:54 +04:00
'line oblique 45 normal 3px wide A failed')
2014-06-25 03:48:38 +04:00
img, draw = self.create_base_image_draw((20, 20))
draw.line((14, 14, 5, 5), BLACK, 3)
2014-08-28 15:44:19 +04:00
self.assert_image_equal(img, expected,
'line oblique 45 inverted 3px wide A failed')
expected = Image.open(os.path.join(IMAGES_PATH,
'line_oblique_45_w3px_b.png'))
2014-06-25 03:48:38 +04:00
expected.load()
img, draw = self.create_base_image_draw((20, 20))
draw.line((14, 5, 5, 14), BLACK, 3)
2014-08-28 15:44:19 +04:00
self.assert_image_equal(img, expected,
'line oblique 45 normal 3px wide B failed')
2014-06-25 03:48:38 +04:00
img, draw = self.create_base_image_draw((20, 20))
draw.line((5, 14, 14, 5), BLACK, 3)
2014-08-28 15:44:19 +04:00
self.assert_image_equal(img, expected,
'line oblique 45 inverted 3px wide B failed')
2014-06-25 03:48:38 +04:00
2014-06-10 13:10:47 +04:00
if __name__ == '__main__':
unittest.main()
2014-05-12 18:32:04 +04:00
# End of file