def test_append_edges(self): data = GraphData( torch.randn(10, 5), torch.randn(5, 4), torch.randn(1, 3), torch.randint(0, 10, torch.Size([2, 5])), ) e = torch.randn(3, 4) edges = torch.randint(0, 10, torch.Size([2, 3])) data.append_edges(e, edges)
def test_invalid_append_edges(self): data = GraphData( torch.randn(10, 5), torch.randn(5, 4), torch.randn(1, 3), torch.randint(0, 10, torch.Size([2, 5])), ) e = torch.randn(3, 4) edges = torch.randint(0, 10, torch.Size([2, 4])) with pytest.raises(RuntimeError): data.append_edges(e, edges)