def test_set_retry_policy(self): wf = Workflow() wf.set_retry_policy('random', 30, 60) assert wf.retry_policy == ['random', 30, 60]
def test_set_retry_policy_invalid(self): wf = Workflow() with pytest.raises(WorkflowException, match='Invalid retry policy'): wf.set_retry_policy('dummy')