()
| 65 | ) |
| 66 | @unittest.skipIf(dgl.backend.backend_name == "mxnet", reason="Skip MXNet") |
| 67 | def test_fraud(): |
| 68 | transform = dgl.AddSelfLoop(allow_duplicate=True) |
| 69 | |
| 70 | g = data.FraudDataset("amazon")[0] |
| 71 | assert g.num_nodes() == 11944 |
| 72 | num_edges1 = g.num_edges() |
| 73 | g2 = data.FraudDataset("amazon", transform=transform)[0] |
| 74 | # 3 edge types |
| 75 | assert g2.num_edges() - num_edges1 == g.num_nodes() * 3 |
| 76 | |
| 77 | g = data.FraudAmazonDataset()[0] |
| 78 | assert g.num_nodes() == 11944 |
| 79 | g2 = data.FraudAmazonDataset(transform=transform)[0] |
| 80 | # 3 edge types |
| 81 | assert g2.num_edges() - g.num_edges() == g.num_nodes() * 3 |
| 82 | |
| 83 | g = data.FraudYelpDataset()[0] |
| 84 | assert g.num_nodes() == 45954 |
| 85 | g2 = data.FraudYelpDataset(transform=transform)[0] |
| 86 | # 3 edge types |
| 87 | assert g2.num_edges() - g.num_edges() == g.num_nodes() * 3 |
| 88 | |
| 89 | |
| 90 | @unittest.skipIf( |
no test coverage detected