()
| 73 | ) |
| 74 | @unittest.skipIf(dgl.backend.backend_name == "mxnet", reason="Skip MXNet") |
| 75 | def test_mask_nodes_by_property(): |
| 76 | num_nodes = 1000 |
| 77 | property_values = np.random.uniform(size=num_nodes) |
| 78 | part_ratios = [0.3, 0.1, 0.1, 0.3, 0.2] |
| 79 | split_masks = data.utils.mask_nodes_by_property( |
| 80 | property_values, part_ratios |
| 81 | ) |
| 82 | assert "in_valid_mask" in split_masks |
| 83 | |
| 84 | |
| 85 | @unittest.skipIf( |