Pillow/Tests/test_imagepalette.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

201 lines
5.0 KiB
Python
Raw Normal View History

from __future__ import annotations
2024-01-20 14:23:03 +03:00
from pathlib import Path
import pytest
from PIL import Image, ImagePalette
2021-06-28 15:24:43 +03:00
from .helper import assert_image_equal, assert_image_equal_tofile
def test_sanity() -> None:
2021-06-23 12:16:04 +03:00
palette = ImagePalette.ImagePalette("RGB", list(range(256)) * 3)
assert len(palette.colors) == 256
def test_reload() -> None:
2021-08-24 10:58:02 +03:00
with Image.open("Tests/images/hopper.gif") as im:
original = im.copy()
im.palette.dirty = 1
assert_image_equal(im.convert("RGB"), original.convert("RGB"))
2021-06-28 15:24:43 +03:00
def test_getcolor() -> None:
2020-02-25 12:57:27 +03:00
palette = ImagePalette.ImagePalette()
2021-06-23 12:22:21 +03:00
assert len(palette.palette) == 0
assert len(palette.colors) == 0
2020-02-25 12:57:27 +03:00
test_map = {}
for i in range(256):
test_map[palette.getcolor((i, i, i))] = i
assert len(test_map) == 256
# Colors can be converted between RGB and RGBA
rgba_palette = ImagePalette.ImagePalette("RGBA")
assert rgba_palette.getcolor((0, 0, 0)) == rgba_palette.getcolor((0, 0, 0, 255))
assert palette.getcolor((0, 0, 0)) == palette.getcolor((0, 0, 0, 255))
# An error is raised when the palette is full
2020-02-25 12:57:27 +03:00
with pytest.raises(ValueError):
palette.getcolor((1, 2, 3))
# But not if the image is not using one of the palette entries
palette.getcolor((1, 2, 3), image=Image.new("P", (1, 1)))
2017-09-01 13:36:51 +03:00
2020-02-25 12:57:27 +03:00
# Test unknown color specifier
with pytest.raises(ValueError):
2024-06-22 03:09:11 +03:00
palette.getcolor("unknown") # type: ignore[arg-type]
def test_getcolor_rgba_color_rgb_palette() -> None:
2022-10-10 04:02:10 +03:00
palette = ImagePalette.ImagePalette("RGB")
# Opaque RGBA colors are converted
assert palette.getcolor((0, 0, 0, 255)) == palette.getcolor((0, 0, 0))
with pytest.raises(ValueError):
palette.getcolor((0, 0, 0, 128))
2021-06-30 13:32:48 +03:00
@pytest.mark.parametrize(
"index, palette",
[
# Test when the palette is not full
2021-06-30 13:32:48 +03:00
(0, ImagePalette.ImagePalette()),
# Test when the palette is full
2021-06-30 13:32:48 +03:00
(255, ImagePalette.ImagePalette("RGB", list(range(256)) * 3)),
],
)
2024-02-12 13:06:17 +03:00
def test_getcolor_not_special(index: int, palette: ImagePalette.ImagePalette) -> None:
2021-06-30 13:32:48 +03:00
im = Image.new("P", (1, 1))
# Do not use transparency index as a new color
im.info["transparency"] = index
index1 = palette.getcolor((0, 0, 0), im)
assert index1 != index
# Do not use background index as a new color
im.info["background"] = index1
index2 = palette.getcolor((0, 0, 1), im)
assert index2 not in (index, index1)
def test_file(tmp_path: Path) -> None:
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)
2024-06-24 14:04:33 +03:00
lut = ImagePalette.load(f)
2020-02-25 12:57:27 +03:00
# load returns raw palette information
2024-06-24 14:04:33 +03:00
assert len(lut[0]) == 768
assert lut[1] == "RGB"
2020-02-25 12:57:27 +03:00
2024-06-24 14:04:33 +03:00
p = ImagePalette.raw(lut[1], lut[0])
2020-02-25 12:57:27 +03:00
assert isinstance(p, ImagePalette.ImagePalette)
assert p.palette == palette.tobytes()
def test_make_linear_lut() -> None:
2020-02-25 12:57:27 +03:00
# 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() -> None:
2020-02-25 12:57:27 +03:00
# Update after FIXME
# Arrange
black = 1
white = 255
# Act
with pytest.raises(NotImplementedError):
ImagePalette.make_linear_lut(black, white)
def test_make_gamma_lut() -> None:
2020-02-25 12:57:27 +03:00
# 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: Path) -> None:
2020-02-25 12:57:27 +03:00
# 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() -> None:
2020-02-25 12:57:27 +03:00
# Arrange
data_in = list(range(256)) * 3
palette = ImagePalette.ImagePalette("RGB", data_in)
# Act
mode, data_out = palette.getdata()
# Assert
2021-06-28 15:24:43 +03:00
assert mode == "RGB"
2020-02-25 12:57:27 +03:00
def test_rawmode_getdata() -> None:
2020-02-25 12:57:27 +03:00
# 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: Path) -> None:
2020-02-25 12:57:27 +03:00
# 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")
assert_image_equal_tofile(img, outfile)
2020-02-25 12:57:27 +03:00
def test_invalid_palette() -> None:
with pytest.raises(OSError):
2020-02-25 12:57:27 +03:00
ImagePalette.load("Tests/images/hopper.jpg")