a3a05105f0a696a99f0b65384f9bed8e8702256f,test/utils/test_loop.py,,test_add_self_loops,#,5

Before Change

def test_add_self_loops():
    edge_index = torch.LongTensor([[0, 1, 0], [1, 0, 0]])
    expected_output = [[0, 1, 0, 0, 1], [1, 0, 0, 0, 1]]

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


def test_remove_self_loops():

After Change

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

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


def test_remove_self_loops():

Instances