mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-26 18:06:18 +03:00
e534991409
- Change WebPAnimEncoder/Decoder to use RGBX mode instead of RGB (since internally it is using RGBA always)
138 lines
4.1 KiB
Python
138 lines
4.1 KiB
Python
from helper import unittest, PillowTestCase, hopper
|
|
|
|
from PIL import Image
|
|
|
|
try:
|
|
from PIL import _webp
|
|
except ImportError:
|
|
# Skip in setUp()
|
|
pass
|
|
|
|
|
|
class TestFileWebp(PillowTestCase):
|
|
|
|
def setUp(self):
|
|
try:
|
|
from PIL import _webp
|
|
except ImportError:
|
|
self.skipTest('WebP support not installed')
|
|
|
|
def test_version(self):
|
|
_webp.WebPDecoderVersion()
|
|
_webp.WebPDecoderBuggyAlpha()
|
|
|
|
def test_read_rgb(self):
|
|
"""
|
|
Can we read a RGB mode webp file without error.
|
|
Does it have the bits we expect?
|
|
"""
|
|
|
|
file_path = "Tests/images/hopper.webp"
|
|
image = Image.open(file_path)
|
|
|
|
self.assertEqual(image.mode, "RGBX")
|
|
self.assertEqual(image.size, (128, 128))
|
|
self.assertEqual(image.format, "WEBP")
|
|
image.load()
|
|
image.getdata()
|
|
|
|
# generated with:
|
|
# dwebp -ppm ../../Tests/images/hopper.webp -o hopper_webp_bits.ppm
|
|
target = Image.open('Tests/images/hopper_webp_bits.ppm').convert("RGBX")
|
|
self.assert_image_similar(image, target, 20.0)
|
|
|
|
def test_write_rgb(self):
|
|
"""
|
|
Can we write a RGB mode file to webp without error.
|
|
Does it have the bits we expect?
|
|
"""
|
|
|
|
temp_file = self.tempfile("temp.webp")
|
|
|
|
hopper("RGBX").save(temp_file)
|
|
image = Image.open(temp_file)
|
|
|
|
self.assertEqual(image.mode, "RGBX")
|
|
self.assertEqual(image.size, (128, 128))
|
|
self.assertEqual(image.format, "WEBP")
|
|
image.load()
|
|
image.getdata()
|
|
|
|
# If we're using the exact same version of WebP, this test should pass.
|
|
# but it doesn't if the WebP is generated on Ubuntu and tested on
|
|
# Fedora.
|
|
|
|
# generated with: dwebp -ppm temp.webp -o hopper_webp_write.ppm
|
|
# target = Image.open('Tests/images/hopper_webp_write.ppm')
|
|
# self.assert_image_equal(image, target)
|
|
|
|
# This test asserts that the images are similar. If the average pixel
|
|
# difference between the two images is less than the epsilon value,
|
|
# then we're going to accept that it's a reasonable lossy version of
|
|
# the image. The old lena images for WebP are showing ~16 on
|
|
# Ubuntu, the jpegs are showing ~18.
|
|
target = hopper("RGBX")
|
|
self.assert_image_similar(image, target, 12.0)
|
|
|
|
def test_write_unsupported_mode_L(self):
|
|
"""
|
|
Saving a black-and-white file to webp format should work, and be
|
|
similar to the original file.
|
|
"""
|
|
|
|
temp_file = self.tempfile("temp.webp")
|
|
hopper("L").save(temp_file)
|
|
image = Image.open(temp_file)
|
|
|
|
self.assertEqual(image.mode, "RGBX")
|
|
self.assertEqual(image.size, (128, 128))
|
|
self.assertEqual(image.format, "WEBP")
|
|
|
|
image.load()
|
|
image.getdata()
|
|
target = hopper("L").convert("RGBX")
|
|
|
|
self.assert_image_similar(image, target, 10.0)
|
|
|
|
def test_write_unsupported_mode_P(self):
|
|
"""
|
|
Saving a palette-based file to webp format should work, and be
|
|
similar to the original file.
|
|
"""
|
|
|
|
temp_file = self.tempfile("temp.webp")
|
|
hopper("P").save(temp_file)
|
|
image = Image.open(temp_file)
|
|
|
|
self.assertEqual(image.mode, "RGBX")
|
|
self.assertEqual(image.size, (128, 128))
|
|
self.assertEqual(image.format, "WEBP")
|
|
|
|
image.load()
|
|
image.getdata()
|
|
target = hopper("P").convert("RGBX")
|
|
|
|
self.assert_image_similar(image, target, 50.0)
|
|
|
|
def test_WebPEncode_with_invalid_args(self):
|
|
"""
|
|
Calling encoder functions with no arguments should result in an error.
|
|
"""
|
|
|
|
if _webp.HAVE_WEBPMUX:
|
|
self.assertRaises(TypeError, _webp.WebPAnimEncoder)
|
|
self.assertRaises(TypeError, _webp.WebPEncode)
|
|
|
|
def test_WebPDecode_with_invalid_args(self):
|
|
"""
|
|
Calling decoder functions with no arguments should result in an error.
|
|
"""
|
|
|
|
if _webp.HAVE_WEBPMUX:
|
|
self.assertRaises(TypeError, _webp.WebPAnimDecoder)
|
|
self.assertRaises(TypeError, _webp.WebPDecode)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|