diff options
| author | tlacombe <lacombe1993@gmail.com> | 2020-02-14 14:58:53 +0100 |
|---|---|---|
| committer | tlacombe <lacombe1993@gmail.com> | 2020-02-14 14:58:53 +0100 |
| commit | dc5c7ac2167bfa467b52d0a36ecb9999fe03ba91 (patch) | |
| tree | ed3788bcef8ad4946e3f91acd93d3ebb6f13f0d0 /src/python/test/test_wasserstein_barycenter.py | |
| parent | dc4442bc402ac25290eb529b57407607434bb7ae (diff) | |
added two more tests for barycenter
Diffstat (limited to 'src/python/test/test_wasserstein_barycenter.py')
| -rwxr-xr-x | src/python/test/test_wasserstein_barycenter.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/python/test/test_wasserstein_barycenter.py b/src/python/test/test_wasserstein_barycenter.py index 07242582..a58a4d62 100755 --- a/src/python/test/test_wasserstein_barycenter.py +++ b/src/python/test/test_wasserstein_barycenter.py @@ -41,4 +41,5 @@ def test_lagrangian_barycenter(): assert np.abs(log["energy"] - 2) < eps assert np.array_equal(log["groupings"][0] , np.array([[0, -1]])) assert np.array_equal(log["groupings"][1] , np.array([[0, 0]])) + assert np.linalg.norm(lagrangian_barycenter(pdiagset=[dg8, dg4], init=np.array([[0.2, 0.6], [0.5, 0.7]]), verbose=False) - np.array([[1, 3]])) < eps assert lagrangian_barycenter(pdiagset = []) is None |