Commit 24595a8f authored by rusty1s's avatar rusty1s
Browse files

fix test

parent 5d12a68a
...@@ -27,4 +27,4 @@ def test_rw(device): ...@@ -27,4 +27,4 @@ def test_rw(device):
walk_length = 4 walk_length = 4
out = random_walk(row, col, start, walk_length, num_nodes=3) out = random_walk(row, col, start, walk_length, num_nodes=3)
assert out.tolist() == [[0, 1, 0, 1], [1, 0, 1, 0], [2, 2, 2, 2]] assert out.tolist() == [[0, 1, 0, 1, 0], [1, 0, 1, 0, 1], [2, 2, 2, 2, 2]]
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