mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-09-21 11:28:58 +03:00
33dabf986f
The unittest in helper.py has not offered an interesting abstraction
since dbe9f85c7d so import from the more
typical stdlib location.
122 lines
4.1 KiB
Python
122 lines
4.1 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):
|
|
im = Image.open(TEST_FILE)
|
|
|
|
temp_file = self.tempfile("temp.icns")
|
|
im.save(temp_file)
|
|
|
|
reread = Image.open(temp_file)
|
|
|
|
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):
|
|
im = Image.open(TEST_FILE)
|
|
|
|
temp_file = self.tempfile("temp.icns")
|
|
provided_im = Image.new("RGBA", (32, 32), (255, 0, 0, 128))
|
|
im.save(temp_file, append_images=[provided_im])
|
|
|
|
reread = Image.open(temp_file)
|
|
self.assert_image_similar(reread, im, 1)
|
|
|
|
reread = Image.open(temp_file)
|
|
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)
|