Test PSD seek

This commit is contained in:
Hugo 2017-02-27 16:09:05 +02:00
parent 5c3b10ee87
commit 921a25d144

View File

@ -46,6 +46,29 @@ class TestImagePsd(PillowTestCase):
except EOFError:
self.assertTrue(im.tell() < n_frames)
def test_seek_tell(self):
im = Image.open(test_file)
layer_number = im.tell()
self.assertEqual(layer_number, 0)
im.seek(0)
layer_number = im.tell()
self.assertEqual(layer_number, 0)
im.seek(1)
layer_number = im.tell()
self.assertEqual(layer_number, 1)
im.seek(2)
layer_number = im.tell()
self.assertEqual(layer_number, 2)
def test_seek_eoferror(self):
im = Image.open(test_file)
self.assertRaises(EOFError, lambda: im.seek(-1))
if __name__ == '__main__':
unittest.main()