Updated the no_face_detected test case

This commit is contained in:
kinaryml 2023-04-14 11:27:27 -07:00
parent e17d401c6e
commit c3a1b1add7

View File

@ -180,9 +180,7 @@ class FaceStylizerTest(parameterized.TestCase):
image_processing_options = _ImageProcessingOptions(roi) image_processing_options = _ImageProcessingOptions(roi)
# Performs face stylization on the input. # Performs face stylization on the input.
stylized_image = stylizer.stylize(test_image, image_processing_options) stylized_image = stylizer.stylize(test_image, image_processing_options)
self.assertIsInstance(stylized_image, _Image) self.assertIsNone(stylized_image)
self.assertEqual(stylized_image.width, _MODEL_IMAGE_SIZE)
self.assertEqual(stylized_image.height, _MODEL_IMAGE_SIZE)
def test_missing_result_callback(self): def test_missing_result_callback(self):
options = _FaceStylizerOptions( options = _FaceStylizerOptions(