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'))