Test matrix convert

This commit is contained in:
hugovk 2017-01-29 14:03:38 +02:00
parent 6d11fa23dd
commit cea36587ab

View File

@ -20,10 +20,10 @@ class TestImageConvert(PillowTestCase):
convert(im, mode)
# Check 0
im = Image.new(mode, (0,0))
im = Image.new(mode, (0, 0))
for mode in modes:
convert(im, mode)
def test_default(self):
im = hopper("P")
@ -137,6 +137,37 @@ class TestImageConvert(PillowTestCase):
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,
lambda: im.convert(mode='CMYK', matrix=matrix))
def test_matrix(self):
# Arrange
im = hopper('RGB')
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='RGB', matrix=matrix)
# Assert
self.assertEqual(converted_im.mode, 'RGB')
self.assertEqual(converted_im.size, im.size)
if __name__ == '__main__':
unittest.main()