def test_add_celery_group(self, some_sigs): wf = Workflow() sigs = some_sigs group = celery_canvas.group(sigs[:4]) dependencies = [ wf.add_signature(sigs[-2]), wf.add_signature(sigs[-1]), ] res = wf.add_celery_group(group, dependencies) assert set(res) == set([ wf.nodes['task-0'], wf.nodes['task-1'], wf.nodes['task-2'], wf.nodes['task-3'], ]) # use build_dependency_asserts(wf) to regenerate # build_dependency_asserts(wf) assert "task-8" in wf.nodes["task-0"].dependencies # noqa assert "task-9" in wf.nodes["task-0"].dependencies # noqa assert "task-8" in wf.nodes["task-1"].dependencies # noqa assert "task-9" in wf.nodes["task-1"].dependencies # noqa assert "task-8" in wf.nodes["task-2"].dependencies # noqa assert "task-9" in wf.nodes["task-2"].dependencies # noqa assert "task-8" in wf.nodes["task-3"].dependencies # noqa assert "task-9" in wf.nodes["task-3"].dependencies # noqa assert wf.nodes["task-8"].dependencies == {} # noqa assert wf.nodes["task-9"].dependencies == {} # noqa
def test_add_celery_chain(self, some_sigs): wf = Workflow() sigs = some_sigs chain = celery_canvas.chain(sigs[:-2]) dependencies = [ wf.add_signature(sigs[-2]), wf.add_signature(sigs[-1]), ] res = wf.add_celery_chain(chain, dependencies) assert res == [wf.nodes[sigs[-3].id]] # use build_dependency_asserts(wf) to regenerate # build_dependency_asserts(wf) assert "task-8" in wf.nodes["task-0"].dependencies # noqa assert "task-9" in wf.nodes["task-0"].dependencies # noqa assert "task-0" in wf.nodes["task-1"].dependencies # noqa assert "task-1" in wf.nodes["task-2"].dependencies # noqa assert "task-2" in wf.nodes["task-3"].dependencies # noqa assert "task-3" in wf.nodes["task-4"].dependencies # noqa assert "task-4" in wf.nodes["task-5"].dependencies # noqa assert "task-5" in wf.nodes["task-6"].dependencies # noqa assert "task-6" in wf.nodes["task-7"].dependencies # noqa assert wf.nodes["task-8"].dependencies == {} # noqa assert wf.nodes["task-9"].dependencies == {} # noqa