def test_traverser(): graph = generate_dep_graph(foo_pipeline) traverser = Traverser(graph) assert traverser.fetch_upstream(item_name='return_one', depth=1) == set() assert traverser.fetch_downstream(item_name='return_one', depth=1) == set(['add_nums']) assert traverser.fetch_upstream(item_name='multiply_two', depth=0) == set() assert traverser.fetch_upstream(item_name='multiply_two', depth=2) == set( ['add_nums', 'return_one', 'return_two'], ) assert traverser.fetch_downstream(item_name='multiply_two', depth=2) == set(['add_one'])
def test_traverser(): graph = generate_dep_graph(foo_pipeline) traverser = Traverser(graph) assert traverser.fetch_upstream(item_name="return_one", depth=1) == set() assert traverser.fetch_downstream(item_name="return_one", depth=1) == {"add_nums"} assert traverser.fetch_upstream(item_name="multiply_two", depth=0) == set() assert traverser.fetch_upstream(item_name="multiply_two", depth=2) == { "add_nums", "return_one", "return_two", } assert traverser.fetch_downstream(item_name="multiply_two", depth=2) == {"add_one"}
def test_generate_dep_graph(): graph = generate_dep_graph(foo_pipeline) assert graph == { "upstream": { "return_one": set(), "return_two": set(), "add_nums": {"return_one", "return_two"}, "multiply_two": {"add_nums"}, "add_one": {"multiply_two"}, }, "downstream": { "return_one": {"add_nums"}, "return_two": {"add_nums"}, "add_nums": {"multiply_two"}, "multiply_two": {"add_one"}, "add_one": set(), }, }
def test_generate_dep_graph(): graph = generate_dep_graph(foo_pipeline) assert graph == { 'upstream': { 'return_one': set(), 'return_two': set(), 'add_nums': set(['return_one', 'return_two']), 'multiply_two': set(['add_nums']), 'add_one': set(['multiply_two']), }, 'downstream': { 'return_one': set(['add_nums']), 'return_two': set(['add_nums']), 'add_nums': set(['multiply_two']), 'multiply_two': set(['add_one']), 'add_one': set(), }, }
def test_traverser_invalid(): graph = generate_dep_graph(foo_pipeline) traverser = Traverser(graph) assert traverser.fetch_upstream(item_name="some_solid", depth=1) == set()