mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-11 04:07:21 +03:00
144 lines
3.7 KiB
Python
144 lines
3.7 KiB
Python
from helper import unittest, PillowTestCase
|
|
|
|
from PIL import ImagePalette, Image
|
|
|
|
ImagePalette = ImagePalette.ImagePalette
|
|
|
|
|
|
class TestImagePalette(PillowTestCase):
|
|
|
|
def test_sanity(self):
|
|
|
|
ImagePalette("RGB", list(range(256))*3)
|
|
self.assertRaises(
|
|
ValueError, lambda: ImagePalette("RGB", list(range(256))*2))
|
|
|
|
def test_getcolor(self):
|
|
|
|
palette = ImagePalette()
|
|
|
|
test_map = {}
|
|
for i in range(256):
|
|
test_map[palette.getcolor((i, i, i))] = i
|
|
|
|
self.assertEqual(len(test_map), 256)
|
|
self.assertRaises(ValueError, lambda: palette.getcolor((1, 2, 3)))
|
|
|
|
def test_file(self):
|
|
|
|
palette = ImagePalette("RGB", list(range(256))*3)
|
|
|
|
f = self.tempfile("temp.lut")
|
|
|
|
palette.save(f)
|
|
|
|
from PIL.ImagePalette import load, raw
|
|
|
|
p = load(f)
|
|
|
|
# load returns raw palette information
|
|
self.assertEqual(len(p[0]), 768)
|
|
self.assertEqual(p[1], "RGB")
|
|
|
|
p = raw(p[1], p[0])
|
|
self.assertIsInstance(p, ImagePalette)
|
|
self.assertEqual(p.palette, palette.tobytes())
|
|
|
|
def test_make_linear_lut(self):
|
|
# Arrange
|
|
from PIL.ImagePalette import make_linear_lut
|
|
black = 0
|
|
white = 255
|
|
|
|
# Act
|
|
lut = make_linear_lut(black, white)
|
|
|
|
# Assert
|
|
self.assertIsInstance(lut, list)
|
|
self.assertEqual(len(lut), 256)
|
|
# Check values
|
|
for i in range(0, len(lut)):
|
|
self.assertEqual(lut[i], i)
|
|
|
|
def test_make_linear_lut_not_yet_implemented(self):
|
|
# Update after FIXME
|
|
# Arrange
|
|
from PIL.ImagePalette import make_linear_lut
|
|
black = 1
|
|
white = 255
|
|
|
|
# Act
|
|
self.assertRaises(
|
|
NotImplementedError,
|
|
lambda: make_linear_lut(black, white))
|
|
|
|
def test_make_gamma_lut(self):
|
|
# Arrange
|
|
from PIL.ImagePalette import make_gamma_lut
|
|
exp = 5
|
|
|
|
# Act
|
|
lut = make_gamma_lut(exp)
|
|
|
|
# Assert
|
|
self.assertIsInstance(lut, list)
|
|
self.assertEqual(len(lut), 256)
|
|
# Check a few values
|
|
self.assertEqual(lut[0], 0)
|
|
self.assertEqual(lut[63], 0)
|
|
self.assertEqual(lut[127], 8)
|
|
self.assertEqual(lut[191], 60)
|
|
self.assertEqual(lut[255], 255)
|
|
|
|
def test_rawmode_valueerrors(self):
|
|
# Arrange
|
|
from PIL.ImagePalette import raw
|
|
palette = raw("RGB", list(range(256))*3)
|
|
|
|
# Act / Assert
|
|
self.assertRaises(ValueError, palette.tobytes)
|
|
self.assertRaises(ValueError, lambda: palette.getcolor((1, 2, 3)))
|
|
f = self.tempfile("temp.lut")
|
|
self.assertRaises(ValueError, lambda: palette.save(f))
|
|
|
|
def test_getdata(self):
|
|
# Arrange
|
|
data_in = list(range(256))*3
|
|
palette = ImagePalette("RGB", data_in)
|
|
|
|
# Act
|
|
mode, data_out = palette.getdata()
|
|
|
|
# Assert
|
|
self.assertEqual(mode, "RGB;L")
|
|
|
|
def test_rawmode_getdata(self):
|
|
# Arrange
|
|
from PIL.ImagePalette import raw
|
|
data_in = list(range(256))*3
|
|
palette = raw("RGB", data_in)
|
|
|
|
# Act
|
|
rawmode, data_out = palette.getdata()
|
|
|
|
# Assert
|
|
self.assertEqual(rawmode, "RGB")
|
|
self.assertEqual(data_in, data_out)
|
|
|
|
def test_2bit_palette(self):
|
|
# issue #2258, 2 bit palettes are corrupted.
|
|
outfile = self.tempfile('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')
|
|
|
|
reloaded = Image.open(outfile)
|
|
|
|
self.assert_image_equal(img, reloaded)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|