mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-27 18:36:17 +03:00
75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
import pytest
|
|
from PIL import Image
|
|
|
|
from .helper import assert_image, assert_image_similar, hopper
|
|
|
|
|
|
def test_sanity():
|
|
image = hopper()
|
|
converted = image.quantize()
|
|
assert_image(converted, "P", converted.size)
|
|
assert_image_similar(converted.convert("RGB"), image, 10)
|
|
|
|
image = hopper()
|
|
converted = image.quantize(palette=hopper("P"))
|
|
assert_image(converted, "P", converted.size)
|
|
assert_image_similar(converted.convert("RGB"), image, 60)
|
|
|
|
|
|
def test_libimagequant_quantize():
|
|
image = hopper()
|
|
try:
|
|
converted = image.quantize(100, Image.LIBIMAGEQUANT)
|
|
except ValueError as ex:
|
|
if "dependency" in str(ex).lower():
|
|
pytest.skip("libimagequant support not available")
|
|
else:
|
|
raise
|
|
assert_image(converted, "P", converted.size)
|
|
assert_image_similar(converted.convert("RGB"), image, 15)
|
|
assert len(converted.getcolors()) == 100
|
|
|
|
|
|
def test_octree_quantize():
|
|
image = hopper()
|
|
converted = image.quantize(100, Image.FASTOCTREE)
|
|
assert_image(converted, "P", converted.size)
|
|
assert_image_similar(converted.convert("RGB"), image, 20)
|
|
assert len(converted.getcolors()) == 100
|
|
|
|
|
|
def test_rgba_quantize():
|
|
image = hopper("RGBA")
|
|
with pytest.raises(ValueError):
|
|
image.quantize(method=0)
|
|
|
|
assert image.quantize().convert().mode == "RGBA"
|
|
|
|
|
|
def test_quantize():
|
|
with Image.open("Tests/images/caption_6_33_22.png") as image:
|
|
image = image.convert("RGB")
|
|
converted = image.quantize()
|
|
assert_image(converted, "P", converted.size)
|
|
assert_image_similar(converted.convert("RGB"), image, 1)
|
|
|
|
|
|
def test_quantize_no_dither():
|
|
image = hopper()
|
|
with Image.open("Tests/images/caption_6_33_22.png") as palette:
|
|
palette = palette.convert("P")
|
|
|
|
converted = image.quantize(dither=0, palette=palette)
|
|
assert_image(converted, "P", converted.size)
|
|
|
|
|
|
def test_quantize_dither_diff():
|
|
image = hopper()
|
|
with Image.open("Tests/images/caption_6_33_22.png") as palette:
|
|
palette = palette.convert("P")
|
|
|
|
dither = image.quantize(dither=1, palette=palette)
|
|
nodither = image.quantize(dither=0, palette=palette)
|
|
|
|
assert dither.tobytes() != nodither.tobytes()
|