Pillow/Tests/test_psdraw.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

73 lines
1.7 KiB
Python
Raw Normal View History

from __future__ import annotations
2024-01-20 14:23:03 +03:00
import os
import sys
from io import BytesIO
from pathlib import Path
2014-12-11 14:06:53 +03:00
import pytest
from PIL import Image, PSDraw
2024-02-12 13:06:17 +03:00
def _create_document(ps: PSDraw.PSDraw) -> None:
2020-01-27 14:46:52 +03:00
title = "hopper"
box = (1 * 72, 2 * 72, 7 * 72, 10 * 72) # in points
2014-12-11 14:06:53 +03:00
2020-01-27 14:46:52 +03:00
ps.begin_document(title)
2014-12-11 14:06:53 +03:00
2020-01-27 14:46:52 +03:00
# draw diagonal lines in a cross
ps.line((1 * 72, 2 * 72), (7 * 72, 10 * 72))
ps.line((7 * 72, 2 * 72), (1 * 72, 10 * 72))
2014-12-11 14:06:53 +03:00
2020-01-27 14:46:52 +03:00
# draw the image (75 dpi)
with Image.open("Tests/images/hopper.ppm") as im:
ps.image(box, im, 75)
ps.rectangle(box)
2014-12-27 23:50:17 +03:00
2020-01-27 14:46:52 +03:00
# draw title
ps.setfont("Courier", 36)
ps.text((3 * 72, 4 * 72), title)
2014-12-11 14:06:53 +03:00
2020-01-27 14:46:52 +03:00
ps.end_document()
2014-12-11 14:06:53 +03:00
def test_draw_postscript(tmp_path: Path) -> None:
2020-01-27 14:46:52 +03:00
# Based on Pillow tutorial, but there is no textsize:
# https://pillow.readthedocs.io/en/latest/handbook/tutorial.html#drawing-postscript
2020-01-27 14:46:52 +03:00
# Arrange
tempfile = str(tmp_path / "temp.ps")
with open(tempfile, "wb") as fp:
# Act
ps = PSDraw.PSDraw(fp)
_create_document(ps)
2020-01-27 14:46:52 +03:00
# Assert
# Check non-zero file was created
assert os.path.isfile(tempfile)
assert os.path.getsize(tempfile) > 0
2014-12-11 14:06:53 +03:00
@pytest.mark.parametrize("buffer", (True, False))
2024-02-12 13:06:17 +03:00
def test_stdout(buffer: bool) -> None:
2020-01-27 14:46:52 +03:00
# Temporarily redirect stdout
old_stdout = sys.stdout
2024-06-22 03:09:11 +03:00
class MyStdOut:
buffer = BytesIO()
2024-06-22 03:09:11 +03:00
mystdout: MyStdOut | BytesIO = MyStdOut() if buffer else BytesIO()
2024-06-23 23:59:00 +03:00
sys.stdout = mystdout # type: ignore[assignment]
2020-01-27 14:46:52 +03:00
ps = PSDraw.PSDraw()
_create_document(ps)
2020-01-27 14:46:52 +03:00
# Reset stdout
sys.stdout = old_stdout
2024-06-22 03:09:11 +03:00
if isinstance(mystdout, MyStdOut):
mystdout = mystdout.buffer
assert mystdout.getvalue() != b""