Pillow/Tests/test_imagepalette.py

151 lines
3.2 KiB
Python
Raw Normal View History

import pytest
from PIL import Image, ImagePalette
2020-02-25 12:57:27 +03:00
from .helper import assert_image_equal
2020-02-25 12:57:27 +03:00
def test_sanity():
2020-02-25 12:57:27 +03:00
ImagePalette.ImagePalette("RGB", list(range(256)) * 3)
with pytest.raises(ValueError):
ImagePalette.ImagePalette("RGB", list(range(256)) * 2)
2020-02-25 12:57:27 +03:00
def test_getcolor():
2020-02-25 12:57:27 +03:00
palette = ImagePalette.ImagePalette()
2020-02-25 12:57:27 +03:00
test_map = {}
for i in range(256):
test_map[palette.getcolor((i, i, i))] = i
2020-02-25 12:57:27 +03:00
assert len(test_map) == 256
with pytest.raises(ValueError):
palette.getcolor((1, 2, 3))
2017-09-01 13:36:51 +03:00
2020-02-25 12:57:27 +03:00
# Test unknown color specifier
with pytest.raises(ValueError):
palette.getcolor("unknown")
2020-02-25 12:57:27 +03:00
def test_file(tmp_path):
2020-02-25 12:57:27 +03:00
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)
2020-02-25 12:57:27 +03:00
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):
2020-02-25 12:57:27 +03:00
ImagePalette.load("Tests/images/hopper.jpg")