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