def test_flow_parameters(): f = Flow(name="test") x = Parameter("x") y = Parameter("y", default=1) f.add_task(x) f.add_task(y) assert f.parameters() == {x, y}
def test_raise_error_if_two_parameters_have_same_name(): f = Flow(name="test") f.add_task(Parameter("x")) assert "x" in {p.name for p in f.parameters()} with pytest.raises(ValueError): f.add_task(Parameter("x"))