From c3a1b1add7cabceca1b851966ae1b254312035ab Mon Sep 17 00:00:00 2001 From: kinaryml Date: Fri, 14 Apr 2023 11:27:27 -0700 Subject: [PATCH] Updated the no_face_detected test case --- mediapipe/tasks/python/test/vision/face_stylizer_test.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/mediapipe/tasks/python/test/vision/face_stylizer_test.py b/mediapipe/tasks/python/test/vision/face_stylizer_test.py index c104d6a8c..c57618801 100644 --- a/mediapipe/tasks/python/test/vision/face_stylizer_test.py +++ b/mediapipe/tasks/python/test/vision/face_stylizer_test.py @@ -180,9 +180,7 @@ class FaceStylizerTest(parameterized.TestCase): image_processing_options = _ImageProcessingOptions(roi) # Performs face stylization on the input. stylized_image = stylizer.stylize(test_image, image_processing_options) - self.assertIsInstance(stylized_image, _Image) - self.assertEqual(stylized_image.width, _MODEL_IMAGE_SIZE) - self.assertEqual(stylized_image.height, _MODEL_IMAGE_SIZE) + self.assertIsNone(stylized_image) def test_missing_result_callback(self): options = _FaceStylizerOptions(