import io import os import shutil import tempfile import PIL import pytest from PIL import Image, ImageDraw, ImagePalette, UnidentifiedImageError from .helper import ( assert_image_equal, assert_image_similar, assert_not_all_same, hopper, is_win32, ) class TestImage: def test_image_modes_success(self): for mode in [ "1", "P", "PA", "L", "LA", "La", "F", "I", "I;16", "I;16L", "I;16B", "I;16N", "RGB", "RGBX", "RGBA", "RGBa", "CMYK", "YCbCr", "LAB", "HSV", ]: Image.new(mode, (1, 1)) def test_image_modes_fail(self): for mode in [ "", "bad", "very very long", "BGR;15", "BGR;16", "BGR;24", "BGR;32", ]: with pytest.raises(ValueError) as e: Image.new(mode, (1, 1)) assert str(e.value) == "unrecognized image mode" def test_exception_inheritance(self): assert issubclass(UnidentifiedImageError, IOError) def test_sanity(self): im = Image.new("L", (100, 100)) assert repr(im)[:45] == "= 7 with pytest.warns(DeprecationWarning): assert test_module.PILLOW_VERSION < "9.9.0" with pytest.warns(DeprecationWarning): assert test_module.PILLOW_VERSION <= "9.9.0" with pytest.warns(DeprecationWarning): assert test_module.PILLOW_VERSION != "7.0.0" with pytest.warns(DeprecationWarning): assert test_module.PILLOW_VERSION >= "7.0.0" with pytest.warns(DeprecationWarning): assert test_module.PILLOW_VERSION > "7.0.0" def test_overrun(self): for file in [ "fli_overrun.bin", "sgi_overrun.bin", "sgi_overrun_expandrow.bin", "sgi_overrun_expandrow2.bin", "pcx_overrun.bin", "pcx_overrun2.bin", ]: with Image.open(os.path.join("Tests/images", file)) as im: try: im.load() assert False except OSError as e: assert str(e) == "buffer overrun when reading image file" with Image.open("Tests/images/fli_overrun2.bin") as im: try: im.seek(1) assert False except OSError as e: assert str(e) == "buffer overrun when reading image file" class MockEncoder: pass def mock_encode(*args): encoder = MockEncoder() encoder.args = args return encoder class TestRegistry: def test_encode_registry(self): Image.register_encoder("MOCK", mock_encode) assert "MOCK" in Image.ENCODERS enc = Image._getencoder("RGB", "MOCK", ("args",), extra=("extra",)) assert isinstance(enc, MockEncoder) assert enc.args == ("RGB", "args", "extra") def test_encode_registry_fail(self): with pytest.raises(IOError): Image._getencoder("RGB", "DoesNotExist", ("args",), extra=("extra",))