from ..graph import load_graph
from ..graph.analysis import link_is_required
[docs]
def test_graph_link_is_required_conditions1():
graph = {"id": "test", "schema_version": "1.1"}
nodes = [
{"id": "start", "task_type": "method", "task_identifier": "dummy"},
{"id": "fan", "task_type": "method", "task_identifier": "dummy"},
{"id": "on_false1", "task_type": "method", "task_identifier": "dummy"},
{"id": "on_true1", "task_type": "method", "task_identifier": "dummy"},
{"id": "on_false2", "task_type": "method", "task_identifier": "dummy"},
{"id": "on_true2", "task_type": "method", "task_identifier": "dummy"},
{"id": "merge", "task_type": "method", "task_identifier": "dummy"},
{"id": "end", "task_type": "method", "task_identifier": "dummy"},
]
links = [
{"source": "start", "target": "fan", "map_all_data": True},
{
"source": "fan",
"target": "on_false1",
"map_all_data": True,
"conditions": [{"source_output": "result", "value": False}],
},
{
"source": "fan",
"target": "on_true1",
"map_all_data": True,
"conditions": [{"source_output": "result", "value": True}],
},
{"source": "on_false1", "target": "on_false2", "map_all_data": True},
{"source": "on_true1", "target": "on_true2", "map_all_data": True},
{"source": "on_false2", "target": "merge", "map_all_data": True},
{"source": "on_true2", "target": "merge", "map_all_data": True},
{"source": "merge", "target": "end", "map_all_data": True},
]
taskgraph = load_graph({"graph": graph, "nodes": nodes, "links": links})
assert link_is_required(taskgraph.graph, "start", "fan")
assert not link_is_required(taskgraph.graph, "fan", "on_false1")
assert not link_is_required(taskgraph.graph, "fan", "on_true1")
assert not link_is_required(taskgraph.graph, "on_false1", "on_false2")
assert not link_is_required(taskgraph.graph, "on_true1", "on_true2")
assert not link_is_required(taskgraph.graph, "on_false2", "merge")
assert not link_is_required(taskgraph.graph, "on_true2", "merge")
assert not link_is_required(
taskgraph.graph, "merge", "end"
) # TODO: this should be True because branches merge again
[docs]
def test_graph_link_is_required_conditions2():
graph = {"id": "test", "schema_version": "1.1"}
nodes = [
{"id": "start", "task_type": "method", "task_identifier": "dummy"},
{"id": "fan", "task_type": "method", "task_identifier": "dummy"},
{"id": "always1", "task_type": "method", "task_identifier": "dummy"},
{"id": "on_true1", "task_type": "method", "task_identifier": "dummy"},
{"id": "always2", "task_type": "method", "task_identifier": "dummy"},
{"id": "on_true2", "task_type": "method", "task_identifier": "dummy"},
{"id": "merge", "task_type": "method", "task_identifier": "dummy"},
{"id": "end_always", "task_type": "method", "task_identifier": "dummy"},
]
links = [
{"source": "start", "target": "fan", "map_all_data": True},
{"source": "fan", "target": "always1", "map_all_data": True},
{
"source": "fan",
"target": "on_true1",
"map_all_data": True,
"conditions": [{"source_output": "result", "value": True}],
},
{"source": "always1", "target": "always2", "map_all_data": True},
{"source": "on_true1", "target": "on_true2", "map_all_data": True},
{"source": "always2", "target": "merge", "map_all_data": True},
{"source": "on_true2", "target": "merge", "map_all_data": True},
{"source": "merge", "target": "end_always", "map_all_data": True},
]
taskgraph = load_graph({"graph": graph, "nodes": nodes, "links": links})
assert link_is_required(taskgraph.graph, "start", "fan")
assert link_is_required(taskgraph.graph, "fan", "always1")
assert not link_is_required(taskgraph.graph, "fan", "on_true1")
assert link_is_required(taskgraph.graph, "always1", "always2")
assert not link_is_required(taskgraph.graph, "on_true1", "on_true2")
assert link_is_required(taskgraph.graph, "always2", "merge")
assert not link_is_required(taskgraph.graph, "on_true2", "merge")
assert not link_is_required(
taskgraph.graph, "merge", "end_always"
) # TODO: this should be True because branches merge again
[docs]
def test_graph_link_is_required_errors():
graph = {"id": "test", "schema_version": "1.1"}
nodes = [
{"id": "start", "task_type": "method", "task_identifier": "dummy"},
{"id": "fan", "task_type": "method", "task_identifier": "dummy"},
{"id": "always1", "task_type": "method", "task_identifier": "dummy"},
{"id": "on_true1", "task_type": "method", "task_identifier": "dummy"},
{"id": "on_error1", "task_type": "method", "task_identifier": "dummy"},
{"id": "always2", "task_type": "method", "task_identifier": "dummy"},
{"id": "on_true2", "task_type": "method", "task_identifier": "dummy"},
{"id": "on_error2", "task_type": "method", "task_identifier": "dummy"},
{"id": "merge", "task_type": "method", "task_identifier": "dummy"},
{"id": "end_always", "task_type": "method", "task_identifier": "dummy"},
{"id": "end_on_error", "task_type": "method", "task_identifier": "dummy"},
]
links = [
{"source": "start", "target": "fan", "map_all_data": True},
{"source": "fan", "target": "always1", "map_all_data": True},
{
"source": "fan",
"target": "on_true1",
"map_all_data": True,
"conditions": [{"source_output": "result", "value": True}],
},
{
"source": "fan",
"target": "on_error1",
"map_all_data": True,
"on_error": True,
},
{"source": "always1", "target": "always2", "map_all_data": True},
{"source": "on_true1", "target": "on_true2", "map_all_data": True},
{"source": "on_error1", "target": "on_error2", "map_all_data": True},
{"source": "always2", "target": "merge", "map_all_data": True},
{"source": "on_true2", "target": "merge", "map_all_data": True},
{"source": "on_error2", "target": "merge", "map_all_data": True},
{"source": "merge", "target": "end_always", "map_all_data": True},
{"source": "on_error2", "target": "end_on_error", "map_all_data": True},
]
taskgraph = load_graph({"graph": graph, "nodes": nodes, "links": links})
assert link_is_required(taskgraph.graph, "start", "fan")
assert link_is_required(taskgraph.graph, "fan", "always1")
assert not link_is_required(taskgraph.graph, "fan", "on_true1")
assert not link_is_required(taskgraph.graph, "fan", "on_error1")
assert not link_is_required(taskgraph.graph, "always1", "always2")
assert not link_is_required(taskgraph.graph, "on_true1", "on_true2")
assert not link_is_required(taskgraph.graph, "on_error1", "on_error2")
assert not link_is_required(taskgraph.graph, "always2", "merge")
assert not link_is_required(taskgraph.graph, "on_true2", "merge")
assert not link_is_required(taskgraph.graph, "on_error2", "merge")
assert not link_is_required(
taskgraph.graph, "merge", "end_always"
) # TODO: this should be True because branches merge again
assert not link_is_required(taskgraph.graph, "on_error2", "end_on_error")