mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-27 10:26:19 +03:00
4de5477b61
With the introduction and use of pytest, it is simple and easy to
execute specific tests in isolation through documented command line
arguments. Either by specifying the module path or through the `-k
EXPRESSION` argument. There is no longer any need to provide the
boilerplate:
if __name__ == '__main__':
unittest.main()
To every test file. It is simply noise.
The pattern remains in test files that aren't named with `test_*` as
those files are not discovered and executed by pytest by default.
124 lines
4.0 KiB
Python
124 lines
4.0 KiB
Python
from .helper import unittest, PillowTestCase
|
|
|
|
from PIL import Image, IcnsImagePlugin
|
|
|
|
import io
|
|
import sys
|
|
|
|
# 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
|
|
im = Image.open(TEST_FILE)
|
|
|
|
# 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
|
|
im = Image.open(TEST_FILE)
|
|
for w, h, r in im.info['sizes']:
|
|
wr = w * r
|
|
hr = h * r
|
|
im2 = Image.open(TEST_FILE)
|
|
im2.size = (w, h, r)
|
|
im2.load()
|
|
self.assertEqual(im2.mode, 'RGBA')
|
|
self.assertEqual(im2.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).
|
|
im = Image.open('Tests/images/pillow2.icns')
|
|
for w, h, r in im.info['sizes']:
|
|
wr = w * r
|
|
hr = h * r
|
|
im2 = Image.open('Tests/images/pillow2.icns')
|
|
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
|
|
|
|
im = Image.open('Tests/images/pillow3.icns')
|
|
for w, h, r in im.info['sizes']:
|
|
wr = w * r
|
|
hr = h * r
|
|
im2 = Image.open('Tests/images/pillow3.icns')
|
|
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)
|