diff options
author | alain <alain.rakotomamonjy@gmail.com> | 2018-09-24 09:05:47 +0200 |
---|---|---|
committer | alain <alain.rakotomamonjy@gmail.com> | 2018-09-24 09:05:47 +0200 |
commit | 653fd0084c529bc74dabf93c68a9bdd5ac8f377a (patch) | |
tree | a44b24bf3deb5fd1b555d5891d52842dfee0bec9 /test | |
parent | 4367a343aeb0ceccbb99acc0f92797af020bb537 (diff) |
adding greenkhorn
Diffstat (limited to 'test')
-rw-r--r-- | test/test_bregman.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/test/test_bregman.py b/test/test_bregman.py index c8e9179..03e38bf 100644 --- a/test/test_bregman.py +++ b/test/test_bregman.py @@ -71,12 +71,14 @@ def test_sinkhorn_variants(): Ges = ot.sinkhorn( u, u, M, 1, method='sinkhorn_epsilon_scaling', stopThr=1e-10) Gerr = ot.sinkhorn(u, u, M, 1, method='do_not_exists', stopThr=1e-10) + G_green = ot.sinkhorn(u, u, M, 1, method='greenkhorn', stopThr=1e-10) # check values np.testing.assert_allclose(G0, Gs, atol=1e-05) np.testing.assert_allclose(G0, Ges, atol=1e-05) np.testing.assert_allclose(G0, Gerr) - + np.testing.assert_allclose(G0, G_green, atol = 1e-32) + print(G0,G_green) def test_bary(): |