mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-10 19:56:47 +03:00
115 lines
2.8 KiB
Python
115 lines
2.8 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from PIL import Image
|
|
|
|
from .helper import assert_image_equal, hopper
|
|
|
|
try:
|
|
import tkinter as tk
|
|
|
|
from PIL import ImageTk
|
|
|
|
dir(ImageTk)
|
|
HAS_TK = True
|
|
except (OSError, ImportError):
|
|
# Skipped via pytestmark
|
|
HAS_TK = False
|
|
|
|
TK_MODES = ("1", "L", "P", "RGB", "RGBA")
|
|
|
|
|
|
pytestmark = pytest.mark.skipif(not HAS_TK, reason="Tk not installed")
|
|
|
|
|
|
def setup_module() -> None:
|
|
try:
|
|
# setup tk
|
|
tk.Frame()
|
|
# root = tk.Tk()
|
|
except RuntimeError as v:
|
|
pytest.skip(f"RuntimeError: {v}")
|
|
except tk.TclError as v:
|
|
pytest.skip(f"TCL Error: {v}")
|
|
|
|
|
|
def test_kw() -> None:
|
|
TEST_JPG = "Tests/images/hopper.jpg"
|
|
TEST_PNG = "Tests/images/hopper.png"
|
|
with Image.open(TEST_JPG) as im1:
|
|
with Image.open(TEST_PNG) as im2:
|
|
with open(TEST_PNG, "rb") as fp:
|
|
data = fp.read()
|
|
kw = {"file": TEST_JPG, "data": data}
|
|
|
|
# Test "file"
|
|
im = ImageTk._get_image_from_kw(kw)
|
|
assert im is not None
|
|
assert_image_equal(im, im1)
|
|
|
|
# Test "data"
|
|
im = ImageTk._get_image_from_kw(kw)
|
|
assert im is not None
|
|
assert_image_equal(im, im2)
|
|
|
|
# Test no relevant entry
|
|
im = ImageTk._get_image_from_kw(kw)
|
|
assert im is None
|
|
|
|
|
|
@pytest.mark.parametrize("mode", TK_MODES)
|
|
def test_photoimage(mode: str) -> None:
|
|
# test as image:
|
|
im = hopper(mode)
|
|
|
|
# this should not crash
|
|
im_tk = ImageTk.PhotoImage(im)
|
|
|
|
assert im_tk.width() == im.width
|
|
assert im_tk.height() == im.height
|
|
|
|
reloaded = ImageTk.getimage(im_tk)
|
|
assert_image_equal(reloaded, im.convert("RGBA"))
|
|
|
|
with pytest.raises(ValueError):
|
|
ImageTk.PhotoImage()
|
|
with pytest.raises(ValueError):
|
|
ImageTk.PhotoImage(mode)
|
|
|
|
|
|
def test_photoimage_apply_transparency() -> None:
|
|
with Image.open("Tests/images/pil123p.png") as im:
|
|
im_tk = ImageTk.PhotoImage(im)
|
|
reloaded = ImageTk.getimage(im_tk)
|
|
assert_image_equal(reloaded, im.convert("RGBA"))
|
|
|
|
|
|
@pytest.mark.parametrize("mode", TK_MODES)
|
|
def test_photoimage_blank(mode: str) -> None:
|
|
# test a image using mode/size:
|
|
im_tk = ImageTk.PhotoImage(mode, (100, 100))
|
|
|
|
assert im_tk.width() == 100
|
|
assert im_tk.height() == 100
|
|
|
|
im = Image.new(mode, (100, 100))
|
|
reloaded = ImageTk.getimage(im_tk)
|
|
assert_image_equal(reloaded.convert(mode), im)
|
|
|
|
|
|
def test_bitmapimage() -> None:
|
|
im = hopper("1")
|
|
|
|
# this should not crash
|
|
im_tk = ImageTk.BitmapImage(im)
|
|
|
|
assert im_tk.width() == im.width
|
|
assert im_tk.height() == im.height
|
|
|
|
# reloaded = ImageTk.getimage(im_tk)
|
|
# assert_image_equal(reloaded, im)
|
|
|
|
with pytest.raises(ValueError):
|
|
ImageTk.BitmapImage()
|