mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-14 13:46:57 +03:00
113 lines
3.2 KiB
Python
113 lines
3.2 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from PIL import Image, ImageSequence, PsdImagePlugin, TiffImagePlugin
|
|
|
|
from .helper import assert_image_equal, hopper, skip_unless_feature
|
|
|
|
|
|
def test_sanity(tmp_path: Path) -> None:
|
|
test_file = str(tmp_path / "temp.im")
|
|
|
|
im = hopper("RGB")
|
|
im.save(test_file)
|
|
|
|
seq = ImageSequence.Iterator(im)
|
|
|
|
index = 0
|
|
for frame in seq:
|
|
assert_image_equal(im, frame)
|
|
assert im.tell() == index
|
|
index += 1
|
|
|
|
assert index == 1
|
|
|
|
with pytest.raises(AttributeError):
|
|
ImageSequence.Iterator(0) # type: ignore[arg-type]
|
|
|
|
|
|
def test_iterator() -> None:
|
|
with Image.open("Tests/images/multipage.tiff") as im:
|
|
assert isinstance(im, TiffImagePlugin.TiffImageFile)
|
|
|
|
i = ImageSequence.Iterator(im)
|
|
for index in range(0, im.n_frames):
|
|
assert i[index] == next(i)
|
|
with pytest.raises(IndexError):
|
|
i[index + 1]
|
|
with pytest.raises(StopIteration):
|
|
next(i)
|
|
|
|
|
|
def test_iterator_min_frame() -> None:
|
|
with Image.open("Tests/images/hopper.psd") as im:
|
|
assert isinstance(im, PsdImagePlugin.PsdImageFile)
|
|
|
|
i = ImageSequence.Iterator(im)
|
|
for index in range(1, im.n_frames):
|
|
assert i[index] == next(i)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"libtiff", (pytest.param(True, marks=skip_unless_feature("libtiff")), False)
|
|
)
|
|
def test_multipage_tiff(monkeypatch: pytest.MonkeyPatch, libtiff: bool) -> None:
|
|
monkeypatch.setattr(TiffImagePlugin, "READ_LIBTIFF", libtiff)
|
|
with Image.open("Tests/images/multipage.tiff") as im:
|
|
for index, frame in enumerate(ImageSequence.Iterator(im)):
|
|
frame.load()
|
|
assert index == im.tell()
|
|
frame.convert("RGB")
|
|
|
|
|
|
def test_consecutive() -> None:
|
|
with Image.open("Tests/images/multipage.tiff") as im:
|
|
first_frame = None
|
|
for frame in ImageSequence.Iterator(im):
|
|
if first_frame is None:
|
|
first_frame = frame.copy()
|
|
assert first_frame is not None
|
|
for frame in ImageSequence.Iterator(im):
|
|
assert_image_equal(frame, first_frame)
|
|
break
|
|
|
|
|
|
def test_palette_mmap() -> None:
|
|
# Using mmap in ImageFile can require to reload the palette.
|
|
with Image.open("Tests/images/multipage-mmap.tiff") as im:
|
|
palette = im.getpalette()
|
|
assert palette is not None
|
|
color1 = palette[:3]
|
|
im.seek(0)
|
|
|
|
palette = im.getpalette()
|
|
assert palette is not None
|
|
color2 = palette[:3]
|
|
assert color1 == color2
|
|
|
|
|
|
def test_all_frames() -> None:
|
|
# Test a single image
|
|
with Image.open("Tests/images/iss634.gif") as im:
|
|
ims = ImageSequence.all_frames(im)
|
|
|
|
assert len(ims) == 42
|
|
for i, im_frame in enumerate(ims):
|
|
assert im_frame is not im
|
|
|
|
im.seek(i)
|
|
assert_image_equal(im, im_frame)
|
|
|
|
# Test a series of images
|
|
ims = ImageSequence.all_frames([im, hopper(), im])
|
|
assert len(ims) == 85
|
|
|
|
# Test an operation
|
|
ims = ImageSequence.all_frames(im, lambda im_frame: im_frame.rotate(90))
|
|
for i, im_frame in enumerate(ims):
|
|
im.seek(i)
|
|
assert_image_equal(im.rotate(90), im_frame)
|