Pillow/Tests/test_image_fromqimage.py

45 lines
1.2 KiB
Python
Raw Normal View History

from PIL import Image, ImageQt
from .helper import PillowTestCase, hopper
from .test_imageqt import PillowQtTestCase
class TestFromQImage(PillowQtTestCase, PillowTestCase):
files_to_test = [
hopper(),
2019-06-13 18:54:24 +03:00
Image.open("Tests/images/transparent.png"),
Image.open("Tests/images/7x13.png"),
]
2015-12-10 01:35:35 +03:00
def roundtrip(self, expected):
# PIL -> Qt
intermediate = expected.toqimage()
# Qt -> PIL
result = ImageQt.fromqimage(intermediate)
if intermediate.hasAlphaChannel():
2019-06-13 18:54:24 +03:00
self.assert_image_equal(result, expected.convert("RGBA"))
else:
2019-06-13 18:54:24 +03:00
self.assert_image_equal(result, expected.convert("RGB"))
def test_sanity_1(self):
for im in self.files_to_test:
2019-06-13 18:54:24 +03:00
self.roundtrip(im.convert("1"))
def test_sanity_rgb(self):
for im in self.files_to_test:
2019-06-13 18:54:24 +03:00
self.roundtrip(im.convert("RGB"))
def test_sanity_rgba(self):
for im in self.files_to_test:
2019-06-13 18:54:24 +03:00
self.roundtrip(im.convert("RGBA"))
def test_sanity_l(self):
for im in self.files_to_test:
2019-06-13 18:54:24 +03:00
self.roundtrip(im.convert("L"))
def test_sanity_p(self):
for im in self.files_to_test:
2019-06-13 18:54:24 +03:00
self.roundtrip(im.convert("P"))