mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-26 01:46:18 +03:00
263 lines
7.6 KiB
Python
263 lines
7.6 KiB
Python
import pytest
|
|
from PIL import Image, ImageMath, ImageMode
|
|
|
|
from .helper import convert_to_comparable
|
|
|
|
codecs = dir(Image.core)
|
|
|
|
|
|
# There are several internal implementations
|
|
remarkable_factors = [
|
|
# special implementations
|
|
1,
|
|
2,
|
|
3,
|
|
4,
|
|
5,
|
|
6,
|
|
# 1xN implementation
|
|
(1, 2),
|
|
(1, 3),
|
|
(1, 4),
|
|
(1, 7),
|
|
# Nx1 implementation
|
|
(2, 1),
|
|
(3, 1),
|
|
(4, 1),
|
|
(7, 1),
|
|
# general implementation with different paths
|
|
(4, 6),
|
|
(5, 6),
|
|
(4, 7),
|
|
(5, 7),
|
|
(19, 17),
|
|
]
|
|
|
|
gradients_image = Image.open("Tests/images/radial_gradients.png")
|
|
gradients_image.load()
|
|
|
|
|
|
def test_args_factor():
|
|
im = Image.new("L", (10, 10))
|
|
|
|
assert (4, 4) == im.reduce(3).size
|
|
assert (4, 10) == im.reduce((3, 1)).size
|
|
assert (10, 4) == im.reduce((1, 3)).size
|
|
|
|
with pytest.raises(ValueError):
|
|
im.reduce(0)
|
|
with pytest.raises(TypeError):
|
|
im.reduce(2.0)
|
|
with pytest.raises(ValueError):
|
|
im.reduce((0, 10))
|
|
|
|
|
|
def test_args_box():
|
|
im = Image.new("L", (10, 10))
|
|
|
|
assert (5, 5) == im.reduce(2, (0, 0, 10, 10)).size
|
|
assert (1, 1) == im.reduce(2, (5, 5, 6, 6)).size
|
|
|
|
with pytest.raises(TypeError):
|
|
im.reduce(2, "stri")
|
|
with pytest.raises(TypeError):
|
|
im.reduce(2, 2)
|
|
with pytest.raises(ValueError):
|
|
im.reduce(2, (0, 0, 11, 10))
|
|
with pytest.raises(ValueError):
|
|
im.reduce(2, (0, 0, 10, 11))
|
|
with pytest.raises(ValueError):
|
|
im.reduce(2, (-1, 0, 10, 10))
|
|
with pytest.raises(ValueError):
|
|
im.reduce(2, (0, -1, 10, 10))
|
|
with pytest.raises(ValueError):
|
|
im.reduce(2, (0, 5, 10, 5))
|
|
with pytest.raises(ValueError):
|
|
im.reduce(2, (5, 0, 5, 10))
|
|
|
|
|
|
def test_unsupported_modes():
|
|
im = Image.new("P", (10, 10))
|
|
with pytest.raises(ValueError):
|
|
im.reduce(3)
|
|
|
|
im = Image.new("1", (10, 10))
|
|
with pytest.raises(ValueError):
|
|
im.reduce(3)
|
|
|
|
im = Image.new("I;16", (10, 10))
|
|
with pytest.raises(ValueError):
|
|
im.reduce(3)
|
|
|
|
|
|
def get_image(mode):
|
|
mode_info = ImageMode.getmode(mode)
|
|
if mode_info.basetype == "L":
|
|
bands = [gradients_image]
|
|
for _ in mode_info.bands[1:]:
|
|
# rotate previous image
|
|
band = bands[-1].transpose(Image.ROTATE_90)
|
|
bands.append(band)
|
|
# Correct alpha channel by transforming completely transparent pixels.
|
|
# Low alpha values also emphasize error after alpha multiplication.
|
|
if mode.endswith("A"):
|
|
bands[-1] = bands[-1].point(lambda x: int(85 + x / 1.5))
|
|
im = Image.merge(mode, bands)
|
|
else:
|
|
assert len(mode_info.bands) == 1
|
|
im = gradients_image.convert(mode)
|
|
# change the height to make a not-square image
|
|
return im.crop((0, 0, im.width, im.height - 5))
|
|
|
|
|
|
def compare_reduce_with_box(im, factor):
|
|
box = (11, 13, 146, 164)
|
|
reduced = im.reduce(factor, box=box)
|
|
reference = im.crop(box).reduce(factor)
|
|
assert reduced == reference
|
|
|
|
|
|
def compare_reduce_with_reference(im, factor, average_diff=0.4, max_diff=1):
|
|
"""Image.reduce() should look very similar to Image.resize(BOX).
|
|
|
|
A reference image is compiled from a large source area
|
|
and possible last column and last row.
|
|
+-----------+
|
|
|..........c|
|
|
|..........c|
|
|
|..........c|
|
|
|rrrrrrrrrrp|
|
|
+-----------+
|
|
"""
|
|
reduced = im.reduce(factor)
|
|
|
|
if not isinstance(factor, (list, tuple)):
|
|
factor = (factor, factor)
|
|
|
|
reference = Image.new(im.mode, reduced.size)
|
|
area_size = (im.size[0] // factor[0], im.size[1] // factor[1])
|
|
area_box = (0, 0, area_size[0] * factor[0], area_size[1] * factor[1])
|
|
area = im.resize(area_size, Image.BOX, area_box)
|
|
reference.paste(area, (0, 0))
|
|
|
|
if area_size[0] < reduced.size[0]:
|
|
assert reduced.size[0] - area_size[0] == 1
|
|
last_column_box = (area_box[2], 0, im.size[0], area_box[3])
|
|
last_column = im.resize((1, area_size[1]), Image.BOX, last_column_box)
|
|
reference.paste(last_column, (area_size[0], 0))
|
|
|
|
if area_size[1] < reduced.size[1]:
|
|
assert reduced.size[1] - area_size[1] == 1
|
|
last_row_box = (0, area_box[3], area_box[2], im.size[1])
|
|
last_row = im.resize((area_size[0], 1), Image.BOX, last_row_box)
|
|
reference.paste(last_row, (0, area_size[1]))
|
|
|
|
if area_size[0] < reduced.size[0] and area_size[1] < reduced.size[1]:
|
|
last_pixel_box = (area_box[2], area_box[3], im.size[0], im.size[1])
|
|
last_pixel = im.resize((1, 1), Image.BOX, last_pixel_box)
|
|
reference.paste(last_pixel, area_size)
|
|
|
|
assert_compare_images(reduced, reference, average_diff, max_diff)
|
|
|
|
|
|
def assert_compare_images(a, b, max_average_diff, max_diff=255):
|
|
assert a.mode == b.mode, "got mode %r, expected %r" % (a.mode, b.mode)
|
|
assert a.size == b.size, "got size %r, expected %r" % (a.size, b.size)
|
|
|
|
a, b = convert_to_comparable(a, b)
|
|
|
|
bands = ImageMode.getmode(a.mode).bands
|
|
for band, ach, bch in zip(bands, a.split(), b.split()):
|
|
ch_diff = ImageMath.eval("convert(abs(a - b), 'L')", a=ach, b=bch)
|
|
ch_hist = ch_diff.histogram()
|
|
|
|
average_diff = sum(i * num for i, num in enumerate(ch_hist)) / (
|
|
a.size[0] * a.size[1]
|
|
)
|
|
msg = (
|
|
"average pixel value difference {:.4f} > expected {:.4f} "
|
|
"for '{}' band".format(average_diff, max_average_diff, band)
|
|
)
|
|
assert max_average_diff >= average_diff, msg
|
|
|
|
last_diff = [i for i, num in enumerate(ch_hist) if num > 0][-1]
|
|
assert (
|
|
max_diff >= last_diff
|
|
), "max pixel value difference {} > expected {} for '{}' band".format(
|
|
last_diff, max_diff, band
|
|
)
|
|
|
|
|
|
def test_mode_L():
|
|
im = get_image("L")
|
|
for factor in remarkable_factors:
|
|
compare_reduce_with_reference(im, factor)
|
|
compare_reduce_with_box(im, factor)
|
|
|
|
|
|
def test_mode_LA():
|
|
im = get_image("LA")
|
|
for factor in remarkable_factors:
|
|
compare_reduce_with_reference(im, factor, 0.8, 5)
|
|
|
|
# With opaque alpha, an error should be way smaller.
|
|
im.putalpha(Image.new("L", im.size, 255))
|
|
for factor in remarkable_factors:
|
|
compare_reduce_with_reference(im, factor)
|
|
compare_reduce_with_box(im, factor)
|
|
|
|
|
|
def test_mode_La():
|
|
im = get_image("La")
|
|
for factor in remarkable_factors:
|
|
compare_reduce_with_reference(im, factor)
|
|
compare_reduce_with_box(im, factor)
|
|
|
|
|
|
def test_mode_RGB():
|
|
im = get_image("RGB")
|
|
for factor in remarkable_factors:
|
|
compare_reduce_with_reference(im, factor)
|
|
compare_reduce_with_box(im, factor)
|
|
|
|
|
|
def test_mode_RGBA():
|
|
im = get_image("RGBA")
|
|
for factor in remarkable_factors:
|
|
compare_reduce_with_reference(im, factor, 0.8, 5)
|
|
|
|
# With opaque alpha, an error should be way smaller.
|
|
im.putalpha(Image.new("L", im.size, 255))
|
|
for factor in remarkable_factors:
|
|
compare_reduce_with_reference(im, factor)
|
|
compare_reduce_with_box(im, factor)
|
|
|
|
|
|
def test_mode_RGBa():
|
|
im = get_image("RGBa")
|
|
for factor in remarkable_factors:
|
|
compare_reduce_with_reference(im, factor)
|
|
compare_reduce_with_box(im, factor)
|
|
|
|
|
|
def test_mode_I():
|
|
im = get_image("I")
|
|
for factor in remarkable_factors:
|
|
compare_reduce_with_reference(im, factor)
|
|
compare_reduce_with_box(im, factor)
|
|
|
|
|
|
def test_mode_F():
|
|
im = get_image("F")
|
|
for factor in remarkable_factors:
|
|
compare_reduce_with_reference(im, factor, 0, 0)
|
|
compare_reduce_with_box(im, factor)
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
"jpeg2k_decoder" not in codecs, reason="JPEG 2000 support not available"
|
|
)
|
|
def test_jpeg2k():
|
|
with Image.open("Tests/images/test-card-lossless.jp2") as im:
|
|
assert im.reduce(2).size == (320, 240)
|