Pillow/Tests/test_file_ico.py

import io

import pytest
from PIL import IcoImagePlugin, Image, ImageDraw

from .helper import PillowTestCase, assert_image_equal, hopper

TEST_ICO_FILE = "Tests/images/hopper.ico"


class TestFileIco(PillowTestCase):
    def test_sanity(self):
        with Image.open(TEST_ICO_FILE) as im:
            im.load()
        assert im.mode == "RGBA"
        assert im.size == (16, 16)
        assert im.format == "ICO"
        assert im.get_format_mimetype() == "image/x-icon"

    def test_invalid_file(self):
        with open("Tests/images/flower.jpg", "rb") as fp:
            with pytest.raises(SyntaxError):
                IcoImagePlugin.IcoImageFile(fp)

    def test_save_to_bytes(self):
        output = io.BytesIO()
        im = hopper()
        im.save(output, "ico", sizes=[(32, 32), (64, 64)])

        # the default image
        output.seek(0)
        with Image.open(output) as reloaded:
            assert reloaded.info["sizes"] == {(32, 32), (64, 64)}

            assert im.mode == reloaded.mode
            assert (64, 64) == reloaded.size
            assert reloaded.format == "ICO"
            assert_image_equal(reloaded, hopper().resize((64, 64), Image.LANCZOS))

        # the other one
        output.seek(0)
        with Image.open(output) as reloaded:
            reloaded.size = (32, 32)

            assert im.mode == reloaded.mode
            assert (32, 32) == reloaded.size
            assert reloaded.format == "ICO"
            assert_image_equal(reloaded, hopper().resize((32, 32), Image.LANCZOS))

    def test_incorrect_size(self):
        with Image.open(TEST_ICO_FILE) as im:
            with pytest.raises(ValueError):
                im.size = (1, 1)

    def test_save_256x256(self):
        """Issue #2264 https://github.com/python-pillow/Pillow/issues/2264"""
        # Arrange
        with Image.open("Tests/images/hopper_256x256.ico") as im:
            outfile = self.tempfile("temp_saved_hopper_256x256.ico")

            # Act
            im.save(outfile)
        with Image.open(outfile) as im_saved:

            # Assert
            assert im_saved.size == (256, 256)

    def test_only_save_relevant_sizes(self):
        """Issue #2266 https://github.com/python-pillow/Pillow/issues/2266
        Should save in 16x16, 24x24, 32x32, 48x48 sizes
        and not in 16x16, 24x24, 32x32, 48x48, 48x48, 48x48, 48x48 sizes
        """
        # Arrange
        with Image.open("Tests/images/python.ico") as im:  # 16x16, 32x32, 48x48
            outfile = self.tempfile("temp_saved_python.ico")
            # Act
            im.save(outfile)

        with Image.open(outfile) as im_saved:
            # Assert
            assert im_saved.info["sizes"] == {(16, 16), (24, 24), (32, 32), (48, 48)}

    def test_unexpected_size(self):
        # This image has been manually hexedited to state that it is 16x32
        # while the image within is still 16x16
        def open():
            with Image.open("Tests/images/hopper_unexpected.ico") as im:
                assert im.size == (16, 16)

        pytest.warns(UserWarning, open)

    def test_draw_reloaded(self):
        with Image.open(TEST_ICO_FILE) as im:
            outfile = self.tempfile("temp_saved_hopper_draw.ico")

            draw = ImageDraw.Draw(im)
            draw.line((0, 0) + im.size, "#f00")
            im.save(outfile)

        with Image.open(outfile) as im:
            im.save("Tests/images/hopper_draw.ico")
            with Image.open("Tests/images/hopper_draw.ico") as reloaded:
                assert_image_equal(im, reloaded)