Pillow/Tests/test_imagepalette.py
2014-06-10 12:10:47 +03:00

51 lines
1.1 KiB
Python

from helper import unittest, PillowTestCase, tearDownModule
from PIL import ImagePalette
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()
map = {}
for i in range(256):
map[palette.getcolor((i, i, i))] = i
self.assertEqual(len(map), 256)
self.assertRaises(ValueError, lambda: palette.getcolor((1, 2, 3)))
def test_file(self):
palette = ImagePalette()
file = self.tempfile("temp.lut")
palette.save(file)
from PIL.ImagePalette import load, raw
p = load(file)
# 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)
if __name__ == '__main__':
unittest.main()
# End of file