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)
