mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-10 19:56:47 +03:00
286 lines
7.8 KiB
Python
286 lines
7.8 KiB
Python
import os
|
|
import re
|
|
from io import BytesIO
|
|
|
|
import pytest
|
|
|
|
from PIL import Image, ImageFile, Jpeg2KImagePlugin, features
|
|
|
|
from .helper import (
|
|
assert_image_equal,
|
|
assert_image_similar,
|
|
assert_image_similar_tofile,
|
|
is_big_endian,
|
|
skip_unless_feature,
|
|
)
|
|
|
|
EXTRA_DIR = "Tests/images/jpeg2000"
|
|
|
|
pytestmark = skip_unless_feature("jpg_2000")
|
|
|
|
test_card = Image.open("Tests/images/test-card.png")
|
|
test_card.load()
|
|
|
|
# OpenJPEG 2.0.0 outputs this debugging message sometimes; we should
|
|
# ignore it---it doesn't represent a test failure.
|
|
# 'Not enough memory to handle tile data'
|
|
|
|
|
|
def roundtrip(im, **options):
|
|
out = BytesIO()
|
|
im.save(out, "JPEG2000", **options)
|
|
test_bytes = out.tell()
|
|
out.seek(0)
|
|
im = Image.open(out)
|
|
im.bytes = test_bytes # for testing only
|
|
im.load()
|
|
return im
|
|
|
|
|
|
def test_sanity():
|
|
# Internal version number
|
|
assert re.search(r"\d+\.\d+\.\d+$", features.version_codec("jpg_2000"))
|
|
|
|
with Image.open("Tests/images/test-card-lossless.jp2") as im:
|
|
px = im.load()
|
|
assert px[0, 0] == (0, 0, 0)
|
|
assert im.mode == "RGB"
|
|
assert im.size == (640, 480)
|
|
assert im.format == "JPEG2000"
|
|
assert im.get_format_mimetype() == "image/jp2"
|
|
|
|
|
|
def test_jpf():
|
|
with Image.open("Tests/images/balloon.jpf") as im:
|
|
assert im.format == "JPEG2000"
|
|
assert im.get_format_mimetype() == "image/jpx"
|
|
|
|
|
|
def test_invalid_file():
|
|
invalid_file = "Tests/images/flower.jpg"
|
|
|
|
with pytest.raises(SyntaxError):
|
|
Jpeg2KImagePlugin.Jpeg2KImageFile(invalid_file)
|
|
|
|
|
|
def test_bytesio():
|
|
with open("Tests/images/test-card-lossless.jp2", "rb") as f:
|
|
data = BytesIO(f.read())
|
|
assert_image_similar_tofile(test_card, data, 1.0e-3)
|
|
|
|
|
|
# These two test pre-written JPEG 2000 files that were not written with
|
|
# PIL (they were made using Adobe Photoshop)
|
|
|
|
|
|
def test_lossless(tmp_path):
|
|
with Image.open("Tests/images/test-card-lossless.jp2") as im:
|
|
im.load()
|
|
outfile = str(tmp_path / "temp_test-card.png")
|
|
im.save(outfile)
|
|
assert_image_similar(im, test_card, 1.0e-3)
|
|
|
|
|
|
def test_lossy_tiled():
|
|
assert_image_similar_tofile(
|
|
test_card, "Tests/images/test-card-lossy-tiled.jp2", 2.0
|
|
)
|
|
|
|
|
|
def test_lossless_rt():
|
|
im = roundtrip(test_card)
|
|
assert_image_equal(im, test_card)
|
|
|
|
|
|
def test_lossy_rt():
|
|
im = roundtrip(test_card, quality_layers=[20])
|
|
assert_image_similar(im, test_card, 2.0)
|
|
|
|
|
|
def test_tiled_rt():
|
|
im = roundtrip(test_card, tile_size=(128, 128))
|
|
assert_image_equal(im, test_card)
|
|
|
|
|
|
def test_tiled_offset_rt():
|
|
im = roundtrip(test_card, tile_size=(128, 128), tile_offset=(0, 0), offset=(32, 32))
|
|
assert_image_equal(im, test_card)
|
|
|
|
|
|
def test_tiled_offset_too_small():
|
|
with pytest.raises(ValueError):
|
|
roundtrip(test_card, tile_size=(128, 128), tile_offset=(0, 0), offset=(128, 32))
|
|
|
|
|
|
def test_irreversible_rt():
|
|
im = roundtrip(test_card, irreversible=True, quality_layers=[20])
|
|
assert_image_similar(im, test_card, 2.0)
|
|
|
|
|
|
def test_prog_qual_rt():
|
|
im = roundtrip(test_card, quality_layers=[60, 40, 20], progression="LRCP")
|
|
assert_image_similar(im, test_card, 2.0)
|
|
|
|
|
|
def test_prog_res_rt():
|
|
im = roundtrip(test_card, num_resolutions=8, progression="RLCP")
|
|
assert_image_equal(im, test_card)
|
|
|
|
|
|
def test_default_num_resolutions():
|
|
for num_resolutions in range(2, 6):
|
|
d = 1 << (num_resolutions - 1)
|
|
im = test_card.resize((d - 1, d - 1))
|
|
with pytest.raises(OSError):
|
|
roundtrip(im, num_resolutions=num_resolutions)
|
|
reloaded = roundtrip(im)
|
|
assert_image_equal(im, reloaded)
|
|
|
|
|
|
def test_reduce():
|
|
with Image.open("Tests/images/test-card-lossless.jp2") as im:
|
|
assert callable(im.reduce)
|
|
|
|
im.reduce = 2
|
|
assert im.reduce == 2
|
|
|
|
im.load()
|
|
assert im.size == (160, 120)
|
|
|
|
im.thumbnail((40, 40))
|
|
assert im.size == (40, 30)
|
|
|
|
|
|
def test_layers_type(tmp_path):
|
|
outfile = str(tmp_path / "temp_layers.jp2")
|
|
for quality_layers in [[100, 50, 10], (100, 50, 10), None]:
|
|
test_card.save(outfile, quality_layers=quality_layers)
|
|
|
|
for quality_layers in ["quality_layers", ("100", "50", "10")]:
|
|
with pytest.raises(ValueError):
|
|
test_card.save(outfile, quality_layers=quality_layers)
|
|
|
|
|
|
def test_layers():
|
|
out = BytesIO()
|
|
test_card.save(out, "JPEG2000", quality_layers=[100, 50, 10], progression="LRCP")
|
|
out.seek(0)
|
|
|
|
with Image.open(out) as im:
|
|
im.layers = 1
|
|
im.load()
|
|
assert_image_similar(im, test_card, 13)
|
|
|
|
out.seek(0)
|
|
with Image.open(out) as im:
|
|
im.layers = 3
|
|
im.load()
|
|
assert_image_similar(im, test_card, 0.4)
|
|
|
|
|
|
def test_rgba():
|
|
# Arrange
|
|
with Image.open("Tests/images/rgb_trns_ycbc.j2k") as j2k:
|
|
with Image.open("Tests/images/rgb_trns_ycbc.jp2") as jp2:
|
|
|
|
# Act
|
|
j2k.load()
|
|
jp2.load()
|
|
|
|
# Assert
|
|
assert j2k.mode == "RGBA"
|
|
assert jp2.mode == "RGBA"
|
|
|
|
|
|
def test_16bit_monochrome_has_correct_mode():
|
|
with Image.open("Tests/images/16bit.cropped.j2k") as j2k:
|
|
j2k.load()
|
|
assert j2k.mode == "I;16"
|
|
|
|
with Image.open("Tests/images/16bit.cropped.jp2") as jp2:
|
|
jp2.load()
|
|
assert jp2.mode == "I;16"
|
|
|
|
|
|
@pytest.mark.xfail(is_big_endian(), reason="Fails on big-endian")
|
|
def test_16bit_monochrome_jp2_like_tiff():
|
|
with Image.open("Tests/images/16bit.cropped.tif") as tiff_16bit:
|
|
assert_image_similar_tofile(tiff_16bit, "Tests/images/16bit.cropped.jp2", 1e-3)
|
|
|
|
|
|
@pytest.mark.xfail(is_big_endian(), reason="Fails on big-endian")
|
|
def test_16bit_monochrome_j2k_like_tiff():
|
|
with Image.open("Tests/images/16bit.cropped.tif") as tiff_16bit:
|
|
assert_image_similar_tofile(tiff_16bit, "Tests/images/16bit.cropped.j2k", 1e-3)
|
|
|
|
|
|
def test_16bit_j2k_roundtrips():
|
|
with Image.open("Tests/images/16bit.cropped.j2k") as j2k:
|
|
im = roundtrip(j2k)
|
|
assert_image_equal(im, j2k)
|
|
|
|
|
|
def test_16bit_jp2_roundtrips():
|
|
with Image.open("Tests/images/16bit.cropped.jp2") as jp2:
|
|
im = roundtrip(jp2)
|
|
assert_image_equal(im, jp2)
|
|
|
|
|
|
def test_unbound_local():
|
|
# prepatch, a malformed jp2 file could cause an UnboundLocalError exception.
|
|
with pytest.raises(OSError):
|
|
with Image.open("Tests/images/unbound_variable.jp2"):
|
|
pass
|
|
|
|
|
|
def test_parser_feed():
|
|
# Arrange
|
|
with open("Tests/images/test-card-lossless.jp2", "rb") as f:
|
|
data = f.read()
|
|
|
|
# Act
|
|
p = ImageFile.Parser()
|
|
p.feed(data)
|
|
|
|
# Assert
|
|
assert p.image.size == (640, 480)
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
not os.path.exists(EXTRA_DIR), reason="Extra image files not installed"
|
|
)
|
|
@pytest.mark.parametrize("name", ("subsampling_1", "subsampling_2", "zoo1", "zoo2"))
|
|
def test_subsampling_decode(name):
|
|
test = f"{EXTRA_DIR}/{name}.jp2"
|
|
reference = f"{EXTRA_DIR}/{name}.ppm"
|
|
|
|
with Image.open(test) as im:
|
|
epsilon = 3 # for YCbCr images
|
|
with Image.open(reference) as im2:
|
|
width, height = im2.size
|
|
if name[-1] == "2":
|
|
# RGB reference images are downscaled
|
|
epsilon = 3e-3
|
|
width, height = width * 2, height * 2
|
|
expected = im2.resize((width, height), Image.NEAREST)
|
|
assert_image_similar(im, expected, epsilon)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"test_file",
|
|
[
|
|
"Tests/images/crash-4fb027452e6988530aa5dabee76eecacb3b79f8a.j2k",
|
|
"Tests/images/crash-7d4c83eb92150fb8f1653a697703ae06ae7c4998.j2k",
|
|
"Tests/images/crash-ccca68ff40171fdae983d924e127a721cab2bd50.j2k",
|
|
"Tests/images/crash-d2c93af851d3ab9a19e34503626368b2ecde9c03.j2k",
|
|
],
|
|
)
|
|
def test_crashes(test_file):
|
|
with open(test_file, "rb") as f:
|
|
with Image.open(f) as im:
|
|
# Valgrind should not complain here
|
|
try:
|
|
im.load()
|
|
except OSError:
|
|
pass
|