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)