mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-11 12:17:14 +03:00
d50445ff30
Similar to the recent adoption of Black. isort is a Python utility to sort imports alphabetically and automatically separate into sections. By using isort, contributors can quickly and automatically conform to the projects style without thinking. Just let the tool do it. Uses the configuration recommended by the Black to avoid conflicts of style. Rewrite TestImageQt.test_deprecated to no rely on import order.
104 lines
3.4 KiB
Python
104 lines
3.4 KiB
Python
import io
|
|
|
|
from PIL import IcoImagePlugin, Image, ImageDraw
|
|
|
|
from .helper import PillowTestCase, hopper
|
|
|
|
TEST_ICO_FILE = "Tests/images/hopper.ico"
|
|
|
|
|
|
class TestFileIco(PillowTestCase):
|
|
def test_sanity(self):
|
|
im = Image.open(TEST_ICO_FILE)
|
|
im.load()
|
|
self.assertEqual(im.mode, "RGBA")
|
|
self.assertEqual(im.size, (16, 16))
|
|
self.assertEqual(im.format, "ICO")
|
|
self.assertEqual(im.get_format_mimetype(), "image/x-icon")
|
|
|
|
def test_invalid_file(self):
|
|
with open("Tests/images/flower.jpg", "rb") as fp:
|
|
self.assertRaises(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)
|
|
reloaded = Image.open(output)
|
|
self.assertEqual(reloaded.info["sizes"], {(32, 32), (64, 64)})
|
|
|
|
self.assertEqual(im.mode, reloaded.mode)
|
|
self.assertEqual((64, 64), reloaded.size)
|
|
self.assertEqual(reloaded.format, "ICO")
|
|
self.assert_image_equal(reloaded, hopper().resize((64, 64), Image.LANCZOS))
|
|
|
|
# the other one
|
|
output.seek(0)
|
|
reloaded = Image.open(output)
|
|
reloaded.size = (32, 32)
|
|
|
|
self.assertEqual(im.mode, reloaded.mode)
|
|
self.assertEqual((32, 32), reloaded.size)
|
|
self.assertEqual(reloaded.format, "ICO")
|
|
self.assert_image_equal(reloaded, hopper().resize((32, 32), Image.LANCZOS))
|
|
|
|
def test_incorrect_size(self):
|
|
im = Image.open(TEST_ICO_FILE)
|
|
with self.assertRaises(ValueError):
|
|
im.size = (1, 1)
|
|
|
|
def test_save_256x256(self):
|
|
"""Issue #2264 https://github.com/python-pillow/Pillow/issues/2264"""
|
|
# Arrange
|
|
im = Image.open("Tests/images/hopper_256x256.ico")
|
|
outfile = self.tempfile("temp_saved_hopper_256x256.ico")
|
|
|
|
# Act
|
|
im.save(outfile)
|
|
im_saved = Image.open(outfile)
|
|
|
|
# Assert
|
|
self.assertEqual(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
|
|
im = Image.open("Tests/images/python.ico") # 16x16, 32x32, 48x48
|
|
outfile = self.tempfile("temp_saved_python.ico")
|
|
|
|
# Act
|
|
im.save(outfile)
|
|
im_saved = Image.open(outfile)
|
|
|
|
# Assert
|
|
self.assertEqual(
|
|
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
|
|
im = self.assert_warning(
|
|
UserWarning, Image.open, "Tests/images/hopper_unexpected.ico"
|
|
)
|
|
self.assertEqual(im.size, (16, 16))
|
|
|
|
def test_draw_reloaded(self):
|
|
im = Image.open(TEST_ICO_FILE)
|
|
outfile = self.tempfile("temp_saved_hopper_draw.ico")
|
|
|
|
draw = ImageDraw.Draw(im)
|
|
draw.line((0, 0) + im.size, "#f00")
|
|
im.save(outfile)
|
|
|
|
im = Image.open(outfile)
|
|
im.save("Tests/images/hopper_draw.ico")
|
|
reloaded = Image.open("Tests/images/hopper_draw.ico")
|
|
self.assert_image_equal(im, reloaded)
|