mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-09-21 03:18:57 +03:00
73 lines
1.7 KiB
Python
73 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
from io import BytesIO
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from PIL import Image, PSDraw
|
|
|
|
|
|
def _create_document(ps: PSDraw.PSDraw) -> None:
|
|
title = "hopper"
|
|
box = (1 * 72, 2 * 72, 7 * 72, 10 * 72) # in points
|
|
|
|
ps.begin_document(title)
|
|
|
|
# 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))
|
|
|
|
# draw the image (75 dpi)
|
|
with Image.open("Tests/images/hopper.ppm") as im:
|
|
ps.image(box, im, 75)
|
|
ps.rectangle(box)
|
|
|
|
# draw title
|
|
ps.setfont("Courier", 36)
|
|
ps.text((3 * 72, 4 * 72), title)
|
|
|
|
ps.end_document()
|
|
|
|
|
|
def test_draw_postscript(tmp_path: Path) -> None:
|
|
# Based on Pillow tutorial, but there is no textsize:
|
|
# https://pillow.readthedocs.io/en/latest/handbook/tutorial.html#drawing-postscript
|
|
|
|
# Arrange
|
|
tempfile = str(tmp_path / "temp.ps")
|
|
with open(tempfile, "wb") as fp:
|
|
# Act
|
|
ps = PSDraw.PSDraw(fp)
|
|
_create_document(ps)
|
|
|
|
# Assert
|
|
# Check non-zero file was created
|
|
assert os.path.isfile(tempfile)
|
|
assert os.path.getsize(tempfile) > 0
|
|
|
|
|
|
@pytest.mark.parametrize("buffer", (True, False))
|
|
def test_stdout(buffer: bool) -> None:
|
|
# Temporarily redirect stdout
|
|
old_stdout = sys.stdout
|
|
|
|
class MyStdOut:
|
|
buffer = BytesIO()
|
|
|
|
mystdout: MyStdOut | BytesIO = MyStdOut() if buffer else BytesIO()
|
|
|
|
sys.stdout = mystdout # type: ignore[assignment]
|
|
|
|
ps = PSDraw.PSDraw()
|
|
_create_document(ps)
|
|
|
|
# Reset stdout
|
|
sys.stdout = old_stdout
|
|
|
|
if isinstance(mystdout, MyStdOut):
|
|
mystdout = mystdout.buffer
|
|
assert mystdout.getvalue() != b""
|