test: add more tolerance (#218)

azayz · web-flow · commit bdf34f341549 · 2022-03-22T20:33:13.000+01:00
diff --git a/tests/unit/math/distance/test_numpy.py b/tests/unit/math/distance/test_numpy.py
@@ -28,7 +28,7 @@
     ),
 )
 def test_cosine(x_mat, y_mat, result):
-    np.testing.assert_allclose(cosine(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(cosine(x_mat, y_mat), result, decimal=3)
 
 
 @pytest.mark.parametrize(
@@ -51,7 +51,7 @@ def test_cosine(x_mat, y_mat, result):
     ),
 )
 def test_sparse_cosine(x_mat, y_mat, result):
-    np.testing.assert_allclose(sparse_cosine(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(sparse_cosine(x_mat, y_mat), result, decimal=3)
 
 
 @pytest.mark.parametrize(
@@ -68,7 +68,7 @@ def test_sparse_cosine(x_mat, y_mat, result):
     ),
 )
 def test_sqeuclidean(x_mat, y_mat, result):
-    np.testing.assert_allclose(sqeuclidean(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(sqeuclidean(x_mat, y_mat), result, decimal=3)
 
 
 @pytest.mark.parametrize(
@@ -85,7 +85,7 @@ def test_sqeuclidean(x_mat, y_mat, result):
     ),
 )
 def test_sparse_sqeuclidean(x_mat, y_mat, result):
-    np.testing.assert_allclose(sparse_sqeuclidean(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(sparse_sqeuclidean(x_mat, y_mat), result, decimal=3)
 
 
 @pytest.mark.parametrize(
@@ -102,7 +102,7 @@ def test_sparse_sqeuclidean(x_mat, y_mat, result):
     ),
 )
 def test_euclidean(x_mat, y_mat, result):
-    np.testing.assert_allclose(euclidean(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(euclidean(x_mat, y_mat), result, decimal=3)
 
 
 @pytest.mark.parametrize(
@@ -123,4 +123,4 @@ def test_euclidean(x_mat, y_mat, result):
     ),
 )
 def test_sparse_euclidean(x_mat, y_mat, result):
-    np.testing.assert_allclose(sparse_euclidean(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(sparse_euclidean(x_mat, y_mat), result, decimal=3)
diff --git a/tests/unit/math/distance/test_paddle.py b/tests/unit/math/distance/test_paddle.py
@@ -31,7 +31,7 @@
     ),
 )
 def test_cosine(x_mat, y_mat, result):
-    np.testing.assert_allclose(cosine(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(cosine(x_mat, y_mat), result, decimal=3)
 
 
 @pytest.mark.parametrize(
@@ -60,7 +60,7 @@ def test_cosine(x_mat, y_mat, result):
     ),
 )
 def test_sqeuclidean(x_mat, y_mat, result):
-    np.testing.assert_allclose(sqeuclidean(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(sqeuclidean(x_mat, y_mat), result, decimal=3)
 
 
 @pytest.mark.parametrize(
@@ -89,4 +89,4 @@ def test_sqeuclidean(x_mat, y_mat, result):
     ),
 )
 def test_euclidean(x_mat, y_mat, result):
-    np.testing.assert_allclose(euclidean(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(euclidean(x_mat, y_mat), result, decimal=3)
diff --git a/tests/unit/math/distance/test_tensorflow.py b/tests/unit/math/distance/test_tensorflow.py
@@ -31,7 +31,7 @@
     ),
 )
 def test_cosine(x_mat, y_mat, result):
-    np.testing.assert_allclose(cosine(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(cosine(x_mat, y_mat), result, decimal=3)
 
 
 @pytest.mark.parametrize(
@@ -60,7 +60,7 @@ def test_cosine(x_mat, y_mat, result):
     ),
 )
 def test_sqeuclidean(x_mat, y_mat, result):
-    np.testing.assert_allclose(sqeuclidean(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(sqeuclidean(x_mat, y_mat), result, decimal=3)
 
 
 @pytest.mark.parametrize(
@@ -89,4 +89,4 @@ def test_sqeuclidean(x_mat, y_mat, result):
     ),
 )
 def test_euclidean(x_mat, y_mat, result):
-    np.testing.assert_allclose(euclidean(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(euclidean(x_mat, y_mat), result, decimal=3)
diff --git a/tests/unit/math/distance/test_torch.py b/tests/unit/math/distance/test_torch.py
@@ -31,7 +31,7 @@
     ),
 )
 def test_cosine(x_mat, y_mat, result):
-    np.testing.assert_allclose(cosine(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(cosine(x_mat, y_mat), result, decimal=3)
 
 
 @pytest.mark.parametrize(
@@ -55,12 +55,12 @@ def test_cosine(x_mat, y_mat, result):
         (
             torch.tensor([[1.0, 2.0, 3.0]]),
             torch.tensor([[19.0, 53.0, 201.0]]),
-            np.array([[42129]]),
+            np.array([[42128.996]]),
         ),
     ),
 )
 def test_sqeuclidean(x_mat, y_mat, result):
-    np.testing.assert_allclose(sqeuclidean(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(sqeuclidean(x_mat, y_mat), result, decimal=3)
 
 
 @pytest.mark.parametrize(
@@ -89,4 +89,4 @@ def test_sqeuclidean(x_mat, y_mat, result):
     ),
 )
 def test_euclidean(x_mat, y_mat, result):
-    np.testing.assert_allclose(euclidean(x_mat, y_mat), result, rtol=1e-5)
+    np.testing.assert_almost_equal(euclidean(x_mat, y_mat), result, decimal=3)

Original file line number	Diff line number	Diff line change
`@@ -31,7 +31,7 @@`
`31`	`31`	`),`
`32`	`32`	`)`
`33`	`33`	`def test_cosine(x_mat, y_mat, result):`
`34`		`- np.testing.assert_allclose(cosine(x_mat, y_mat), result, rtol=1e-5)`
	`34`	`+ np.testing.assert_almost_equal(cosine(x_mat, y_mat), result, decimal=3)`
`35`	`35`
`36`	`36`
`37`	`37`	`@pytest.mark.parametrize(`
`@@ -60,7 +60,7 @@ def test_cosine(x_mat, y_mat, result):`
`60`	`60`	`),`
`61`	`61`	`)`
`62`	`62`	`def test_sqeuclidean(x_mat, y_mat, result):`
`63`		`- np.testing.assert_allclose(sqeuclidean(x_mat, y_mat), result, rtol=1e-5)`
	`63`	`+ np.testing.assert_almost_equal(sqeuclidean(x_mat, y_mat), result, decimal=3)`
`64`	`64`
`65`	`65`
`66`	`66`	`@pytest.mark.parametrize(`
`@@ -89,4 +89,4 @@ def test_sqeuclidean(x_mat, y_mat, result):`
`89`	`89`	`),`
`90`	`90`	`)`
`91`	`91`	`def test_euclidean(x_mat, y_mat, result):`
`92`		`- np.testing.assert_allclose(euclidean(x_mat, y_mat), result, rtol=1e-5)`
	`92`	`+ np.testing.assert_almost_equal(euclidean(x_mat, y_mat), result, decimal=3)`