Pillow/Tests/test_file_jxl.py
2024-03-02 00:38:52 +00:00

73 lines
1.8 KiB
Python

from __future__ import annotations
import re
import pytest
from PIL import Image, JxlImagePlugin, features
from .helper import (
assert_image_similar_tofile,
skip_unless_feature,
)
try:
from PIL import _jxl
HAVE_JXL = True
except ImportError:
HAVE_JXL = False
# cjxl v0.9.2 41b8cdab
# hopper.jxl: cjxl hopper.png hopper.jxl -q 75 -e 8
class TestUnsupportedJxl:
def test_unsupported(self) -> None:
if HAVE_JXL:
JxlImagePlugin.SUPPORTED = False
file_path = "Tests/images/hopper.jxl"
with pytest.warns(UserWarning):
with pytest.raises(OSError):
with Image.open(file_path):
pass
if HAVE_JXL:
JxlImagePlugin.SUPPORTED = True
@skip_unless_feature("jxl")
class TestFileJxl:
def setup_method(self) -> None:
self.rgb_mode = "RGB"
def test_version(self) -> None:
_jxl.JxlDecoderVersion()
assert re.search(r"\d+\.\d+\.\d+$", features.version_module("jxl"))
def test_read_rgb(self) -> None:
"""
Can we read a RGB mode Jpeg XL file without error?
Does it have the bits we expect?
"""
with Image.open("Tests/images/hopper.jxl") as image:
assert image.mode == self.rgb_mode
assert image.size == (128, 128)
assert image.format == "JPEG XL"
image.load()
image.getdata()
# generated with:
# djxl hopper.jxl hopper_jxl_bits.ppm
assert_image_similar_tofile(image, "Tests/images/hopper_jxl_bits.ppm", 1.0)
def test_JxlDecode_with_invalid_args(self) -> None:
"""
Calling decoder functions with no arguments should result in an error.
"""
with pytest.raises(TypeError):
_jxl.PILJxlDecoder()