MCPcopy Create free account
hub / github.com/dmlc/dgl / test_minigc

Function test_minigc

tests/python/common/data/test_data.py:25–33  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

23)
24@unittest.skipIf(dgl.backend.backend_name == "mxnet", reason="Skip MXNet")
25def test_minigc():
26 ds = data.MiniGCDataset(16, 10, 20)
27 g, l = list(zip(*ds))
28 print(g, l)
29 g1 = ds[0][0]
30 transform = dgl.AddSelfLoop(allow_duplicate=True)
31 ds = data.MiniGCDataset(16, 10, 20, transform=transform)
32 g2 = ds[0][0]
33 assert g2.num_edges() - g1.num_edges() == g1.num_nodes()
34
35
36@unittest.skipIf(

Callers 1

test_data.pyFile · 0.85

Calls 2

num_edgesMethod · 0.45
num_nodesMethod · 0.45

Tested by

no test coverage detected