def test_portgraph_edge_connected_to_ports(): pg = PortGraph() vid1 = pg.add_vertex() vid2 = pg.add_vertex() # method add_edge is forbidden assert_raises(UserWarning, lambda: pg.add_edge((vid1, vid2))) # connect to None port raises InvalidPort assert_raises(InvalidPort, lambda: pg.connect(None, None)) pid1 = pg.add_out_port(vid1, "out") pid2 = pg.add_in_port(vid2, "in") assert_raises(InvalidPort, lambda: pg.connect(pid1, None)) assert_raises(InvalidPort, lambda: pg.connect(None, pid2)) # connect from in port to out port raises InvalidPort assert_raises(InvalidPort, lambda: pg.connect(pid2, pid1)) # test connect is working eid = pg.connect(pid1, pid2) assert pg.source_port(eid) == pid1 assert pg.target_port(eid) == pid2
def test_portgraph_edge_connected_to_ports(): pg = PortGraph() vid1 = pg.add_vertex() vid2 = pg.add_vertex() # method add_edge is forbidden assert_raises(UserWarning, lambda: pg.add_edge((vid1, vid2))) # connect to None port raises InvalidPort assert_raises(InvalidPort, lambda: pg.connect(None, None)) pid1 = pg.add_out_port(vid1, "out") pid2 = pg.add_in_port(vid2, "in") assert_raises(InvalidPort, lambda: pg.connect(pid1, None)) assert_raises(InvalidPort, lambda: pg.connect(None, pid2)) # connect from in port to out port raises InvalidPort assert_raises(InvalidPort, lambda: pg.connect(pid2, pid1)) # test connect is working eid = pg.connect(pid1, pid2) assert pg.source_port(eid) == pid1 assert pg.target_port(eid) == pid2