()
| 3 | |
| 4 | |
| 5 | def test_fb13(): |
| 6 | args = build_args_from_dict({"dataset": "fb13"}) |
| 7 | assert args.dataset == "fb13" |
| 8 | dataset = build_dataset(args) |
| 9 | data = dataset.data |
| 10 | assert len(data.edge_index) == 2 |
| 11 | assert data.edge_index[0].shape[0] == 345873 |
| 12 | assert data.edge_attr.shape[0] == 345873 |
| 13 | |
| 14 | |
| 15 | def test_fb15k(): |
no test coverage detected