Merge pull request #465 from NickleDave/fix-test-functional-fixes-#464

Fix `test_functional`, fixes #464
vocalpy · Mar 14, 2022 · 4599740 · 4599740
2 parents f340542 + 9c04459
commit 4599740
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_nn/test_functional.py b/tests/test_nn/test_functional.py
@@ -17,7 +17,7 @@ def test_onehot():
     # convert labels to one hot tensor
     one_hot = F.one_hot(labels, num_classes)
 
-    assert pytest.approx(one_hot[0, labels[0, 0, 0], 0, 0].item(), 1.0)
-    assert pytest.approx(one_hot[0, labels[0, 1, 0], 1, 0].item(), 1.0)
-    assert pytest.approx(one_hot[1, labels[1, 0, 0], 0, 0].item(), 1.0)
-    assert pytest.approx(one_hot[1, labels[1, 1, 0], 1, 0].item(), 1.0)
+    assert 1.0 == pytest.approx(one_hot[0, labels[0, 0, 0], 0, 0].item())
+    assert 1.0 == pytest.approx(one_hot[0, labels[0, 1, 0], 1, 0].item())
+    assert 1.0 == pytest.approx(one_hot[1, labels[1, 0, 0], 0, 0].item())
+    assert 1.0 == pytest.approx(one_hot[1, labels[1, 1, 0], 1, 0].item())