diff options
author | RĂ©mi Flamary <remi.flamary@gmail.com> | 2018-09-24 15:29:30 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-09-24 15:29:30 +0200 |
commit | 22d310d554239a854a5027397a5a6dff7cfe8d3c (patch) | |
tree | 0c8672255fd7c8c467af8d73a592a5bd7d3a22a9 /test | |
parent | c9b99df8fffec1dcc6802ef43b6192774817c5fb (diff) | |
parent | 1b24b1fd60a7126cd1646525ac5d7cf25f382a3a (diff) |
Merge pull request #66 from rflamary/greenkhorn
[MRG] Adding greenkhorn
Diffstat (limited to 'test')
-rw-r--r-- | test/test_bregman.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/test/test_bregman.py b/test/test_bregman.py index 01ec655..58afd7a 100644 --- a/test/test_bregman.py +++ b/test/test_bregman.py @@ -71,11 +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-5) + print(G0, G_green) def test_bary(): |