diff options
author | Alex Tong <alexanderytong@gmail.com> | 2020-03-05 12:05:16 -0500 |
---|---|---|
committer | Alex Tong <alexanderytong@gmail.com> | 2020-03-06 08:49:36 -0500 |
commit | d82e6eb1af99a982a4934d6bc019a9ab4ad5c880 (patch) | |
tree | 9b5b5bb699eac9a24f222e72c42a6e18549fd7c1 /test/test_bregman.py | |
parent | 0baf83bbff6bd0c67244b3019509fe7518fb2d75 (diff) |
Fix convolutional_barycenter kernel for non-symmetric images
Add authorship
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) |