Updated names of test case methods

This commit is contained in:
kinaryml 2022-10-18 04:34:17 -07:00
parent 69f4daf9bc
commit f166eb32e8

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)