mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-10 19:56:47 +03:00
80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
from PIL import Image
|
|
|
|
from .helper import fromstring, skip_unless_feature, tostring
|
|
|
|
pytestmark = skip_unless_feature("jpg")
|
|
|
|
|
|
def draft_roundtrip(
|
|
in_mode: str,
|
|
in_size: tuple[int, int],
|
|
req_mode: str | None,
|
|
req_size: tuple[int, int] | None,
|
|
) -> Image.Image:
|
|
im = Image.new(in_mode, in_size)
|
|
data = tostring(im, "JPEG")
|
|
im = fromstring(data)
|
|
mode, box = im.draft(req_mode, req_size)
|
|
scale, _ = im.decoderconfig
|
|
assert box[:2] == (0, 0)
|
|
assert (im.width - scale) < box[2] <= im.width
|
|
assert (im.height - scale) < box[3] <= im.height
|
|
return im
|
|
|
|
|
|
def test_size() -> None:
|
|
for in_size, req_size, out_size in [
|
|
((435, 361), (2048, 2048), (435, 361)), # bigger
|
|
((435, 361), (435, 361), (435, 361)), # same
|
|
((128, 128), (64, 64), (64, 64)),
|
|
((128, 128), (32, 32), (32, 32)),
|
|
((128, 128), (16, 16), (16, 16)),
|
|
# large requested width
|
|
((435, 361), (218, 128), (435, 361)), # almost 2x
|
|
((435, 361), (217, 128), (218, 181)), # more than 2x
|
|
((435, 361), (109, 64), (218, 181)), # almost 4x
|
|
((435, 361), (108, 64), (109, 91)), # more than 4x
|
|
((435, 361), (55, 32), (109, 91)), # almost 8x
|
|
((435, 361), (54, 32), (55, 46)), # more than 8x
|
|
((435, 361), (27, 16), (55, 46)), # more than 16x
|
|
# and vice versa
|
|
((435, 361), (128, 181), (435, 361)), # almost 2x
|
|
((435, 361), (128, 180), (218, 181)), # more than 2x
|
|
((435, 361), (64, 91), (218, 181)), # almost 4x
|
|
((435, 361), (64, 90), (109, 91)), # more than 4x
|
|
((435, 361), (32, 46), (109, 91)), # almost 8x
|
|
((435, 361), (32, 45), (55, 46)), # more than 8x
|
|
((435, 361), (16, 22), (55, 46)), # more than 16x
|
|
]:
|
|
im = draft_roundtrip("L", in_size, None, req_size)
|
|
im.load()
|
|
assert im.size == out_size
|
|
|
|
|
|
def test_mode() -> None:
|
|
for in_mode, req_mode, out_mode in [
|
|
("RGB", "1", "RGB"),
|
|
("RGB", "L", "L"),
|
|
("RGB", "RGB", "RGB"),
|
|
("RGB", "YCbCr", "YCbCr"),
|
|
("L", "1", "L"),
|
|
("L", "L", "L"),
|
|
("L", "RGB", "L"),
|
|
("L", "YCbCr", "L"),
|
|
("CMYK", "1", "CMYK"),
|
|
("CMYK", "L", "CMYK"),
|
|
("CMYK", "RGB", "CMYK"),
|
|
("CMYK", "YCbCr", "CMYK"),
|
|
]:
|
|
im = draft_roundtrip(in_mode, (64, 64), req_mode, None)
|
|
im.load()
|
|
assert im.mode == out_mode
|
|
|
|
|
|
def test_several_drafts() -> None:
|
|
im = draft_roundtrip("L", (128, 128), None, (64, 64))
|
|
im.draft(None, (64, 64))
|
|
im.load()
|