From cea36587ab209f00173cdd8716232fcc3673a868 Mon Sep 17 00:00:00 2001 From: hugovk Date: Sun, 29 Jan 2017 14:03:38 +0200 Subject: [PATCH] Test matrix convert --- Tests/test_image_convert.py | 35 +++++++++++++++++++++++++++++++++-- 1 file changed, 33 insertions(+), 2 deletions(-) diff --git a/Tests/test_image_convert.py b/Tests/test_image_convert.py index 54ffde10b..6128dc9d9 100644 --- a/Tests/test_image_convert.py +++ b/Tests/test_image_convert.py @@ -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()