diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2017-07-24 15:23:14 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2017-07-24 15:23:14 +0200 |
commit | 83ecc6df836d1a6b05bd641dfef465cc02b25b8f (patch) | |
tree | b3db628c53c16ec0b1efc3b2a9b0e13fe0f96de2 /test | |
parent | 709d8cbc9f9961a5175eb64ae497b854e0b9b184 (diff) |
bregman coverage
Diffstat (limited to 'test')
-rw-r--r-- | test/test_bregman.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/test/test_bregman.py b/test/test_bregman.py index b65de11..78666c7 100644 --- a/test/test_bregman.py +++ b/test/test_bregman.py @@ -33,18 +33,20 @@ def test_sinkhorn_empty(): M = ot.dist(x, x) - G = ot.sinkhorn([], [], M, 1, stopThr=1e-10) + G, log = ot.sinkhorn([], [], M, 1, stopThr=1e-10, verbose=True, log=True) # check constratints assert np.allclose(u, G.sum(1), atol=1e-05) # cf convergence sinkhorn assert np.allclose(u, G.sum(0), atol=1e-05) # cf convergence sinkhorn - G = ot.sinkhorn([], [], M, 1, stopThr=1e-10, method='sinkhorn_stabilized') + G, log = ot.sinkhorn([], [], M, 1, stopThr=1e-10, + method='sinkhorn_stabilized', verbose=True, log=True) # check constratints assert np.allclose(u, G.sum(1), atol=1e-05) # cf convergence sinkhorn assert np.allclose(u, G.sum(0), atol=1e-05) # cf convergence sinkhorn - G = ot.sinkhorn( - [], [], M, 1, stopThr=1e-10, method='sinkhorn_epsilon_scaling') + G, log = ot.sinkhorn( + [], [], M, 1, stopThr=1e-10, method='sinkhorn_epsilon_scaling', + verbose=True, log=True) # check constratints assert np.allclose(u, G.sum(1), atol=1e-05) # cf convergence sinkhorn assert np.allclose(u, G.sum(0), atol=1e-05) # cf convergence sinkhorn |