def test_two_fork_connection_mismatch_lane():
source_lane = 1
lane = 3
res = ps.fork_connection(source="A",
sink=["B", "C"],
source_lane=source_lane,
lane=lane)
assert res == [{
"input": {
"process": "A",
"lane": source_lane
},
"output": {
"process": "B",
"lane": lane + 1
}
}, {
"input": {
"process": "A",
"lane": source_lane,
},
"output": {
"process": "C",
"lane": lane + 2
}
}]
def test_multi_fork_connection():
source_lane = 1
res = ps.fork_connection(source="A",
sink=["B", "C", "D"],
source_lane=source_lane,
lane=source_lane)
assert res == [{
"input": {
"process": "A",
"lane": source_lane
},
"output": {
"process": "B",
"lane": source_lane + 1
}
}, {
"input": {
"process": "A",
"lane": source_lane,
},
"output": {
"process": "C",
"lane": source_lane + 2
}
}, {
"input": {
"process": "A",
"lane": source_lane,
},
"output": {
"process": "D",
"lane": source_lane + 3
}
}]