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

Function test_multigraph_node_removal

tests/python/common/test_heterograph-remove.py:45–68  ·  view source on GitHub ↗
(idtype)

Source from the content-addressed store, hash-verified

43
44@parametrize_idtype
45def test_multigraph_node_removal(idtype):
46 g = create_graph(idtype, 5)
47 for i in range(5):
48 g.add_edges(i, i)
49 g.add_edges(i, i)
50 assert g.num_nodes() == 5
51 assert g.num_edges() == 10
52
53 # remove nodes
54 g.remove_nodes([2, 3])
55 assert g.num_nodes() == 3
56 assert g.num_edges() == 6
57
58 # add nodes
59 g.add_nodes(1)
60 g.add_edges(1, 1)
61 g.add_edges(1, 1)
62 assert g.num_nodes() == 4
63 assert g.num_edges() == 8
64
65 # remove nodes
66 g.remove_nodes([0])
67 assert g.num_nodes() == 3
68 assert g.num_edges() == 6
69
70
71@parametrize_idtype

Callers 1

Calls 6

remove_nodesMethod · 0.80
create_graphFunction · 0.70
add_edgesMethod · 0.45
num_nodesMethod · 0.45
num_edgesMethod · 0.45
add_nodesMethod · 0.45

Tested by

no test coverage detected