mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-10 19:56:47 +03:00
146 lines
5.0 KiB
Python
146 lines
5.0 KiB
Python
from helper import unittest, PillowTestCase
|
|
from io import BytesIO
|
|
from PIL import Image
|
|
|
|
|
|
test_files = ["Tests/images/sugarshack.mpo", "Tests/images/frozenpond.mpo"]
|
|
|
|
|
|
class TestFileMpo(PillowTestCase):
|
|
|
|
def setUp(self):
|
|
codecs = dir(Image.core)
|
|
if "jpeg_encoder" not in codecs or "jpeg_decoder" not in codecs:
|
|
self.skipTest("jpeg support not available")
|
|
|
|
def frame_roundtrip(self, im, **options):
|
|
# Note that for now, there is no MPO saving functionality
|
|
out = BytesIO()
|
|
im.save(out, "MPO", **options)
|
|
test_bytes = out.tell()
|
|
out.seek(0)
|
|
im = Image.open(out)
|
|
im.bytes = test_bytes # for testing only
|
|
return im
|
|
|
|
def test_sanity(self):
|
|
for test_file in test_files:
|
|
im = Image.open(test_file)
|
|
im.load()
|
|
self.assertEqual(im.mode, "RGB")
|
|
self.assertEqual(im.size, (640, 480))
|
|
self.assertEqual(im.format, "MPO")
|
|
|
|
def test_app(self):
|
|
for test_file in test_files:
|
|
# Test APP/COM reader (@PIL135)
|
|
im = Image.open(test_file)
|
|
self.assertEqual(im.applist[0][0], 'APP1')
|
|
self.assertEqual(im.applist[1][0], 'APP2')
|
|
self.assertEqual(im.applist[1][1][:16],
|
|
b'MPF\x00MM\x00*\x00\x00\x00\x08\x00\x03\xb0\x00')
|
|
self.assertEqual(len(im.applist), 2)
|
|
|
|
def test_exif(self):
|
|
for test_file in test_files:
|
|
im = Image.open(test_file)
|
|
info = im._getexif()
|
|
self.assertEqual(info[272], 'Nintendo 3DS')
|
|
self.assertEqual(info[296], 2)
|
|
self.assertEqual(info[34665], 188)
|
|
|
|
def test_mp(self):
|
|
for test_file in test_files:
|
|
im = Image.open(test_file)
|
|
mpinfo = im._getmp()
|
|
self.assertEqual(mpinfo[45056], b'0100')
|
|
self.assertEqual(mpinfo[45057], 2)
|
|
|
|
def test_mp_attribute(self):
|
|
for test_file in test_files:
|
|
im = Image.open(test_file)
|
|
mpinfo = im._getmp()
|
|
frameNumber = 0
|
|
for mpentry in mpinfo[45058]:
|
|
mpattr = mpentry['Attribute']
|
|
if frameNumber:
|
|
self.assertFalse(mpattr['RepresentativeImageFlag'])
|
|
else:
|
|
self.assertTrue(mpattr['RepresentativeImageFlag'])
|
|
self.assertFalse(mpattr['DependentParentImageFlag'])
|
|
self.assertFalse(mpattr['DependentChildImageFlag'])
|
|
self.assertEqual(mpattr['ImageDataFormat'], 'JPEG')
|
|
self.assertEqual(mpattr['MPType'],
|
|
'Multi-Frame Image: (Disparity)')
|
|
self.assertEqual(mpattr['Reserved'], 0)
|
|
frameNumber += 1
|
|
|
|
def test_seek(self):
|
|
for test_file in test_files:
|
|
im = Image.open(test_file)
|
|
self.assertEqual(im.tell(), 0)
|
|
# prior to first image raises an error, both blatant and borderline
|
|
self.assertRaises(EOFError, im.seek, -1)
|
|
self.assertRaises(EOFError, im.seek, -523)
|
|
# after the final image raises an error,
|
|
# both blatant and borderline
|
|
self.assertRaises(EOFError, im.seek, 2)
|
|
self.assertRaises(EOFError, im.seek, 523)
|
|
# bad calls shouldn't change the frame
|
|
self.assertEqual(im.tell(), 0)
|
|
# this one will work
|
|
im.seek(1)
|
|
self.assertEqual(im.tell(), 1)
|
|
# and this one, too
|
|
im.seek(0)
|
|
self.assertEqual(im.tell(), 0)
|
|
|
|
def test_n_frames(self):
|
|
im = Image.open("Tests/images/sugarshack.mpo")
|
|
self.assertEqual(im.n_frames, 2)
|
|
self.assertTrue(im.is_animated)
|
|
|
|
def test_eoferror(self):
|
|
im = Image.open("Tests/images/sugarshack.mpo")
|
|
|
|
n_frames = im.n_frames
|
|
while True:
|
|
n_frames -= 1
|
|
try:
|
|
im.seek(n_frames)
|
|
break
|
|
except EOFError:
|
|
self.assertTrue(im.tell() < n_frames)
|
|
|
|
def test_image_grab(self):
|
|
for test_file in test_files:
|
|
im = Image.open(test_file)
|
|
self.assertEqual(im.tell(), 0)
|
|
im0 = im.tobytes()
|
|
im.seek(1)
|
|
self.assertEqual(im.tell(), 1)
|
|
im1 = im.tobytes()
|
|
im.seek(0)
|
|
self.assertEqual(im.tell(), 0)
|
|
im02 = im.tobytes()
|
|
self.assertEqual(im0, im02)
|
|
self.assertNotEqual(im0, im1)
|
|
|
|
def test_save(self):
|
|
# Note that only individual frames can be saved at present
|
|
for test_file in test_files:
|
|
im = Image.open(test_file)
|
|
self.assertEqual(im.tell(), 0)
|
|
jpg0 = self.frame_roundtrip(im)
|
|
self.assert_image_similar(im, jpg0, 30)
|
|
im.seek(1)
|
|
self.assertEqual(im.tell(), 1)
|
|
jpg1 = self.frame_roundtrip(im)
|
|
self.assert_image_similar(im, jpg1, 30)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|
|
|
|
# End of file
|