diff options
author | Mokhtar Z. Alaya <mzalaya@Mokhtars-iMac.local> | 2020-01-18 07:45:34 +0100 |
---|---|---|
committer | Mokhtar Z. Alaya <mzalaya@Mokhtars-iMac.local> | 2020-01-18 07:45:34 +0100 |
commit | b3fb1ef40a482f0989686b79373060d764b62d38 (patch) | |
tree | d93a3641d6ef2c21eb6a70c12770669274343263 /test/test_bregman.py | |
parent | 3be0c215143e16c59ddd3be902416e91c3292937 (diff) |
clean
Diffstat (limited to 'test/test_bregman.py')
-rw-r--r-- | test/test_bregman.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/test/test_bregman.py b/test/test_bregman.py index 2398d45..e376715 100644 --- a/test/test_bregman.py +++ b/test/test_bregman.py @@ -348,7 +348,7 @@ def test_screenkhorn(): x = rng.randn(n, 2) M = ot.dist(x, x) - G_sink = ot.sinkhorn(a, b, M, 1e-03) - G_screen = ot.bregman.screenkhorn(a, b, M, 1e-03, uniform=True, verbose=True) - np.testing.assert_allclose(G_sink.sum(0), G_screen.sum(0), atol=1e-02) - np.testing.assert_allclose(G_sink.sum(1), G_screen.sum(1), atol=1e-02)
\ No newline at end of file + G_s = ot.sinkhorn(a, b, M, 1e-03) + G_sc = ot.bregman.screenkhorn(a, b, M, 1e-03, uniform=True, verbose=True) + np.testing.assert_allclose(G_s.sum(0), G_sc.sum(0), atol=1e-02) + np.testing.assert_allclose(G_s.sum(1), G_sc.sum(1), atol=1e-02)
\ No newline at end of file |