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