mirror of
https://github.com/python-pillow/Pillow.git
synced 2025-01-30 03:04:18 +03:00
7da17ad41e
The previous test configuration made it difficult to run a single test with the pytest CLI. There were two major issues: - The Tests directory was not a package. It now includes a __init__.py file and imports from other tests modules are done with relative imports. - setup.cfg always specified the Tests directory. So even if a specific test were specified as a CLI arg, this configuration would also always include all tests. This configuration has been removed to allow specifying a single test on the command line. Contributors can now run specific tests with a single command such as: $ tox -e py37 -- Tests/test_file_pdf.py::TestFilePdf.test_rgb This makes it easy and faster to iterate on a single test failure and is very familiar to those that have previously used tox and pytest. When running tox or pytest with no arguments, they still discover and runs all tests in the Tests directory.
128 lines
4.0 KiB
Python
128 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)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|