diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2018-05-31 11:26:32 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-05-31 11:26:32 +0200 |
commit | af99d3ff66062811a81454ea03e6b831a1292ae4 (patch) | |
tree | 6ade5d0ec2360bb3630635f3fdfad93b758a6323 /test/test_smooth.py | |
parent | 8d288976d398749c5261daca22ee4d772bd5b489 (diff) |
add smooth.py + first tests
Diffstat (limited to 'test/test_smooth.py')
-rw-r--r-- | test/test_smooth.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/test/test_smooth.py b/test/test_smooth.py index f951bf9..4ca44f8 100644 --- a/test/test_smooth.py +++ b/test/test_smooth.py @@ -30,4 +30,8 @@ def test_smooth_ot_dual(): u, G.sum(1), atol=1e-05) # cf convergence sinkhorn np.testing.assert_allclose( u, G.sum(0), atol=1e-05) # cf convergence sinkhorn + + + G2 = ot.sinkhorn(u, u, M, 1, stopThr=1e-10) + np.testing.assert_allclose( G, G2 , atol=1e-05)
\ No newline at end of file |