mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-25 17:36:18 +03:00
150 lines
3.2 KiB
Python
150 lines
3.2 KiB
Python
import pytest
|
|
from PIL import Image, ImagePalette
|
|
|
|
from .helper import assert_image_equal
|
|
|
|
|
|
def test_sanity():
|
|
|
|
ImagePalette.ImagePalette("RGB", list(range(256)) * 3)
|
|
with pytest.raises(ValueError):
|
|
ImagePalette.ImagePalette("RGB", list(range(256)) * 2)
|
|
|
|
|
|
def test_getcolor():
|
|
|
|
palette = ImagePalette.ImagePalette()
|
|
|
|
test_map = {}
|
|
for i in range(256):
|
|
test_map[palette.getcolor((i, i, i))] = i
|
|
|
|
assert len(test_map) == 256
|
|
with pytest.raises(ValueError):
|
|
palette.getcolor((1, 2, 3))
|
|
|
|
# Test unknown color specifier
|
|
with pytest.raises(ValueError):
|
|
palette.getcolor("unknown")
|
|
|
|
|
|
def test_file(tmp_path):
|
|
|
|
palette = ImagePalette.ImagePalette("RGB", list(range(256)) * 3)
|
|
|
|
f = str(tmp_path / "temp.lut")
|
|
|
|
palette.save(f)
|
|
|
|
p = ImagePalette.load(f)
|
|
|
|
# load returns raw palette information
|
|
assert len(p[0]) == 768
|
|
assert p[1] == "RGB"
|
|
|
|
p = ImagePalette.raw(p[1], p[0])
|
|
assert isinstance(p, ImagePalette.ImagePalette)
|
|
assert p.palette == palette.tobytes()
|
|
|
|
|
|
def test_make_linear_lut():
|
|
# Arrange
|
|
black = 0
|
|
white = 255
|
|
|
|
# Act
|
|
lut = ImagePalette.make_linear_lut(black, white)
|
|
|
|
# Assert
|
|
assert isinstance(lut, list)
|
|
assert len(lut) == 256
|
|
# Check values
|
|
for i in range(0, len(lut)):
|
|
assert lut[i] == i
|
|
|
|
|
|
def test_make_linear_lut_not_yet_implemented():
|
|
# Update after FIXME
|
|
# Arrange
|
|
black = 1
|
|
white = 255
|
|
|
|
# Act
|
|
with pytest.raises(NotImplementedError):
|
|
ImagePalette.make_linear_lut(black, white)
|
|
|
|
|
|
def test_make_gamma_lut():
|
|
# Arrange
|
|
exp = 5
|
|
|
|
# Act
|
|
lut = ImagePalette.make_gamma_lut(exp)
|
|
|
|
# Assert
|
|
assert isinstance(lut, list)
|
|
assert len(lut) == 256
|
|
# Check a few values
|
|
assert lut[0] == 0
|
|
assert lut[63] == 0
|
|
assert lut[127] == 8
|
|
assert lut[191] == 60
|
|
assert lut[255] == 255
|
|
|
|
|
|
def test_rawmode_valueerrors(tmp_path):
|
|
# Arrange
|
|
palette = ImagePalette.raw("RGB", list(range(256)) * 3)
|
|
|
|
# Act / Assert
|
|
with pytest.raises(ValueError):
|
|
palette.tobytes()
|
|
with pytest.raises(ValueError):
|
|
palette.getcolor((1, 2, 3))
|
|
f = str(tmp_path / "temp.lut")
|
|
with pytest.raises(ValueError):
|
|
palette.save(f)
|
|
|
|
|
|
def test_getdata():
|
|
# Arrange
|
|
data_in = list(range(256)) * 3
|
|
palette = ImagePalette.ImagePalette("RGB", data_in)
|
|
|
|
# Act
|
|
mode, data_out = palette.getdata()
|
|
|
|
# Assert
|
|
assert mode == "RGB;L"
|
|
|
|
|
|
def test_rawmode_getdata():
|
|
# Arrange
|
|
data_in = list(range(256)) * 3
|
|
palette = ImagePalette.raw("RGB", data_in)
|
|
|
|
# Act
|
|
rawmode, data_out = palette.getdata()
|
|
|
|
# Assert
|
|
assert rawmode == "RGB"
|
|
assert data_in == data_out
|
|
|
|
|
|
def test_2bit_palette(tmp_path):
|
|
# issue #2258, 2 bit palettes are corrupted.
|
|
outfile = str(tmp_path / "temp.png")
|
|
|
|
rgb = b"\x00" * 2 + b"\x01" * 2 + b"\x02" * 2
|
|
img = Image.frombytes("P", (6, 1), rgb)
|
|
img.putpalette(b"\xFF\x00\x00\x00\xFF\x00\x00\x00\xFF") # RGB
|
|
img.save(outfile, format="PNG")
|
|
|
|
with Image.open(outfile) as reloaded:
|
|
assert_image_equal(img, reloaded)
|
|
|
|
|
|
def test_invalid_palette():
|
|
with pytest.raises(OSError):
|
|
ImagePalette.load("Tests/images/hopper.jpg")
|