Pillow/Tests/test_file_icns.py

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)