summaryrefslogtreecommitdiff
path: root/test/test_gromov.py
diff options
context:
space:
mode:
authortvayer <titouan.vayer@gmail.com>2019-05-29 17:52:20 +0200
committertvayer <titouan.vayer@gmail.com>2019-05-29 17:52:20 +0200
commit9bb7d40b563f42bf2875efca860bf0c579307161 (patch)
tree8f39422c461703b37e39057d80e870c5a51a0065 /test/test_gromov.py
parent63093cef7af3350228251aa930872c6f30789432 (diff)
.0
Diffstat (limited to 'test/test_gromov.py')
-rw-r--r--test/test_gromov.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_gromov.py b/test/test_gromov.py
index d7a12f3..b7ede95 100644
--- a/test/test_gromov.py
+++ b/test/test_gromov.py
@@ -39,7 +39,7 @@ def test_gromov():
np.testing.assert_allclose(
q, G.sum(0), atol=1e-04) # cf convergence gromov
- Id = (1 / float(n_samples)) * np.eye(n_samples, n_samples)
+ Id = (1 / 1.0*n_samples) * np.eye(n_samples, n_samples)
np.testing.assert_allclose(
G, np.flipud(Id), atol=1e-04)
@@ -193,7 +193,7 @@ def test_fgw():
np.testing.assert_allclose(
q, G.sum(0), atol=1e-04) # cf convergence fgw
- Id = (1 / float(n_samples)) * np.eye(n_samples, n_samples)
+ Id = (1 / 1.0*n_samples) * np.eye(n_samples, n_samples)
np.testing.assert_allclose(
G, np.flipud(Id), atol=1e-04) # cf convergence gromov