()
| 22 | ) |
| 23 | @unittest.skipIf(dgl.backend.backend_name == "mxnet", reason="Skip MXNet") |
| 24 | def test_reddit(): |
| 25 | # RedditDataset |
| 26 | g = data.RedditDataset()[0] |
| 27 | assert g.num_nodes() == 232965 |
| 28 | assert g.num_edges() == 114615892 |
| 29 | dst = F.asnumpy(g.edges()[1]) |
| 30 | assert np.array_equal(dst, np.sort(dst)) |
| 31 | |
| 32 | transform = dgl.AddSelfLoop(allow_duplicate=True) |
| 33 | g2 = data.RedditDataset(transform=transform)[0] |
| 34 | assert g2.num_edges() - g.num_edges() == g.num_nodes() |
| 35 | |
| 36 | |
| 37 | @unittest.skipIf( |