Pillow/Tests/test_imageqt.py

from .helper import PillowTestCase, hopper

import warnings

deprecated = False
with warnings.catch_warnings():
    warnings.filterwarnings("error", category=DeprecationWarning)
    try:
        from PIL import ImageQt
    except DeprecationWarning:
        deprecated = True
        warnings.filterwarnings("ignore", category=DeprecationWarning)
        from PIL import ImageQt

if ImageQt.qt_is_installed:
    from PIL.ImageQt import qRgba

    def skip_if_qt_is_not_installed(_):
        pass


else:

    def skip_if_qt_is_not_installed(test_case):
        test_case.skipTest("Qt bindings are not installed")


class PillowQtTestCase(object):
    def setUp(self):
        skip_if_qt_is_not_installed(self)

    def tearDown(self):
        pass


class PillowQPixmapTestCase(PillowQtTestCase):
    def setUp(self):
        PillowQtTestCase.setUp(self)
        try:
            if ImageQt.qt_version == "5":
                from PyQt5.QtGui import QGuiApplication
            elif ImageQt.qt_version == "4":
                from PyQt4.QtGui import QGuiApplication
            elif ImageQt.qt_version == "side":
                from PySide.QtGui import QGuiApplication
            elif ImageQt.qt_version == "side2":
                from PySide2.QtGui import QGuiApplication
        except ImportError:
            self.skipTest("QGuiApplication not installed")

        self.app = QGuiApplication([])

    def tearDown(self):
        PillowQtTestCase.tearDown(self)
        self.app.quit()


class TestImageQt(PillowQtTestCase, PillowTestCase):
    def test_rgb(self):
        # 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 == "5":
            from PyQt5.QtGui import qRgb
        elif ImageQt.qt_version == "4":
            from PyQt4.QtGui import qRgb
        elif ImageQt.qt_version == "side":
            from PySide.QtGui import qRgb
        elif ImageQt.qt_version == "side2":
            from PySide2.QtGui import qRgb

        self.assertEqual(qRgb(0, 0, 0), qRgba(0, 0, 0, 255))

        def checkrgb(r, g, b):
            val = ImageQt.rgb(r, g, b)
            val = val % 2 ** 24  # drop the alpha
            self.assertEqual(val >> 16, r)
            self.assertEqual(((val >> 8) % 2 ** 8), g)
            self.assertEqual(val % 2 ** 8, b)

        checkrgb(0, 0, 0)
        checkrgb(255, 0, 0)
        checkrgb(0, 255, 0)
        checkrgb(0, 0, 255)

    def test_image(self):
        for mode in ("1", "RGB", "RGBA", "L", "P"):
            ImageQt.ImageQt(hopper(mode))

    def test_deprecated(self):
        self.assertEqual(ImageQt.qt_version in ["4", "side"], deprecated)