diff options
author | RĂ©mi Flamary <remi.flamary@gmail.com> | 2020-03-13 15:33:22 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-03-13 15:33:22 +0100 |
commit | bb15cdd36aa1ea3e24d3fe36a9c49544c407fdfe (patch) | |
tree | 19359d03211d53ed5fb3fa0b1bf0a918548b30da /test/test_bregman.py | |
parent | 18fa98fb109c935dc8d87f9c93318d8cfd118738 (diff) | |
parent | 599154c22f98eb7c0c5d3f97a6858c474b14dbdd (diff) |
Merge branch 'master' into fgw_fix
Diffstat (limited to 'test/test_bregman.py')
-rw-r--r-- | test/test_bregman.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/test/test_bregman.py b/test/test_bregman.py index f54ba9f..ec4388d 100644 --- a/test/test_bregman.py +++ b/test/test_bregman.py @@ -351,3 +351,10 @@ def test_screenkhorn(): # check marginals 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) + + +def test_convolutional_barycenter_non_square(): + # test for image with height not equal width + A = np.ones((2, 2, 3)) / (2 * 3) + b = ot.bregman.convolutional_barycenter2d(A, 1e-03) + np.testing.assert_allclose(np.ones((2, 3)) / (2 * 3), b, atol=1e-02) |