def test_evaluation_propagated_upstream(): visited = [] def func(txt): visited.append(txt) return txt pg = PortGraph() n = FuncNode(func) pg.add_actor(n, 0) pg.add_actor(n, 1) pg.connect(pg.out_port(0, 'txt'), pg.in_port(1, 'txt')) algo = BruteEvaluation(pg) ws = WorkflowState(pg) env = EvaluationEnvironment() ws.store_param(pg.in_port(0, 'txt'), "txt", 0) algo.eval(env, ws, 0) assert len(visited) == 1 algo.eval(env, ws, 1) assert len(visited) == 2 env.new_execution() algo.eval(env, ws, 1) assert len(visited) == 4
def test_portgraph_in_port(): pg = PortGraph() assert_raises(InvalidVertex, lambda: pg.in_port(0, None)) vid = pg.add_vertex() assert_raises(InvalidPort, lambda: pg.in_port(vid, None)) pg.add_out_port(vid, "toto") assert_raises(InvalidPort, lambda: pg.in_port(vid, None)) for lpid in [0, 1, "a", None]: assert_raises(InvalidPort, lambda: pg.in_port(vid, "toto")) pid = pg.add_in_port(vid, lpid) assert pg.in_port(vid, lpid) == pid
def test_lazy_always_reevaluate_if_inputs_changed(): evaluated = [] def func(txt): evaluated.append(txt) return txt pg = PortGraph() vid = pg.add_actor(FuncNode(func)) assert pg.actor(vid).is_lazy() algo = LazyEvaluation(pg) env = EvaluationEnvironment() ws = WorkflowState(pg) ws.store_param(pg.in_port(vid, 'txt'), 'toto', env.current_execution()) algo.eval(env, ws) assert len(evaluated) == 1 env.new_execution() ws.store_param(pg.in_port(vid, 'txt'), 'toto', env.current_execution()) algo.eval(env, ws) assert len(evaluated) == 2
def test_portgraph_add_in_port(): pg = PortGraph() assert_raises(InvalidVertex, lambda: pg.add_in_port(0, "toto")) vid = pg.add_vertex() pid = pg.add_in_port(vid, "port") # raise error if reuse same global port id assert_raises(IndexError, lambda: pg.add_in_port(vid, "toto", pid)) # raise error if reuse same local port id assert_raises(InvalidPort, lambda: pg.add_in_port(vid, "port")) assert tuple(pg.in_ports(vid)) == (pid,) assert pg.local_id(pid) == "port" assert pg.in_port(vid, "port") == pid
def test_portgraph_big(): pg = PortGraph() vid1 = pg.add_vertex() pid11 = pg.add_out_port(vid1, "out") vid2 = pg.add_vertex() pid21 = pg.add_out_port(vid2, "out") vid3 = pg.add_vertex() pid31 = pg.add_in_port(vid3, "in1") pid32 = pg.add_in_port(vid3, "in2") pid33 = pg.add_out_port(vid3, "res") vid4 = pg.add_vertex() pid41 = pg.add_in_port(vid4, "in") eid1 = pg.connect(pid11, pid31) eid2 = pg.connect(pid21, pid32) pg.connect(pid33, pid41) assert pg.source_port(eid1) == pid11 assert pg.target_port(eid2) == pid32 assert set(pg.out_ports(vid1)) == {pid11} assert set(pg.in_ports(vid3)) == {pid31, pid32} assert set(pg.ports(vid3)) == {pid31, pid32, pid33} assert pg.is_in_port(pid31) assert pg.is_out_port(pid11) assert pg.vertex(pid11) == vid1 assert set(pg.connected_ports(pid11)) == {pid31} assert set(pg.connected_edges(pid21)) == {eid2} assert pg.out_port(vid1, "out") == pid11 assert pg.in_port(vid3, "in1") == pid31 assert_raises(InvalidPort, lambda: pg.connect(pid11, pid33)) pg.remove_port(pid33) assert set(pg.connected_ports(pid41)) == set() assert set(pg.out_edges(vid3)) == set() assert_raises(InvalidPort, lambda: pg.is_in_port(pid33))