diff options
Diffstat (limited to 'test/test_smooth.py')
-rw-r--r-- | test/test_smooth.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/test/test_smooth.py b/test/test_smooth.py index 2afa4f8..31e0b2e 100644 --- a/test/test_smooth.py +++ b/test/test_smooth.py @@ -25,16 +25,16 @@ def test_smooth_ot_dual(): Gl2, log = ot.smooth.smooth_ot_dual(u, u, M, 1, reg_type='l2', log=True, stopThr=1e-10) - # check constratints + # check constraints np.testing.assert_allclose( u, Gl2.sum(1), atol=1e-05) # cf convergence sinkhorn np.testing.assert_allclose( u, Gl2.sum(0), atol=1e-05) # cf convergence sinkhorn - # kl regyularisation + # kl regularisation G = ot.smooth.smooth_ot_dual(u, u, M, 1, reg_type='kl', stopThr=1e-10) - # check constratints + # check constraints np.testing.assert_allclose( u, G.sum(1), atol=1e-05) # cf convergence sinkhorn np.testing.assert_allclose( @@ -60,16 +60,16 @@ def test_smooth_ot_semi_dual(): Gl2, log = ot.smooth.smooth_ot_semi_dual(u, u, M, 1, reg_type='l2', log=True, stopThr=1e-10) - # check constratints + # check constraints np.testing.assert_allclose( u, Gl2.sum(1), atol=1e-05) # cf convergence sinkhorn np.testing.assert_allclose( u, Gl2.sum(0), atol=1e-05) # cf convergence sinkhorn - # kl regyularisation + # kl regularisation G = ot.smooth.smooth_ot_semi_dual(u, u, M, 1, reg_type='kl', stopThr=1e-10) - # check constratints + # check constraints np.testing.assert_allclose( u, G.sum(1), atol=1e-05) # cf convergence sinkhorn np.testing.assert_allclose( |