()
| 88 | ) |
| 89 | @unittest.skipIf(dgl.backend.backend_name == "mxnet", reason="Skip MXNet") |
| 90 | def test_add_node_property_split(): |
| 91 | dataset = data.AmazonCoBuyComputerDataset() |
| 92 | part_ratios = [0.3, 0.1, 0.1, 0.3, 0.2] |
| 93 | for property_name in ["popularity", "locality", "density"]: |
| 94 | data.utils.add_node_property_split(dataset, part_ratios, property_name) |
| 95 | assert "in_valid_mask" in dataset[0].ndata |
| 96 | |
| 97 | |
| 98 | if __name__ == "__main__": |