def test_subportgraph_out_edges(): pg = get_pg() sub = SubPortGraph(pg, (0, 2)) assert_raises(InvalidVertex, lambda: tuple(sub.out_edges(10))) assert set(sub.out_edges(0)) == {0} assert sub.nb_out_edges(0) == 1
def test_subportgraph_nb_connections(): pg = get_pg() sub = SubPortGraph(pg, (0, 2)) assert sub.nb_connections(0) == 1 assert sub.nb_connections(2) == 1 assert sub.nb_connections(3) == 0 assert sub.nb_connections(4) == 0 for pid in (1, 5): assert_raises(InvalidPort, lambda: sub.nb_connections(pid))
def test_subportgraph_connected_ports(): pg = get_pg() sub = SubPortGraph(pg, (0, 2)) assert set(sub.connected_ports(0)) == {2} assert set(sub.connected_ports(2)) == {0} assert len(set(sub.connected_ports(3))) == 0 assert len(set(sub.connected_ports(4))) == 0 assert_raises(InvalidPort, lambda: sub.connected_ports(1).next()) assert_raises(InvalidPort, lambda: sub.connected_ports(5).next())
def test_subportgraph_port(): pg = get_pg() sub = SubPortGraph(pg, (0, 2)) assert sub.source_port(0) == pg.source_port(0) assert_raises(InvalidEdge, lambda: sub.source_port(1)) assert_raises(InvalidEdge, lambda: sub.source_port(2)) assert sub.target_port(0) == pg.target_port(0) assert_raises(InvalidEdge, lambda: sub.target_port(1)) assert_raises(InvalidEdge, lambda: sub.target_port(2))
def test_subportgraph_mirror_functions(): pg = get_pg() sub = SubPortGraph(pg, (0, 2)) assert sub.is_in_port(2) assert sub.is_out_port(0) assert sub.vertex(0) == 0 assert sub.local_id(0) == pg.local_id(0) assert sub.in_port(2, "in1") == pg.in_port(2, "in1") assert sub.out_port(0, "out") == pg.out_port(0, "out") assert sub.actor(0) == pg.actor(0) assert 0 in sub
def test_subportgraph_out_ports(): pg = get_pg() sub = SubPortGraph(pg, (0, 2)) assert set(sub.out_ports()) == {0, 4} assert set(sub.out_ports(0)) == {0} assert_raises(InvalidVertex, lambda: sub.out_ports(1).next()) assert set(sub.out_ports(2)) == {4} assert_raises(InvalidVertex, lambda: sub.out_ports(3).next())
def test_subportgraph_in_ports(): pg = get_pg() sub = SubPortGraph(pg, (0, 2)) assert set(sub.in_ports()) == {2, 3} assert len(set(sub.in_ports(0))) == 0 assert_raises(InvalidVertex, lambda: sub.in_ports(1).next()) assert set(sub.in_ports(2)) == {2, 3} assert_raises(InvalidVertex, lambda: sub.in_ports(3).next())
def test_subportgraph_in_neighbors(): pg = get_pg() sub = SubPortGraph(pg, (0, 2)) assert_raises(InvalidVertex, lambda: tuple(sub.in_neighbors(10))) assert set(sub.in_neighbors(2)) == {0}
def test_subportgraph_edges(): pg = get_pg() sub = SubPortGraph(pg, (0, 2)) assert set(sub.edges()) == {0}
def test_subportgraph_init(): pg = PortGraph() sub = SubPortGraph(pg) assert len(tuple(sub.vertices())) == 0
def test_subportgraph_vertices(): pg = get_pg() sub = SubPortGraph(pg, (0, 2)) assert set(sub.vertices()) == {0, 2}