Pillow/Tests/test_file_webp_lossless.py

44 lines
1.1 KiB
Python
Raw Normal View History

2014-09-05 13:36:24 +04:00
from helper import unittest, PillowTestCase, hopper
2014-01-21 01:05:30 +04:00
from PIL import Image
try:
from PIL import _webp
HAVE_WEBP = True
2015-05-27 02:15:45 +03:00
except ImportError:
HAVE_WEBP = False
2014-01-21 01:05:30 +04:00
2014-06-10 13:10:47 +04:00
class TestFileWebpLossless(PillowTestCase):
2014-01-21 01:05:30 +04:00
2014-06-10 13:10:47 +04:00
def setUp(self):
if not HAVE_WEBP:
2014-06-10 13:10:47 +04:00
self.skipTest('WebP support not installed')
return
2014-01-21 01:05:30 +04:00
2014-06-10 13:10:47 +04:00
if (_webp.WebPDecoderVersion() < 0x0200):
self.skipTest('lossless not included')
2014-01-21 01:05:30 +04:00
2017-10-03 18:05:20 +03:00
# WebPAnimDecoder only returns RGBA or RGBX, never RGB
self.rgb_mode = "RGBX" if _webp.HAVE_WEBPANIM else "RGB"
2014-06-10 13:10:47 +04:00
def test_write_lossless_rgb(self):
temp_file = self.tempfile("temp.webp")
2014-01-21 01:05:30 +04:00
hopper(self.rgb_mode).save(temp_file, lossless=True)
2014-01-21 01:05:30 +04:00
2014-06-10 13:10:47 +04:00
image = Image.open(temp_file)
image.load()
2014-01-21 01:05:30 +04:00
self.assertEqual(image.mode, self.rgb_mode)
2014-06-10 13:10:47 +04:00
self.assertEqual(image.size, (128, 128))
self.assertEqual(image.format, "WEBP")
image.load()
image.getdata()
2014-01-21 01:05:30 +04:00
self.assert_image_equal(image, hopper(self.rgb_mode))
2014-01-21 01:05:30 +04:00
2014-06-10 13:10:47 +04:00
if __name__ == '__main__':
unittest.main()