mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-11-11 04:07:21 +03:00
237 lines
6.8 KiB
Python
237 lines
6.8 KiB
Python
from helper import unittest, PillowTestCase, hopper
|
|
|
|
from PIL import Image
|
|
|
|
|
|
class TestImageConvert(PillowTestCase):
|
|
|
|
def test_sanity(self):
|
|
|
|
def convert(im, mode):
|
|
out = im.convert(mode)
|
|
self.assertEqual(out.mode, mode)
|
|
self.assertEqual(out.size, im.size)
|
|
|
|
modes = "1", "L", "I", "F", "RGB", "RGBA", "RGBX", "CMYK", "YCbCr"
|
|
|
|
for mode in modes:
|
|
im = hopper(mode)
|
|
for mode in modes:
|
|
convert(im, mode)
|
|
|
|
# Check 0
|
|
im = Image.new(mode, (0, 0))
|
|
for mode in modes:
|
|
convert(im, mode)
|
|
|
|
def test_default(self):
|
|
|
|
im = hopper("P")
|
|
self.assert_image(im, "P", im.size)
|
|
im = im.convert()
|
|
self.assert_image(im, "RGB", im.size)
|
|
im = im.convert()
|
|
self.assert_image(im, "RGB", im.size)
|
|
|
|
# ref https://github.com/python-pillow/Pillow/issues/274
|
|
|
|
def _test_float_conversion(self, im):
|
|
orig = im.getpixel((5, 5))
|
|
converted = im.convert('F').getpixel((5, 5))
|
|
self.assertEqual(orig, converted)
|
|
|
|
def test_8bit(self):
|
|
im = Image.open('Tests/images/hopper.jpg')
|
|
self._test_float_conversion(im.convert('L'))
|
|
|
|
def test_16bit(self):
|
|
im = Image.open('Tests/images/16bit.cropped.tif')
|
|
self._test_float_conversion(im)
|
|
|
|
def test_16bit_workaround(self):
|
|
im = Image.open('Tests/images/16bit.cropped.tif')
|
|
self._test_float_conversion(im.convert('I'))
|
|
|
|
def test_rgba_p(self):
|
|
im = hopper('RGBA')
|
|
im.putalpha(hopper('L'))
|
|
|
|
converted = im.convert('P')
|
|
comparable = converted.convert('RGBA')
|
|
|
|
self.assert_image_similar(im, comparable, 20)
|
|
|
|
def test_trns_p(self):
|
|
im = hopper('P')
|
|
im.info['transparency'] = 0
|
|
|
|
f = self.tempfile('temp.png')
|
|
|
|
im_l = im.convert('L')
|
|
self.assertEqual(im_l.info['transparency'], 0) # undone
|
|
im_l.save(f)
|
|
|
|
im_rgb = im.convert('RGB')
|
|
self.assertEqual(im_rgb.info['transparency'], (0, 0, 0)) # undone
|
|
im_rgb.save(f)
|
|
|
|
# ref https://github.com/python-pillow/Pillow/issues/664
|
|
|
|
def test_trns_p_rgba(self):
|
|
# Arrange
|
|
im = hopper('P')
|
|
im.info['transparency'] = 128
|
|
|
|
# Act
|
|
im_rgba = im.convert('RGBA')
|
|
|
|
# Assert
|
|
self.assertNotIn('transparency', im_rgba.info)
|
|
# https://github.com/python-pillow/Pillow/issues/2702
|
|
self.assertEqual(im_rgba.palette, None)
|
|
|
|
def test_trns_l(self):
|
|
im = hopper('L')
|
|
im.info['transparency'] = 128
|
|
|
|
f = self.tempfile('temp.png')
|
|
|
|
im_rgb = im.convert('RGB')
|
|
self.assertEqual(im_rgb.info['transparency'],
|
|
(128, 128, 128)) # undone
|
|
im_rgb.save(f)
|
|
|
|
im_p = im.convert('P')
|
|
self.assertIn('transparency', im_p.info)
|
|
im_p.save(f)
|
|
|
|
im_p = self.assert_warning(
|
|
UserWarning,
|
|
im.convert, 'P', palette=Image.ADAPTIVE)
|
|
self.assertNotIn('transparency', im_p.info)
|
|
im_p.save(f)
|
|
|
|
def test_trns_RGB(self):
|
|
im = hopper('RGB')
|
|
im.info['transparency'] = im.getpixel((0, 0))
|
|
|
|
f = self.tempfile('temp.png')
|
|
|
|
im_l = im.convert('L')
|
|
self.assertEqual(im_l.info['transparency'],
|
|
im_l.getpixel((0, 0))) # undone
|
|
im_l.save(f)
|
|
|
|
im_p = im.convert('P')
|
|
self.assertIn('transparency', im_p.info)
|
|
im_p.save(f)
|
|
|
|
im_rgba = im.convert('RGBA')
|
|
self.assertNotIn('transparency', im_rgba.info)
|
|
im_rgba.save(f)
|
|
|
|
im_p = self.assert_warning(
|
|
UserWarning,
|
|
im.convert, 'P', palette=Image.ADAPTIVE)
|
|
self.assertNotIn('transparency', im_p.info)
|
|
im_p.save(f)
|
|
|
|
def test_gif_with_rgba_palette_to_p(self):
|
|
# See https://github.com/python-pillow/Pillow/issues/2433
|
|
im = Image.open('Tests/images/hopper.gif')
|
|
im.info['transparency'] = 255
|
|
im.load()
|
|
self.assertEqual(im.palette.mode, 'RGBA')
|
|
im_p = im.convert('P')
|
|
|
|
# Should not raise ValueError: unrecognized raw mode
|
|
im_p.load()
|
|
|
|
def test_p_la(self):
|
|
im = hopper('RGBA')
|
|
alpha = hopper('L')
|
|
im.putalpha(alpha)
|
|
|
|
comparable = im.convert('P').convert('LA').getchannel('A')
|
|
|
|
self.assert_image_similar(alpha, comparable, 5)
|
|
|
|
def test_matrix_illegal_conversion(self):
|
|
# Arrange
|
|
im = hopper('CMYK')
|
|
matrix = (
|
|
0.412453, 0.357580, 0.180423, 0,
|
|
0.212671, 0.715160, 0.072169, 0,
|
|
0.019334, 0.119193, 0.950227, 0)
|
|
self.assertNotEqual(im.mode, 'RGB')
|
|
|
|
# Act / Assert
|
|
self.assertRaises(ValueError,
|
|
im.convert, mode='CMYK', matrix=matrix)
|
|
|
|
def test_matrix_wrong_mode(self):
|
|
# Arrange
|
|
im = hopper('L')
|
|
matrix = (
|
|
0.412453, 0.357580, 0.180423, 0,
|
|
0.212671, 0.715160, 0.072169, 0,
|
|
0.019334, 0.119193, 0.950227, 0)
|
|
self.assertEqual(im.mode, 'L')
|
|
|
|
# Act / Assert
|
|
self.assertRaises(ValueError,
|
|
im.convert, mode='L', matrix=matrix)
|
|
|
|
def test_matrix_xyz(self):
|
|
|
|
def matrix_convert(mode):
|
|
# Arrange
|
|
im = hopper('RGB')
|
|
im.info['transparency'] = (255, 0, 0)
|
|
matrix = (
|
|
0.412453, 0.357580, 0.180423, 0,
|
|
0.212671, 0.715160, 0.072169, 0,
|
|
0.019334, 0.119193, 0.950227, 0)
|
|
self.assertEqual(im.mode, 'RGB')
|
|
|
|
# Act
|
|
# Convert an RGB image to the CIE XYZ colour space
|
|
converted_im = im.convert(mode=mode, matrix=matrix)
|
|
|
|
# Assert
|
|
self.assertEqual(converted_im.mode, mode)
|
|
self.assertEqual(converted_im.size, im.size)
|
|
target = Image.open('Tests/images/hopper-XYZ.png')
|
|
if converted_im.mode == 'RGB':
|
|
self.assert_image_similar(converted_im, target, 3)
|
|
self.assertEqual(converted_im.info['transparency'],
|
|
(105, 54, 4))
|
|
else:
|
|
self.assert_image_similar(converted_im,
|
|
target.getchannel(0), 1)
|
|
self.assertEqual(converted_im.info['transparency'], 105)
|
|
|
|
matrix_convert('RGB')
|
|
matrix_convert('L')
|
|
|
|
def test_matrix_identity(self):
|
|
# Arrange
|
|
im = hopper('RGB')
|
|
identity_matrix = (
|
|
1, 0, 0, 0,
|
|
0, 1, 0, 0,
|
|
0, 0, 1, 0)
|
|
self.assertEqual(im.mode, 'RGB')
|
|
|
|
# Act
|
|
# Convert with an identity matrix
|
|
converted_im = im.convert(mode='RGB', matrix=identity_matrix)
|
|
|
|
# Assert
|
|
# No change
|
|
self.assert_image_equal(converted_im, im)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|