a3a05105f0a696a99f0b65384f9bed8e8702256f,test/utils/test_coalesce.py,,test_coalesce,#,5

Before Change

    edge_index = torch.LongTensor([[1, 0, 1, 0, 2, 1], [0, 1, 1, 1, 0, 0]])
    expected_output = [[0, 1, 1, 2], [1, 0, 1, 0]]

    output = coalesce(edge_index)
    assert output.tolist() == expected_output

After Change

def test_coalesce():
    row = torch.LongTensor([1, 0, 1, 0, 2, 1])
    col = torch.LongTensor([0, 1, 1, 1, 0, 0])
    expected_output = [[0, 1, 1, 2], [1, 0, 1, 0]]

    output = coalesce(torch.stack([row, col], dim=0))
    assert output.tolist() == expected_output

Instances