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