()
| 111 | |
| 112 | |
| 113 | def _test_map_to_subgraph(): |
| 114 | g = dgl.graph([]) |
| 115 | g.add_nodes(10) |
| 116 | g.add_edges(F.arange(0, 9), F.arange(1, 10)) |
| 117 | h = g.subgraph([0, 1, 2, 5, 8]) |
| 118 | v = h.map_to_subgraph_nid([0, 8, 2]) |
| 119 | assert np.array_equal(F.asnumpy(v), np.array([0, 4, 2])) |
| 120 | |
| 121 | |
| 122 | def create_test_heterograph(idtype): |