()
| 63 | dgl.backend.backend_name != "pytorch", reason="only supports pytorch" |
| 64 | ) |
| 65 | def test_peptides_structural(): |
| 66 | transform = dgl.AddSelfLoop(allow_duplicate=True) |
| 67 | dataset1 = data.PeptidesStructuralDataset() |
| 68 | g1 = dataset1[0][0] |
| 69 | dataset2 = data.PeptidesStructuralDataset(transform=transform) |
| 70 | g2 = dataset2[0][0] |
| 71 | |
| 72 | assert g2.num_edges() - g1.num_edges() == g1.num_nodes() |
| 73 | |
| 74 | |
| 75 | @unittest.skipIf( |