Python update_plan_from_graph示例

示例#1

def test_status_preserved_on_update(simple):
    new = deepcopy(simple)
    task_under_test = 'echo_stuff'

    assert new.node[task_under_test]['status'] == states.PENDING.name

    simple.node[task_under_test]['status'] = states.SUCCESS.name
    graph.update_plan_from_graph(new, simple)

    updated = graph.get_plan(new.graph['uid'])
    assert new.node[task_under_test]['status'] == states.SUCCESS.name

示例#2

def test_update_plan_with_new_node(simple):
    new = deepcopy(simple)
    new.add_node('one_more', {})
    graph.update_plan_from_graph(new, simple)
    updated = graph.get_plan(new.graph['uid'])
    assert set(updated.nodes()) == {'one_more', 'just_fail', 'echo_stuff'}