Skip to content

GitLab

Commit 76ccc51d authored by rusty1s's avatar rusty1s
Browse files

fix tests

parent af9127e3
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/test_permute.py
View file @ 76ccc51d
...@@ -14,4 +14,4 @@ def test_permute(device): ...@@ -14,4 +14,4 @@ def test_permute(device):
row, col, value = adj.permute(torch.tensor([1, 0, 2])).coo() row, col, value = adj.permute(torch.tensor([1, 0, 2])).coo()
assert row.tolist() == [0, 1, 1, 2, 2] assert row.tolist() == [0, 1, 1, 2, 2]
assert col.tolist() == [1, 0, 1, 0, 2] assert col.tolist() == [1, 0, 1, 0, 2]
print(value) assert value.tolist() == [3, 2, 1, 4, 5]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment