diff --git a/tests/domain/tasks/test_subtasks.py b/tests/domain/tasks/test_subtasks.py index 8cee44960..d379439c6 100644 --- a/tests/domain/tasks/test_subtasks.py +++ b/tests/domain/tasks/test_subtasks.py @@ -58,6 +58,20 @@ def test_start_rejects_from_terminal_status() -> None: st.start() +def test_start_restarts_a_failed_subtask_for_a_rerun() -> None: + # arrange + st = SubTask.create(task_id=uuid4()) + st.start() + st.fail(RuntimeError("degenerate prediction")) + + # act + st.start() + + # assert + assert st.status is SubTaskStatus.IN_PROGRESS + assert st.job_completed is None + + def test_complete_marks_outputs_and_job_completed() -> None: # arrange st = SubTask.create(task_id=uuid4())