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