fix tests

76ccc51d · rusty1s · af9127e3 · 76ccc51d
Commit 76ccc51d authored Feb 19, 2020 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test/test_permute.py test/test_permute.py +1 -1

No files found.
--- a/test/test_permute.py
+++ b/test/test_permute.py
@@ -14,4 +14,4 @@ def test_permute(device):
    row, col, value = adj.permute(torch.tensor([1, 0, 2])).coo()
    assert row.tolist() == [0, 1, 1, 2, 2]
    assert col.tolist() == [1, 0, 1, 0, 2]
-    print(value)
+    assert value.tolist() == [3, 2, 1, 4, 5]