def test_create_metadata_should_succeed(self, content_type, golden_json):
associated_file1 = metadata_info.AssociatedFileMd(
file_path=self._LABEL_FILE_EN, locale="en")
associated_file2 = metadata_info.AssociatedFileMd(
file_path=self._LABEL_FILE_CN, locale="cn")
tensor_md = metadata_info.TensorMd(
name=self._TENSOR_NAME,
description=self._TENSOR_DESCRIPTION,
min_values=[self._TENSOR_MIN],
max_values=[self._TENSOR_MAX],
content_type=content_type,
associated_files=[associated_file1, associated_file2])
tensor_metadata = tensor_md.create_metadata()
metadata_json = _metadata.convert_to_json(
_create_dummy_model_metadata(tensor_metadata))
expected_json = test_utils.load_file(golden_json, "r")
self.assertEqual(metadata_json, expected_json)
def test_create_metadata_should_succeed(self):
file_md = metadata_info.AssociatedFileMd(
file_path="label.txt",
description="The label file.",
file_type=_metadata_fb.AssociatedFileType.TENSOR_AXIS_LABELS,
locale="en")
file_metadata = file_md.create_metadata()
# Create the Flatbuffers object and convert it to the json format.
model_metadata = _metadata_fb.ModelMetadataT()
model_metadata.associatedFiles = [file_metadata]
builder = flatbuffers.Builder(0)
builder.Finish(model_metadata.Pack(builder),
_metadata.MetadataPopulator.METADATA_FILE_IDENTIFIER)
metadata_json = _metadata.convert_to_json(bytes(builder.Output()))
expected_json = test_utils.load_file(self._EXPECTED_META_JSON, "r")
self.assertEqual(metadata_json, expected_json)