()
| 10 | F._default_context_str == "gpu", reason="GPU random choice not implemented" |
| 11 | ) |
| 12 | def test_rand_graph(): |
| 13 | g = dgl.rand_graph(10000, 100000) |
| 14 | assert g.num_nodes() == 10000 |
| 15 | assert g.num_edges() == 100000 |
| 16 | # test random seed |
| 17 | dgl.random.seed(42) |
| 18 | g1 = dgl.rand_graph(100, 30) |
| 19 | dgl.random.seed(42) |
| 20 | g2 = dgl.rand_graph(100, 30) |
| 21 | u1, v1 = g1.edges() |
| 22 | u2, v2 = g2.edges() |
| 23 | assert F.array_equal(u1, u2) |
| 24 | assert F.array_equal(v1, v2) |
| 25 | |
| 26 | |
| 27 | if __name__ == "__main__": |
no test coverage detected