diff options
Diffstat (limited to 'test/test_ot.py')
-rw-r--r-- | test/test_ot.py | 34 |
1 files changed, 18 insertions, 16 deletions
diff --git a/test/test_ot.py b/test/test_ot.py index b69d080..9103ac8 100644 --- a/test/test_ot.py +++ b/test/test_ot.py @@ -36,7 +36,8 @@ def test_emd_emd2(): # check loss=0 assert np.allclose(w, 0) - + + def test_emd2_multi(): from ot.datasets import get_1D_gauss as gauss @@ -72,11 +73,11 @@ def test_emd2_multi(): emdn = ot.emd2(a, b, M) ot.toc('multi proc : {} s') - assert np.allclose(emd1, emdn) - - + assert np.allclose(emd1, emdn) + + def test_sinkhorn(): - # test sinkhorn + # test sinkhorn n = 100 np.random.seed(0) @@ -85,14 +86,15 @@ def test_sinkhorn(): M = ot.dist(x, x) - G = ot.sinkhorn(u, u, M,1,stopThr=1e-10) + G = ot.sinkhorn(u, u, M, 1, stopThr=1e-10) # 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 - + 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 + + def test_sinkhorn_variants(): - # test sinkhorn + # test sinkhorn n = 100 np.random.seed(0) @@ -101,11 +103,11 @@ def test_sinkhorn_variants(): M = ot.dist(x, x) - G0 = ot.sinkhorn(u, u, M,1, method='sinkhorn',stopThr=1e-10) - Gs = ot.sinkhorn(u, u, M,1, method='sinkhorn_stabilized',stopThr=1e-10) - Ges = ot.sinkhorn(u, u, M,1, method='sinkhorn_epsilon_scaling',stopThr=1e-10) + G0 = ot.sinkhorn(u, u, M, 1, method='sinkhorn', stopThr=1e-10) + Gs = ot.sinkhorn(u, u, M, 1, method='sinkhorn_stabilized', stopThr=1e-10) + Ges = ot.sinkhorn( + u, u, M, 1, method='sinkhorn_epsilon_scaling', stopThr=1e-10) # check constratints - assert np.allclose(G0, Gs, atol=1e-05) - assert np.allclose(G0, Ges, atol=1e-05) # - + assert np.allclose(G0, Gs, atol=1e-05) + assert np.allclose(G0, Ges, atol=1e-05) |