mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-25 17:36:18 +03:00
61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
from __future__ import annotations
|
|
|
|
import warnings
|
|
|
|
import pytest
|
|
|
|
from PIL import ImageQt
|
|
|
|
from .helper import assert_image_similar, hopper
|
|
|
|
pytestmark = pytest.mark.skipif(
|
|
not ImageQt.qt_is_installed, reason="Qt bindings are not installed"
|
|
)
|
|
|
|
if ImageQt.qt_is_installed:
|
|
from PIL.ImageQt import qRgba
|
|
|
|
|
|
def test_rgb() -> None:
|
|
# from https://doc.qt.io/archives/qt-4.8/qcolor.html
|
|
# typedef QRgb
|
|
# An ARGB quadruplet on the format #AARRGGBB,
|
|
# equivalent to an unsigned int.
|
|
if ImageQt.qt_version == "6":
|
|
from PyQt6.QtGui import qRgb
|
|
elif ImageQt.qt_version == "side6":
|
|
from PySide6.QtGui import qRgb
|
|
|
|
assert qRgb(0, 0, 0) == qRgba(0, 0, 0, 255)
|
|
|
|
def checkrgb(r: int, g: int, b: int) -> None:
|
|
val = ImageQt.rgb(r, g, b)
|
|
val = val % 2**24 # drop the alpha
|
|
assert val >> 16 == r
|
|
assert ((val >> 8) % 2**8) == g
|
|
assert val % 2**8 == b
|
|
|
|
checkrgb(0, 0, 0)
|
|
checkrgb(255, 0, 0)
|
|
checkrgb(0, 255, 0)
|
|
checkrgb(0, 0, 255)
|
|
|
|
|
|
def test_image() -> None:
|
|
modes = ["1", "RGB", "RGBA", "L", "P"]
|
|
qt_format = ImageQt.QImage.Format if ImageQt.qt_version == "6" else ImageQt.QImage
|
|
if hasattr(qt_format, "Format_Grayscale16"): # Qt 5.13+
|
|
modes.append("I;16")
|
|
|
|
for mode in modes:
|
|
im = hopper(mode)
|
|
roundtripped_im = ImageQt.fromqimage(ImageQt.ImageQt(im))
|
|
if mode not in ("RGB", "RGBA"):
|
|
im = im.convert("RGB")
|
|
assert_image_similar(roundtripped_im, im, 1)
|
|
|
|
|
|
def test_closed_file() -> None:
|
|
with warnings.catch_warnings():
|
|
ImageQt.ImageQt("Tests/images/hopper.gif")
|