No public description

PiperOrigin-RevId: 577873616
This commit is contained in:
Sebastian Schmidt 2023-10-30 09:44:11 -07:00 committed by Copybara-Service
parent 2f4d7b4079
commit 7256bd2638

View File

@ -207,7 +207,9 @@ class ImageTest(absltest.TestCase):
loaded_image = Image.create_from_file(image_path) loaded_image = Image.create_from_file(image_path)
self.assertEqual(loaded_image.width, 720) self.assertEqual(loaded_image.width, 720)
self.assertEqual(loaded_image.height, 382) self.assertEqual(loaded_image.height, 382)
self.assertEqual(loaded_image.channels, 3) # On Mac w/ GPU support, images use 4 channels. Otherwise, all images use
# 3 channels.
self.assertIn(loaded_image.channels, [3, 4])
self.assertEqual(loaded_image.image_format, ImageFormat.SRGB) self.assertEqual(loaded_image.image_format, ImageFormat.SRGB)
if __name__ == '__main__': if __name__ == '__main__':