mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-14 13:46:57 +03:00
114 lines
3.0 KiB
Python
114 lines
3.0 KiB
Python
from helper import unittest, PillowTestCase, lena
|
|
|
|
from PIL import Image
|
|
|
|
|
|
class TestModeI16(PillowTestCase):
|
|
|
|
original = lena().resize((32,32)).convert('I')
|
|
|
|
def verify(self, im1):
|
|
im2 = self.original.copy()
|
|
self.assertEqual(im1.size, im2.size)
|
|
pix1 = im1.load()
|
|
pix2 = im2.load()
|
|
for y in range(im1.size[1]):
|
|
for x in range(im1.size[0]):
|
|
xy = x, y
|
|
p1 = pix1[xy]
|
|
p2 = pix2[xy]
|
|
self.assertEqual(
|
|
p1, p2,
|
|
("got %r from mode %s at %s, expected %r" %
|
|
(p1, im1.mode, xy, p2)))
|
|
|
|
def test_basic(self):
|
|
# PIL 1.1 has limited support for 16-bit image data. Check that
|
|
# create/copy/transform and save works as expected.
|
|
|
|
def basic(mode):
|
|
|
|
imIn = self.original.convert(mode)
|
|
self.verify(imIn)
|
|
|
|
w, h = imIn.size
|
|
|
|
imOut = imIn.copy()
|
|
self.verify(imOut) # copy
|
|
|
|
imOut = imIn.transform((w, h), Image.EXTENT, (0, 0, w, h))
|
|
self.verify(imOut) # transform
|
|
|
|
filename = self.tempfile("temp.im")
|
|
imIn.save(filename)
|
|
|
|
imOut = Image.open(filename)
|
|
|
|
self.verify(imIn)
|
|
self.verify(imOut)
|
|
|
|
imOut = imIn.crop((0, 0, w, h))
|
|
self.verify(imOut)
|
|
|
|
imOut = Image.new(mode, (w, h), None)
|
|
imOut.paste(imIn.crop((0, 0, w//2, h)), (0, 0))
|
|
imOut.paste(imIn.crop((w//2, 0, w, h)), (w//2, 0))
|
|
|
|
self.verify(imIn)
|
|
self.verify(imOut)
|
|
|
|
imIn = Image.new(mode, (1, 1), 1)
|
|
self.assertEqual(imIn.getpixel((0, 0)), 1)
|
|
|
|
imIn.putpixel((0, 0), 2)
|
|
self.assertEqual(imIn.getpixel((0, 0)), 2)
|
|
|
|
if mode == "L":
|
|
max = 255
|
|
else:
|
|
max = 32767
|
|
|
|
imIn = Image.new(mode, (1, 1), 256)
|
|
self.assertEqual(imIn.getpixel((0, 0)), min(256, max))
|
|
|
|
imIn.putpixel((0, 0), 512)
|
|
self.assertEqual(imIn.getpixel((0, 0)), min(512, max))
|
|
|
|
basic("L")
|
|
|
|
basic("I;16")
|
|
basic("I;16B")
|
|
basic("I;16L")
|
|
|
|
basic("I")
|
|
|
|
def test_tobytes(self):
|
|
|
|
def tobytes(mode):
|
|
return Image.new(mode, (1, 1), 1).tobytes()
|
|
|
|
order = 1 if Image._ENDIAN == '<' else -1
|
|
|
|
self.assertEqual(tobytes("L"), b"\x01")
|
|
self.assertEqual(tobytes("I;16"), b"\x01\x00")
|
|
self.assertEqual(tobytes("I;16B"), b"\x00\x01")
|
|
self.assertEqual(tobytes("I"), b"\x01\x00\x00\x00"[::order])
|
|
|
|
def test_convert(self):
|
|
|
|
im = self.original.copy()
|
|
|
|
self.verify(im.convert("I;16"))
|
|
self.verify(im.convert("I;16").convert("L"))
|
|
self.verify(im.convert("I;16").convert("I"))
|
|
|
|
self.verify(im.convert("I;16B"))
|
|
self.verify(im.convert("I;16B").convert("L"))
|
|
self.verify(im.convert("I;16B").convert("I"))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|
|
# End of file
|