Function test_line_graph1

tests/python/common/transforms/test_transform.py:130–137

()

def test_line_graph1():
    N = 5
    G = dgl.from_networkx(nx.star_graph(N)).to(F.ctx())
    G.edata["h"] = F.randn((2 * N, D))
    L = G.line_graph(shared=True)
    assert L.num_nodes() == 2 * N
    assert F.allclose(L.ndata["h"], G.edata["h"])
    assert G.device == F.ctx()


@parametrize_idtype