Pillow/Tests/test_imagesequence.py

76 lines
2.2 KiB
Python
Raw Normal View History

from helper import unittest, PillowTestCase, hopper
from PIL import Image, ImageSequence, TiffImagePlugin
2014-06-10 13:10:47 +04:00
class TestImageSequence(PillowTestCase):
2014-06-10 13:10:47 +04:00
def test_sanity(self):
2015-04-24 11:24:52 +03:00
test_file = self.tempfile("temp.im")
im = hopper("RGB")
2015-04-24 11:24:52 +03:00
im.save(test_file)
2014-06-10 13:10:47 +04:00
seq = ImageSequence.Iterator(im)
2014-06-10 13:10:47 +04:00
index = 0
for frame in seq:
self.assert_image_equal(im, frame)
self.assertEqual(im.tell(), index)
index += 1
2014-06-10 13:10:47 +04:00
self.assertEqual(index, 1)
2015-07-03 08:03:25 +03:00
self.assertRaises(AttributeError, lambda: ImageSequence.Iterator(0))
def test_iterator(self):
im = Image.open('Tests/images/multipage.tiff')
i = ImageSequence.Iterator(im)
for index in range(0, im.n_frames):
self.assertEqual(i[index], next(i))
self.assertRaises(IndexError, lambda: i[index+1])
self.assertRaises(StopIteration, lambda: next(i))
2015-12-02 08:55:50 +03:00
def _test_multipage_tiff(self):
im = Image.open('Tests/images/multipage.tiff')
for index, frame in enumerate(ImageSequence.Iterator(im)):
frame.load()
self.assertEqual(index, im.tell())
frame.convert('RGB')
def test_tiff(self):
2015-12-02 08:55:50 +03:00
self._test_multipage_tiff()
def test_libtiff(self):
codecs = dir(Image.core)
if "libtiff_encoder" not in codecs or "libtiff_decoder" not in codecs:
self.skipTest("tiff support not available")
TiffImagePlugin.READ_LIBTIFF = True
2015-12-02 08:55:50 +03:00
self._test_multipage_tiff()
TiffImagePlugin.READ_LIBTIFF = False
2016-01-27 06:04:05 +03:00
def test_consecutive(self):
im = Image.open('Tests/images/multipage.tiff')
firstFrame = None
for frame in ImageSequence.Iterator(im):
2016-08-04 09:40:12 +03:00
if firstFrame is None:
2016-01-27 06:04:05 +03:00
firstFrame = frame.copy()
for frame in ImageSequence.Iterator(im):
self.assert_image_equal(frame, firstFrame)
break
def test_palette_mmap(self):
# Using mmap in ImageFile can require to reload the palette.
im = Image.open('Tests/images/multipage-mmap.tiff')
color1 = im.getpalette()[0:3]
im.seek(0)
color2 = im.getpalette()[0:3]
self.assertEqual(color1, color2)
2014-06-10 13:10:47 +04:00
if __name__ == '__main__':
unittest.main()