bugfix

d3857770 · rusty1s · 02e6d36a · d3857770 · d3857770
Commit d3857770 authored Dec 16, 2017 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

test/test_add.py test/test_add.py +1 -1

test/test_mean.py test/test_mean.py +1 -1

No files found.
--- a/test/test_add.py
+++ b/test/test_add.py
@@ -19,7 +19,7 @@ def test_scatter_add(str):
    assert output.tolist() == expected_output
    output = scatter_add(index, input, dim=1)
-    assert output.tolist(), expected_output
+    assert output.tolist() == expected_output
    output = Variable(output).fill_(0)
    index = Variable(index)

--- a/test/test_mean.py
+++ b/test/test_mean.py
@@ -19,7 +19,7 @@ def test_scatter_mean(str):
    assert output.tolist() == expected_output
    output = scatter_mean(index, input, dim=1)
-    assert output.tolist(), expected_output
+    assert output.tolist() == expected_output
    output = Variable(output).fill_(0)
    index = Variable(index)