def test_set_parameters_error():
test_workflow = ArgoWorkflow()
test_workflow.body = {
'spec': {
'arguments': {
'parameters': [{
'name': 'test-param-1',
'value': 'test-value-1'
}, {
'name': 'test-param-2'
}]
}
}
}
with pytest.raises(KeyError):
test_workflow.parameters = {'test-param-1': 'new-value'}
def test_set_parameters():
test_workflow = ArgoWorkflow()
test_workflow.body = {
'spec': {
'arguments': {
'parameters': [{
'name': 'test-param-1',
'value': 'test-value-1'
}, {
'name': 'test-param-2',
'value': 'test-value-2'
}]
}
}
}
test_workflow.parameters = {'test-param-2': 'new-value'}
assert test_workflow.parameters == {
'test-param-1': 'test-value-1',
'test-param-2': 'new-value'
}