7f69a4ff73a73e4b2da4f554ee4640e185aa19a1,test/utils/test_coalesce.py,,test_coalesce,#,5

Before Change

    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]]
    expected_perm = [1, 0, 2, 4]

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

After Change

    edge_attr = [[1, 2], [3, 4], [5, 6], [7, 8], [9, 10], [11, 12]]
    edge_attr = torch.tensor(edge_attr)

    out, _ = coalesce(edge_index)
    assert out.tolist() == [[0, 1, 1, 2], [1, 0, 1, 0]]

    out = coalesce(edge_index, edge_attr)
    assert out[0].tolist() == [[0, 1, 1, 2], [1, 0, 1, 0]]

Instances