Pillow/Tests/test_image_point.py

49 lines
1.4 KiB
Python
Raw Normal View History

2014-07-07 21:03:50 +04:00
from helper import unittest, PillowTestCase, lena
2014-06-10 13:10:47 +04:00
import sys
2014-03-26 20:12:51 +04:00
2014-06-10 13:10:47 +04:00
class TestImagePoint(PillowTestCase):
2014-06-10 13:10:47 +04:00
def test_sanity(self):
im = lena()
2014-06-10 13:10:47 +04:00
self.assertRaises(ValueError, lambda: im.point(list(range(256))))
im.point(list(range(256))*3)
im.point(lambda x: x)
2013-12-11 04:05:05 +04:00
2014-06-10 13:10:47 +04:00
im = im.convert("I")
self.assertRaises(ValueError, lambda: im.point(list(range(256))))
im.point(lambda x: x*1)
im.point(lambda x: x+1)
im.point(lambda x: x*1+1)
self.assertRaises(TypeError, lambda: im.point(lambda x: x-1))
self.assertRaises(TypeError, lambda: im.point(lambda x: x/1))
2013-12-11 04:05:05 +04:00
2014-06-10 13:10:47 +04:00
def test_16bit_lut(self):
""" Tests for 16 bit -> 8 bit lut for converting I->L images
see https://github.com/python-pillow/Pillow/issues/440
2014-07-24 01:31:49 +04:00
"""
# This takes _forever_ on PyPy. Open Bug,
# see https://github.com/python-pillow/Pillow/issues/484
2014-07-24 03:01:06 +04:00
#self.skipKnownBadTest(msg="Too Slow on pypy", interpreter='pypy')
2013-12-11 04:05:05 +04:00
2014-06-10 13:10:47 +04:00
im = lena("I")
im.point(list(range(256))*256, 'L')
2014-07-29 08:49:11 +04:00
def test_f_lut(self):
""" Tests for floating point lut of 8bit gray image """
im = lena('L')
lut = [0.5 * float(x) for x in range(256)]
out = im.point(lut, 'F')
int_lut = [x//2 for x in range(256)]
self.assert_image_equal(out.convert('L'), im.point(int_lut, 'L'))
2014-06-10 13:10:47 +04:00
if __name__ == '__main__':
unittest.main()
# End of file