from helper import unittest, PillowTestCase, hopper from PIL import Image try: from PIL import _webp except: # Skip in setUp() pass class TestFileWebp(PillowTestCase): def setUp(self): try: from PIL import _webp except: self.skipTest('WebP support not installed') def test_version(self): _webp.WebPDecoderVersion() _webp.WebPDecoderBuggyAlpha() def test_read_rgb(self): file_path = "Tests/images/hopper.webp" image = Image.open(file_path) self.assertEqual(image.mode, "RGB") 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') 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("RGB").save(temp_file) image = Image.open(temp_file) image.load() self.assertEqual(image.mode, "RGB") 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("RGB") self.assert_image_similar(image, target, 12) if __name__ == '__main__': unittest.main() # End of file