def test_portgraph_remove_port(): pg = PortGraph() vid1 = pg.add_vertex() vid2 = pg.add_vertex() pid1 = pg.add_in_port(vid1, "in") pid2 = pg.add_out_port(vid2, "out") pg.connect(pid2, pid1) # raise error if global port id dos not exist assert_raises(InvalidPort, lambda: pg.remove_port(pid1 + pid2 + 1)) pg.remove_port(pid1) assert len(tuple(pg.ports(vid1))) == 0 assert tuple(pg.ports(vid2)) == (pid2,) assert pg.nb_connections(pid2) == 0
def test_portgraph_remove_vertex(): pg = PortGraph() vid1 = pg.add_vertex() vid2 = pg.add_vertex() pid1 = pg.add_in_port(vid1, "in") pid2 = pg.add_out_port(vid2, "out") pg.connect(pid2, pid1) assert_raises(InvalidVertex, lambda: pg.remove_vertex(vid1 + vid2 + 1)) pg.remove_vertex(vid1) assert tuple(pg.ports()) == (pid2,) assert_raises(InvalidVertex, lambda: pg.ports(vid1).next()) assert tuple(pg.ports(vid2)) == (pid2,) assert pg.nb_connections(pid2) == 0 assert pg.nb_neighbors(vid2) == 0
def test_portgraph_add_vertex(): pg = PortGraph() vid = pg.add_vertex() assert pg.nb_vertices() == 1 assert_raises(InvalidVertex, lambda: pg.add_vertex(vid)) assert len(tuple(pg.ports(vid))) == 0 assert pg.actor(vid) is None
def test_portgraph_ports(): pg = PortGraph() assert_raises(InvalidVertex, lambda: pg.ports(0)) vid1 = pg.add_vertex() vid2 = pg.add_vertex() # test new vertices are created without ports assert len(tuple(pg.ports(vid1))) == 0 assert len(tuple(pg.ports())) == 0 ipid1 = pg.add_in_port(vid1, "in") opids = [pg.add_out_port(vid1, "out%d" % i) for i in range(5)] ipid2 = pg.add_in_port(vid2, "in") opid = pg.add_out_port(vid2, "out") assert sorted(pg.ports(vid1)) == sorted([ipid1] + opids) assert sorted(pg.ports()) == sorted([ipid1] + opids + [ipid2, opid])
def test_portgraph_clear(): pg = PortGraph() vid1 = pg.add_vertex() vid2 = pg.add_vertex() pid1 = pg.add_in_port(vid1, "in") pid2 = pg.add_out_port(vid2, "out") pg.connect(pid2, pid1) pg.clear() assert pg.nb_vertices() == 0 assert pg.nb_edges() == 0 assert len(tuple(pg.ports())) == 0
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))