def testResolveParameterSchema(self): parameters = pipeline_pb2.NodeParameters() text_format.Parse( """ parameters { key: 'key_one' value { field_value { string_value: 'value_one' } schema { value_type { boolean_type {} } } } } parameters { key: 'key_two' value { field_value { int_value: 2 } } }""", parameters) expected_schema = """ field_value { string_value: 'value_one' } schema { value_type { boolean_type {} } } """ parameter_schemas = inputs_utils.resolve_parameters_with_schema( parameters) self.assertEqual(len(parameter_schemas), 2) self.assertProtoPartiallyEquals(expected_schema, parameter_schemas['key_one'])
def testResolveParametersFail(self): parameters = pipeline_pb2.NodeParameters() text_format.Parse( """ parameters { key: 'key_one' value { runtime_parameter {name: 'rp'} } }""", parameters) with self.assertRaisesRegex(RuntimeError, 'Parameter value not ready'): inputs_utils.resolve_parameters(parameters)
def testResolveParameters(self): parameters = pipeline_pb2.NodeParameters() text_format.Parse( """ parameters { key: 'key_one' value { field_value {string_value: 'value_one'} } } parameters { key: 'key_two' value { field_value {int_value: 2} } }""", parameters) parameters = inputs_utils.resolve_parameters(parameters) self.assertEqual(len(parameters), 2) self.assertEqual(parameters['key_one'], 'value_one') self.assertEqual(parameters['key_two'], 2)