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

Before Change

    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

    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]]
    assert out[1].tolist() == [[10, 12], [12, 14], [5, 6], [9, 10]]

Instances