Updated names of test case methods
This commit is contained in:
parent
69f4daf9bc
commit
f166eb32e8
|
@ -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)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user