Pillow/test/test_imagedraw.py

from helper import unittest, PillowTestCase, lena

from PIL import Image
from PIL import ImageColor
from PIL import ImageDraw

import sys

# Image size
w, h = 100, 100

# Bounding box points
x0 = int(w / 4)
x1 = int(x0 * 3)
y0 = int(h / 4)
y1 = int(x0 * 3)

# Two kinds of bounding box
bbox1 = [(x0, y0), (x1, y1)]
bbox2 = [x0, y0, x1, y1]

# Two kinds of coordinate sequences
points1 = [(10, 10), (20, 40), (30, 30)]
points2 = [10, 10, 20, 40, 30, 30]


class TestImageDraw(PillowTestCase):

    def test_sanity(self):
        im = lena("RGB").copy()

        draw = ImageDraw.ImageDraw(im)
        draw = ImageDraw.Draw(im)

        draw.ellipse(list(range(4)))
        draw.line(list(range(10)))
        draw.polygon(list(range(100)))
        draw.rectangle(list(range(4)))

    def test_deprecated(self):
        im = lena().copy()

        draw = ImageDraw.Draw(im)

        self.assert_warning(DeprecationWarning, lambda: draw.setink(0))
        self.assert_warning(DeprecationWarning, lambda: draw.setfill(0))

    def helper_arc(self, bbox):
        # Arrange
        im = Image.new("RGB", (w, h))
        draw = ImageDraw.Draw(im)

        # Act
        # FIXME Fill param should be named outline.
        draw.arc(bbox, 0, 180)
        del draw

        # Assert
        self.assert_image_equal(
            im, Image.open("Tests/images/imagedraw_arc.png"))

    def test_arc1(self):
        self.helper_arc(bbox1)

    def test_arc2(self):
        self.helper_arc(bbox2)

    def test_bitmap(self):
        # Arrange
        small = Image.open("Tests/images/pil123rgba.png").resize((50, 50))
        im = Image.new("RGB", (w, h))
        draw = ImageDraw.Draw(im)

        # Act
        draw.bitmap((10, 10), small)
        del draw

        # Assert
        self.assert_image_equal(
            im, Image.open("Tests/images/imagedraw_bitmap.png"))

    def helper_chord(self, bbox):
        # Arrange
        im = Image.new("RGB", (w, h))
        draw = ImageDraw.Draw(im)

        # Act
        draw.chord(bbox, 0, 180, fill="red", outline="yellow")
        del draw

        # Assert
        self.assert_image_equal(
            im, Image.open("Tests/images/imagedraw_chord.png"))

    def test_chord1(self):
        self.helper_chord(bbox1)

    def test_chord2(self):
        self.helper_chord(bbox2)

    def helper_ellipse(self, bbox):
        # Arrange
        im = Image.new("RGB", (w, h))
        draw = ImageDraw.Draw(im)

        # Act
        draw.ellipse(bbox, fill="green", outline="blue")
        del draw

        # Assert
        self.assert_image_equal(
            im, Image.open("Tests/images/imagedraw_ellipse.png"))

    def test_ellipse1(self):
        self.helper_ellipse(bbox1)

    def test_ellipse2(self):
        self.helper_ellipse(bbox2)

    def helper_line(self, points):
        # Arrange
        im = Image.new("RGB", (w, h))
        draw = ImageDraw.Draw(im)

        # Act
        draw.line(points1, fill="yellow", width=2)
        del draw

        # Assert
        self.assert_image_equal(
            im, Image.open("Tests/images/imagedraw_line.png"))

    def test_line1(self):
        self.helper_line(points1)

    def test_line2(self):
        self.helper_line(points2)

    def helper_pieslice(self, bbox):
        # Arrange
        im = Image.new("RGB", (w, h))
        draw = ImageDraw.Draw(im)

        # Act
        draw.pieslice(bbox, -90, 45, fill="white", outline="blue")
        del draw

        # Assert
        self.assert_image_equal(
            im, Image.open("Tests/images/imagedraw_pieslice.png"))

    def test_pieslice1(self):
        self.helper_pieslice(bbox1)

    def test_pieslice2(self):
        self.helper_pieslice(bbox2)

    def helper_point(self, points):
        # Arrange
        im = Image.new("RGB", (w, h))
        draw = ImageDraw.Draw(im)

        # Act
        draw.point(points1, fill="yellow")
        del draw

        # Assert
        self.assert_image_equal(
            im, Image.open("Tests/images/imagedraw_point.png"))

    def test_point1(self):
        self.helper_point(points1)

    def test_point2(self):
        self.helper_point(points2)

    def helper_polygon(self, points):
        # Arrange
        im = Image.new("RGB", (w, h))
        draw = ImageDraw.Draw(im)

        # Act
        draw.polygon(points1, fill="red", outline="blue")
        del draw

        # Assert
        self.assert_image_equal(
            im, Image.open("Tests/images/imagedraw_polygon.png"))

    def test_polygon1(self):
        self.helper_polygon(points1)

    def test_polygon2(self):
        self.helper_polygon(points2)

    def helper_rectangle(self, bbox):
        # Arrange
        im = Image.new("RGB", (w, h))
        draw = ImageDraw.Draw(im)

        # Act
        draw.rectangle(bbox, fill="black", outline="green")
        del draw

        # Assert
        self.assert_image_equal(
            im, Image.open("Tests/images/imagedraw_rectangle.png"))

    def test_rectangle1(self):
        self.helper_rectangle(bbox1)

    def test_rectangle2(self):
        self.helper_rectangle(bbox2)

    def test_floodfill(self):
        # Arrange
        im = Image.new("RGB", (w, h))
        draw = ImageDraw.Draw(im)
        draw.rectangle(bbox2, outline="yellow", fill="green")
        centre_point = (int(w/2), int(h/2))

        # Act
        ImageDraw.floodfill(im, centre_point, ImageColor.getrgb("red"))
        del draw

        # Assert
        self.assert_image_equal(
            im, Image.open("Tests/images/imagedraw_floodfill.png"))

    @unittest.skipIf(hasattr(sys, 'pypy_version_info'),
                     "Causes fatal RPython error on PyPy")
    def test_floodfill_border(self):
        # floodfill() is experimental

        # Arrange
        im = Image.new("RGB", (w, h))
        draw = ImageDraw.Draw(im)
        draw.rectangle(bbox2, outline="yellow", fill="green")
        centre_point = (int(w/2), int(h/2))

        # Act
        ImageDraw.floodfill(
            im, centre_point, ImageColor.getrgb("red"),
            border=ImageColor.getrgb("black"))
        del draw

        # Assert
        self.assert_image_equal(
            im, Image.open("Tests/images/imagedraw_floodfill2.png"))


if __name__ == '__main__':
    unittest.main()

# End of file