Simplified code

This commit is contained in:
Andrew Murray 2025-06-10 19:18:34 +10:00
parent 8093cf3704
commit 7ebfb871d0
2 changed files with 4 additions and 10 deletions

View File

@ -6,11 +6,7 @@ import pytest
from PIL import Image, QoiImagePlugin
from .helper import (
assert_image_equal,
assert_image_equal_tofile,
hopper,
)
from .helper import assert_image_equal_tofile, hopper
def test_sanity() -> None:
@ -42,15 +38,13 @@ def test_save(tmp_path: Path) -> None:
im = hopper("RGB")
im.save(f, qoi_colorspace="sRGB")
with Image.open(f) as reloaded:
assert_image_equal(im, reloaded)
assert_image_equal_tofile(im, f)
for image in ["Tests/images/default_font.png", "Tests/images/pil123rgba.png"]:
with Image.open(image) as im:
im.save(f)
with Image.open(f) as reloaded:
assert_image_equal(im, reloaded)
assert_image_equal_tofile(im, f)
im = hopper("P")
with pytest.raises(ValueError):

View File

@ -133,7 +133,7 @@ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
fp.write(o8(channels))
fp.write(o8(colorspace))
ImageFile._save(im, fp, [ImageFile._Tile("qoi", (0, 0) + im.size, 0, im.mode)])
ImageFile._save(im, fp, [ImageFile._Tile("qoi", (0, 0) + im.size)])
class QoiEncoder(ImageFile.PyEncoder):