summaryrefslogtreecommitdiff
path: root/src/python/test/test_wasserstein_barycenter.py
diff options
context:
space:
mode:
authortlacombe <lacombe1993@gmail.com>2019-12-16 18:34:55 +0100
committertlacombe <lacombe1993@gmail.com>2019-12-16 18:34:55 +0100
commit20047b94e693f31fd88ca142ba7256767ac753eb (patch)
treeb3850ef965478072a4aef1f934d9b9cc4c05c3e3 /src/python/test/test_wasserstein_barycenter.py
parent0c2fdc65cc1ea676fa8d11c24bba0d34eb5b7a3c (diff)
correction of typo in test_wasserstein_barycenter
Diffstat (limited to 'src/python/test/test_wasserstein_barycenter.py')
-rwxr-xr-xsrc/python/test/test_wasserstein_barycenter.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/python/test/test_wasserstein_barycenter.py b/src/python/test/test_wasserstein_barycenter.py
index ae3f6579..dc82a57c 100755
--- a/src/python/test/test_wasserstein_barycenter.py
+++ b/src/python/test/test_wasserstein_barycenter.py
@@ -29,5 +29,5 @@ def test_lagrangian_barycenter():
dg7 = np.array([[0.1, 0.15], [0.1, 0.7], [0.2, 0.22], [0.55, 0.84], [0.11, 0.91], [0.61, 0.75], [0.33, 0.46], [0.12, 0.41], [0.32, 0.48]])
assert np.linalg.norm(lagrangian_barycenter(pdiagset=[dg1, dg2, dg3, dg4],init=3, verbose=False) - res) < 0.001
- assert np.array_equal(lagrangian_barycenter(pdiagset=[dg4, dg5, dg6], verbose=False), shape=(0,2), np.array([]))
+ assert np.array_equal(lagrangian_barycenter(pdiagset=[dg4, dg5, dg6], verbose=False), np.array([], shape=(0,2)))
assert np.linalg.norm(lagrangian_barycenter(pdiagset=[dg7], verbose=False) - dg7) < 0.001