def test_jpeg_cmyk_to_rgb(self):
image = self._get_image('6pixels_cmyk.jpeg')
new_image = image_utils.jpeg_cmyk_to_rgb(image, quality=100)
self.assertNotEqual(image, new_image)
# Converting between color systems is not bijective, so high rtol.
original_np_image = image_utils.decode_image(image)
new_np_image = image_utils.decode_image(new_image)
np.testing.assert_allclose(original_np_image, new_np_image, rtol=10)
def test_png_4chan_to_jpeg(self):
image = self._get_image('6pixels_4chan.png')
jpeg = image_utils.png_to_jpeg(image)
image_np_jpeg = image_utils.decode_image(jpeg)
np.testing.assert_array_equal(image_np_jpeg, SIX_PIXELS_JPEG)
def test_decode_image(self):
image = self._get_image('6pixels.png')
np_image = image_utils.decode_image(image)
np.testing.assert_array_equal(np_image, SIX_PIXELS)
