MCPcopy
hub / github.com/dmlc/dgl / test_to_cugraph_conversion

Function test_to_cugraph_conversion

tests/cugraph/test_basics.py:14–23  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

12
13
14def test_to_cugraph_conversion():
15 g = dgl.graph((F.tensor([0, 1, 2, 3]), F.tensor([1, 0, 3, 2]))).to("cuda")
16 cugraph_g = g.to_cugraph()
17
18 assert cugraph_g.number_of_nodes() == g.num_nodes()
19 assert cugraph_g.number_of_edges() == g.num_edges()
20
21 assert cugraph_g.has_edge(0, 1)
22 assert cugraph_g.has_edge(1, 0)
23 assert cugraph_g.has_edge(3, 2)
24
25
26def test_from_cugraph_conversion():

Callers

nothing calls this directly

Calls 6

toMethod · 0.45
graphMethod · 0.45
number_of_nodesMethod · 0.45
num_nodesMethod · 0.45
number_of_edgesMethod · 0.45
num_edgesMethod · 0.45

Tested by

no test coverage detected