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)