talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Updated names of test case methods

Browse Source
This commit is contained in:
kinaryml
2022-10-18 04:34:17 -07:00
parent 69f4daf9bc
commit f166eb32e8
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/tasks/python/test/vision/image_segmenter_test.py
+3 -4
View File
@@ -113,7 +113,7 @@ class ImageSegmenterTest(parameterized.TestCase):
@parameterized.parameters( @parameterized.parameters(
(ModelFileType.FILE_NAME,), (ModelFileType.FILE_NAME,),
(ModelFileType.FILE_CONTENT,)) (ModelFileType.FILE_CONTENT,))
def test_succeeds_with_category_mask(self, model_file_type): def test_segment_succeeds_with_category_mask(self, model_file_type):
# Creates segmenter. # Creates segmenter.
if model_file_type is ModelFileType.FILE_NAME: if model_file_type is ModelFileType.FILE_NAME:
base_options = _BaseOptions(model_asset_path=self.model_path) base_options = _BaseOptions(model_asset_path=self.model_path)
@@ -148,7 +148,7 @@ class ImageSegmenterTest(parameterized.TestCase):
# a context. # a context.
segmenter.close() segmenter.close()
def test_succeeds_with_confidence_mask(self): def test_segment_succeeds_with_confidence_mask(self):
# Creates segmenter. # Creates segmenter.
base_options = _BaseOptions(model_asset_path=self.model_path) base_options = _BaseOptions(model_asset_path=self.model_path)
@@ -192,8 +192,7 @@ class ImageSegmenterTest(parameterized.TestCase):
segmenter.close() segmenter.close()
@parameterized.parameters( @parameterized.parameters(
(ModelFileType.FILE_NAME,), (ModelFileType.FILE_NAME), (ModelFileType.FILE_CONTENT))
(ModelFileType.FILE_CONTENT,))
def test_segment_in_context(self, model_file_type): def test_segment_in_context(self, model_file_type):
if model_file_type is ModelFileType.FILE_NAME: if model_file_type is ModelFileType.FILE_NAME:
base_options = _BaseOptions(model_asset_path=self.model_path) base_options = _BaseOptions(model_asset_path=self.model_path)