def test__DAG_add_task(): A, B = get_two_tasks() dag = DAG() dag.add_task(A) assert dag.tasks == {A,}, 'Test Task was not added to the DAG'
def test__DAG_repr(): make_tea = Task('make_tea.py', 'test-env') dag = DAG() dag.add_task(make_tea) assert repr(dag) == "DAG({Task(make_tea.py)})"
def test__Dequindre_repr(): make_tea = Task('make_tea.py', 'test-env') dag = DAG() dag.add_task(make_tea) dq = Dequindre(dag) assert repr(dq) == "Dequindre(DAG({Task(make_tea.py)}))"