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()