def test_ws_retrieve_data_on_input_ports():
pg = PortGraph()
pg.add_vertex(0)
pg.add_out_port(0, "out", 0)
pg.add_vertex(1)
pg.add_in_port(1, "in", 1)
# non existing port
ws = WorkflowState(pg)
assert_raises(KeyError, lambda: ws.get(10))
# lonely input port without data
ws = WorkflowState(pg)
assert_raises(KeyError, lambda: ws.get(1))
# lonely input port with data
ws.store_param(1, "param", 0)
assert ws.get(1) == "param"
# connected input port
pg.connect(0, 1)
ws = WorkflowState(pg)
ws.store(0, "data")
assert ws.get(1) == "data"
def test_ws_retrieve_stored_data_on_output_port():
pg = PortGraph()
pg.add_vertex(0)
pg.add_out_port(0, "out", 0)
ws = WorkflowState(pg)
ws.store(0, "data")
assert_raises(KeyError, lambda: ws.get(1))
assert ws.get(0) == "data"
def test_ws_can_not_store_data_on_input_port():
pg = PortGraph()
pg.add_vertex(0)
pg.add_in_port(0, "in", 0)
ws = WorkflowState(pg)
assert_raises(UserWarning, lambda: ws.store(0, "data"))
def test_ws_retrieve_data_on_multiple_connections_input_ports():
pg = PortGraph()
pg.add_vertex(0)
pg.add_out_port(0, "out", 0)
pg.add_vertex(1)
pg.add_out_port(1, "out", 1)
pg.add_vertex(2)
pg.add_in_port(2, "in", 2)
pg.connect(0, 2)
pg.connect(1, 2)
ws = WorkflowState(pg)
assert_raises(KeyError, lambda: ws.get(2))
ws.store(0, "data0")
assert_raises(KeyError, lambda: ws.get(2))
ws.store(1, "data1")
assert tuple(ws.get(2)) == ("data0", "data1")
def test_ws_is_ready_for_evaluation():
pg = PortGraph()
pg.add_vertex(0)
pg.add_in_port(0, "in", 0)
pg.add_out_port(0, "out", 1)
pg.add_vertex(1)
pg.add_out_port(1, "out", 2)
pg.add_vertex(2)
pg.add_in_port(2, "in", 3)
pg.add_out_port(2, "out", 4)
pg.connect(1, 3)
pg.connect(2, 3)
ws = WorkflowState(pg)
assert not ws.is_ready_for_evaluation()
ws.store(1, "data")
assert not ws.is_ready_for_evaluation()
ws.store_param(0, "param", 0)
assert ws.is_ready_for_evaluation()
