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()
