def test_portgraph_out_port():
pg = PortGraph()
assert_raises(InvalidVertex, lambda: pg.out_port(0, None))
vid = pg.add_vertex()
assert_raises(InvalidPort, lambda: pg.out_port(vid, None))
pg.add_in_port(vid, "toto")
assert_raises(InvalidPort, lambda: pg.out_port(vid, None))
for lpid in [0, 1, "a", None]:
assert_raises(InvalidPort, lambda: pg.out_port(vid, "toto"))
pid = pg.add_out_port(vid, lpid)
assert pg.out_port(vid, lpid) == pid
def test_portgraph_out_port():
pg = PortGraph()
assert_raises(InvalidVertex, lambda: pg.out_port(0, None))
vid = pg.add_vertex()
assert_raises(InvalidPort, lambda: pg.out_port(vid, None))
pg.add_in_port(vid, "toto")
assert_raises(InvalidPort, lambda: pg.out_port(vid, None))
for lpid in [0, 1, "a", None]:
assert_raises(InvalidPort, lambda: pg.out_port(vid, "toto"))
pid = pg.add_out_port(vid, lpid)
assert pg.out_port(vid, lpid) == pid
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_add_out_port():
pg = PortGraph()
assert_raises(InvalidVertex, lambda: pg.add_out_port(0, "toto"))
vid = pg.add_vertex()
pid = pg.add_out_port(vid, "port")
# raise error if reuse same global port id
assert_raises(IndexError, lambda: pg.add_out_port(vid, "toto", pid))
# raise error if reuse same local port id
assert_raises(InvalidPort, lambda: pg.add_out_port(vid, "port"))
assert tuple(pg.out_ports(vid)) == (pid,)
assert pg.local_id(pid) == "port"
assert pg.out_port(vid, "port") == pid
def test_portgraph_add_out_port():
pg = PortGraph()
assert_raises(InvalidVertex, lambda: pg.add_out_port(0, "toto"))
vid = pg.add_vertex()
pid = pg.add_out_port(vid, "port")
# raise error if reuse same global port id
assert_raises(IndexError, lambda: pg.add_out_port(vid, "toto", pid))
# raise error if reuse same local port id
assert_raises(InvalidPort, lambda: pg.add_out_port(vid, "port"))
assert tuple(pg.out_ports(vid)) == (pid,)
assert pg.local_id(pid) == "port"
assert pg.out_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))
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))
