def test_dict_to_proto(self):
self.assertEqual(
proto_utils.dict_to_proto(self.test_dict, foo_pb2.TestProto()),
self.test_proto)
dict_with_obsolete_field = {'obsolete_field': 2, 'string_value': 'x'}
self.assertEqual(
proto_utils.dict_to_proto(dict_with_obsolete_field,
foo_pb2.TestProto()),
foo_pb2.TestProto(string_value='x'))
def setUp(self):
super().setUp()
self.test_proto = foo_pb2.TestProto(
string_value='hello', int_value=2, double_value=0.5)
self.test_dict = {
'string_value': 'hello',
'int_value': 2,
'double_value': 0.5
}
def test_deserialize_proto_message(self):
expected_pb_message = """
string_value: 'hello'
"""
serialized_message = '{"string_value":"hello"}'
message_type = 'tfx.utils.proto.testdata.TestProto'
fd_set = descriptor_pb2.FileDescriptorSet()
foo_pb2.TestProto().DESCRIPTOR.file.CopyToProto(fd_set.file.add())
self.assertProtoPartiallyEquals(
expected_pb_message,
proto_utils.deserialize_proto_message(serialized_message,
message_type, fd_set))
def test_json_to_proto(self):
json_str = '{"obsolete_field":2,"string_value":"x"}'
result = proto_utils.json_to_proto(json_str, foo_pb2.TestProto())
self.assertEqual(result, foo_pb2.TestProto(string_value='x'))
# Make sure that returned type is not message.Message
self.assertEqual(result.string_value, 'x')
def test_json_to_proto(self):
json_str = '{"obsolete_field":2,"string_value":"x"}'
self.assertEqual(
proto_utils.json_to_proto(json_str, foo_pb2.TestProto()),
foo_pb2.TestProto(string_value='x'))