mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-14 05:36:48 +03:00
121 lines
4.2 KiB
Python
121 lines
4.2 KiB
Python
import io
|
|
import sys
|
|
import unittest
|
|
|
|
from PIL import IcnsImagePlugin, Image
|
|
|
|
from .helper import PillowTestCase
|
|
|
|
# sample icon file
|
|
TEST_FILE = "Tests/images/pillow.icns"
|
|
|
|
enable_jpeg2k = hasattr(Image.core, "jp2klib_version")
|
|
|
|
|
|
class TestFileIcns(PillowTestCase):
|
|
def test_sanity(self):
|
|
# Loading this icon by default should result in the largest size
|
|
# (512x512@2x) being loaded
|
|
with Image.open(TEST_FILE) as im:
|
|
|
|
# Assert that there is no unclosed file warning
|
|
self.assert_warning(None, im.load)
|
|
|
|
self.assertEqual(im.mode, "RGBA")
|
|
self.assertEqual(im.size, (1024, 1024))
|
|
self.assertEqual(im.format, "ICNS")
|
|
|
|
@unittest.skipIf(sys.platform != "darwin", "requires macOS")
|
|
def test_save(self):
|
|
temp_file = self.tempfile("temp.icns")
|
|
|
|
with Image.open(TEST_FILE) as im:
|
|
im.save(temp_file)
|
|
|
|
with Image.open(temp_file) as reread:
|
|
self.assertEqual(reread.mode, "RGBA")
|
|
self.assertEqual(reread.size, (1024, 1024))
|
|
self.assertEqual(reread.format, "ICNS")
|
|
|
|
@unittest.skipIf(sys.platform != "darwin", "requires macOS")
|
|
def test_save_append_images(self):
|
|
temp_file = self.tempfile("temp.icns")
|
|
provided_im = Image.new("RGBA", (32, 32), (255, 0, 0, 128))
|
|
|
|
with Image.open(TEST_FILE) as im:
|
|
im.save(temp_file, append_images=[provided_im])
|
|
|
|
with Image.open(temp_file) as reread:
|
|
self.assert_image_similar(reread, im, 1)
|
|
|
|
with Image.open(temp_file) as reread:
|
|
reread.size = (16, 16, 2)
|
|
reread.load()
|
|
self.assert_image_equal(reread, provided_im)
|
|
|
|
def test_sizes(self):
|
|
# Check that we can load all of the sizes, and that the final pixel
|
|
# dimensions are as expected
|
|
with Image.open(TEST_FILE) as im:
|
|
for w, h, r in im.info["sizes"]:
|
|
wr = w * r
|
|
hr = h * r
|
|
im.size = (w, h, r)
|
|
im.load()
|
|
self.assertEqual(im.mode, "RGBA")
|
|
self.assertEqual(im.size, (wr, hr))
|
|
|
|
# Check that we cannot load an incorrect size
|
|
with self.assertRaises(ValueError):
|
|
im.size = (1, 1)
|
|
|
|
def test_older_icon(self):
|
|
# This icon was made with Icon Composer rather than iconutil; it still
|
|
# uses PNG rather than JP2, however (since it was made on 10.9).
|
|
with Image.open("Tests/images/pillow2.icns") as im:
|
|
for w, h, r in im.info["sizes"]:
|
|
wr = w * r
|
|
hr = h * r
|
|
with Image.open("Tests/images/pillow2.icns") as im2:
|
|
im2.size = (w, h, r)
|
|
im2.load()
|
|
self.assertEqual(im2.mode, "RGBA")
|
|
self.assertEqual(im2.size, (wr, hr))
|
|
|
|
def test_jp2_icon(self):
|
|
# This icon was made by using Uli Kusterer's oldiconutil to replace
|
|
# the PNG images with JPEG 2000 ones. The advantage of doing this is
|
|
# that OS X 10.5 supports JPEG 2000 but not PNG; some commercial
|
|
# software therefore does just this.
|
|
|
|
# (oldiconutil is here: https://github.com/uliwitness/oldiconutil)
|
|
|
|
if not enable_jpeg2k:
|
|
return
|
|
|
|
with Image.open("Tests/images/pillow3.icns") as im:
|
|
for w, h, r in im.info["sizes"]:
|
|
wr = w * r
|
|
hr = h * r
|
|
with Image.open("Tests/images/pillow3.icns") as im2:
|
|
im2.size = (w, h, r)
|
|
im2.load()
|
|
self.assertEqual(im2.mode, "RGBA")
|
|
self.assertEqual(im2.size, (wr, hr))
|
|
|
|
def test_getimage(self):
|
|
with open(TEST_FILE, "rb") as fp:
|
|
icns_file = IcnsImagePlugin.IcnsFile(fp)
|
|
|
|
im = icns_file.getimage()
|
|
self.assertEqual(im.mode, "RGBA")
|
|
self.assertEqual(im.size, (1024, 1024))
|
|
|
|
im = icns_file.getimage((512, 512))
|
|
self.assertEqual(im.mode, "RGBA")
|
|
self.assertEqual(im.size, (512, 512))
|
|
|
|
def test_not_an_icns_file(self):
|
|
with io.BytesIO(b"invalid\n") as fp:
|
|
self.assertRaises(SyntaxError, IcnsImagePlugin.IcnsFile, fp)
|