54ff7b04669450ceca33852407891c5645472ca5,test/utils/test_convert.py,,test_to_networkx,#,27

Before Change

def test_to_networkx():
    x = torch.Tensor([[1, 2], [3, 4]])
    pos = torch.Tensor([[0, 0], [1, 1]])
    row = torch.tensor([0, 1, 0])
    col = torch.tensor([1, 0, 0])
    edge_attr = torch.Tensor([1, 2, 3])

    G = to_networkx(torch.stack([row, col], dim=0), x, edge_attr, pos)
    assert G.nodes[0]["x"].tolist() == [1, 2]
    assert G.nodes[1]["x"].tolist() == [3, 4]
    assert G.nodes[0]["pos"].tolist() == [0, 0]
    assert G.nodes[1]["pos"].tolist() == [1, 1]

After Change

def test_to_networkx():
    x = torch.Tensor([[1, 2], [3, 4]])
    pos = torch.Tensor([[0, 0], [1, 1]])
    edge_index = torch.tensor([[0, 1, 0], [1, 0, 0]])
    edge_attr = torch.Tensor([1, 2, 3])
    data = Data(x=x, pos=pos, edge_index=edge_index, weight=edge_attr)

    G = to_networkx(data, node_attrs=["x", "pos"], edge_attrs=["weight"])
    assert G.nodes[0]["x"] == [1, 2]
    assert G.nodes[1]["x"] == [3, 4]
    assert G.nodes[0]["pos"] == [0, 0]
    assert G.nodes[1]["pos"] == [1, 1]

Instances