Pillow/Tests/test_imagedraw.py

from tester import *

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

# 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]


def test_sanity():

    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)))

    success()


def test_deprecated():

    im = lena().copy()

    draw = ImageDraw.Draw(im)

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


def helper_arc(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
    assert_image_equal(im, Image.open("Tests/images/imagedraw_arc.png"))


def test_arc1():
    helper_arc(bbox1)


def test_arc2():
    helper_arc(bbox2)


def test_bitmap():
    # 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
    assert_image_equal(im, Image.open("Tests/images/imagedraw_bitmap.png"))


def helper_chord(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
    assert_image_equal(im, Image.open("Tests/images/imagedraw_chord.png"))


def test_chord1():
    helper_chord(bbox1)


def test_chord2():
    helper_chord(bbox2)


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

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

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


def test_ellipse1():
    helper_ellipse(bbox1)


def test_ellipse2():
    helper_ellipse(bbox2)


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

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

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


def test_line1():
    helper_line(points1)


def test_line2():
    helper_line(points2)


def helper_pieslice(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
    assert_image_equal(im, Image.open("Tests/images/imagedraw_pieslice.png"))


def test_pieslice1():
    helper_pieslice(bbox1)


def test_pieslice2():
    helper_pieslice(bbox2)


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

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

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


def test_point1():
    helper_point(points1)


def test_point2():
    helper_point(points2)


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

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

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


def test_polygon1():
    helper_polygon(points1)


def test_polygon2():
    helper_polygon(points2)


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

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

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


def test_rectangle1():
    helper_rectangle(bbox1)


def test_rectangle2():
    helper_rectangle(bbox2)


def test_floodfill():
    # 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
    assert_image_equal(im, Image.open("Tests/images/imagedraw_floodfill.png"))


def test_floodfill_border():
    # 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
    assert_image_equal(im, Image.open("Tests/images/imagedraw_floodfill2.png"))


# End of file