mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-26 18:06:18 +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.
93 lines
2.8 KiB
Python
93 lines
2.8 KiB
Python
from PIL import Image, ImageQt
|
|
|
|
from .helper import PillowTestCase, hopper
|
|
from .test_imageqt import PillowQtTestCase
|
|
|
|
if ImageQt.qt_is_installed:
|
|
from PIL.ImageQt import QImage
|
|
|
|
try:
|
|
from PyQt5 import QtGui
|
|
from PyQt5.QtWidgets import QWidget, QHBoxLayout, QLabel, QApplication
|
|
|
|
QT_VERSION = 5
|
|
except (ImportError, RuntimeError):
|
|
try:
|
|
from PySide2 import QtGui
|
|
from PySide2.QtWidgets import QWidget, QHBoxLayout, QLabel, QApplication
|
|
|
|
QT_VERSION = 5
|
|
except (ImportError, RuntimeError):
|
|
try:
|
|
from PyQt4 import QtGui
|
|
from PyQt4.QtGui import QWidget, QHBoxLayout, QLabel, QApplication
|
|
|
|
QT_VERSION = 4
|
|
except (ImportError, RuntimeError):
|
|
from PySide import QtGui
|
|
from PySide.QtGui import QWidget, QHBoxLayout, QLabel, QApplication
|
|
|
|
QT_VERSION = 4
|
|
|
|
|
|
class TestToQImage(PillowQtTestCase, PillowTestCase):
|
|
def test_sanity(self):
|
|
for mode in ("RGB", "RGBA", "L", "P", "1"):
|
|
src = hopper(mode)
|
|
data = ImageQt.toqimage(src)
|
|
|
|
self.assertIsInstance(data, QImage)
|
|
self.assertFalse(data.isNull())
|
|
|
|
# reload directly from the qimage
|
|
rt = ImageQt.fromqimage(data)
|
|
if mode in ("L", "P", "1"):
|
|
self.assert_image_equal(rt, src.convert("RGB"))
|
|
else:
|
|
self.assert_image_equal(rt, src)
|
|
|
|
if mode == "1":
|
|
# BW appears to not save correctly on QT4 and QT5
|
|
# kicks out errors on console:
|
|
# libpng warning: Invalid color type/bit depth combination
|
|
# in IHDR
|
|
# libpng error: Invalid IHDR data
|
|
continue
|
|
|
|
# Test saving the file
|
|
tempfile = self.tempfile("temp_{}.png".format(mode))
|
|
data.save(tempfile)
|
|
|
|
# Check that it actually worked.
|
|
reloaded = Image.open(tempfile)
|
|
# Gray images appear to come back in palette mode.
|
|
# They're roughly equivalent
|
|
if QT_VERSION == 4 and mode == "L":
|
|
src = src.convert("P")
|
|
self.assert_image_equal(reloaded, src)
|
|
|
|
def test_segfault(self):
|
|
app = QApplication([])
|
|
ex = Example()
|
|
assert app # Silence warning
|
|
assert ex # Silence warning
|
|
|
|
|
|
if ImageQt.qt_is_installed:
|
|
|
|
class Example(QWidget):
|
|
def __init__(self):
|
|
super(Example, self).__init__()
|
|
|
|
img = hopper().resize((1000, 1000))
|
|
|
|
qimage = ImageQt.ImageQt(img)
|
|
|
|
pixmap1 = QtGui.QPixmap.fromImage(qimage)
|
|
|
|
QHBoxLayout(self) # hbox
|
|
|
|
lbl = QLabel(self)
|
|
# Segfault in the problem
|
|
lbl.setPixmap(pixmap1.copy())
|