python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2024-12-25 09:26:16 +03:00
Code Issues Packages Projects Releases Wiki Activity

tests for icc profile in png

Browse Source
This commit is contained in:
homm 2016-05-12 20:28:58 +03:00
parent 989e162ff6
commit 1a6c6b1250
2 changed files with 25 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
Tests/images/icc_profile.png Normal file
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 5.7 KiB

40
Tests/test_file_png.py
View File

@ -434,26 +434,36 @@ class TestFilePng(PillowTestCase):
self.assertEqual(im.info["transparency"], 0) self.assertEqual(im.info["transparency"], 0)
def test_save_icc_profile_none(self): def test_save_icc_profile(self):
# check saving files with an ICC profile set to None (omit profile) im = Image.open("Tests/images/icc_profile_none.png")
in_file = "Tests/images/icc_profile_none.png" self.assertEqual(im.info['icc_profile'], None)
im = Image.open(in_file)
with_icc = Image.open("Tests/images/icc_profile.png")
expected_icc = with_icc.info['icc_profile']
im = roundtrip(im, icc_profile=expected_icc)
self.assertEqual(im.info['icc_profile'], expected_icc)
def test_discard_icc_profile(self):
im = Image.open('Tests/images/icc_profile.png')
im = roundtrip(im, icc_profile=None)
self.assertNotIn('icc_profile', im.info)
def test_roundtrip_icc_profile(self):
im = Image.open('Tests/images/icc_profile.png')
expected_icc = im.info['icc_profile']
im = roundtrip(im)
self.assertEqual(im.info['icc_profile'], expected_icc)
def test_roundtrip_no_icc_profile(self):
im = Image.open("Tests/images/icc_profile_none.png")
self.assertEqual(im.info['icc_profile'], None) self.assertEqual(im.info['icc_profile'], None)
im = roundtrip(im) im = roundtrip(im)
self.assertNotIn('icc_profile', im.info) self.assertNotIn('icc_profile', im.info)
def test_roundtrip_icc_profile(self):
# check that we can roundtrip the icc profile
im = hopper('RGB')
jpeg_image = Image.open('Tests/images/flower2.jpg')
expected_icc = jpeg_image.info['icc_profile']
im.info['icc_profile'] = expected_icc
im = roundtrip(im)
self.assertEqual(im.info['icc_profile'], expected_icc)
def test_repr_png(self): def test_repr_png(self):
im = hopper() im = hopper()