summaryrefslogtreecommitdiff
path: root/src/python/test/test_wasserstein_barycenter.py
diff options
context:
space:
mode:
authorMarc Glisse <marc.glisse@inria.fr>2020-04-06 19:37:58 +0200
committerMarc Glisse <marc.glisse@inria.fr>2020-04-06 19:37:58 +0200
commitdd96965e521313b6210391f511c82cced9b2a950 (patch)
tree6f7d28b0b3dce6df1d62e1bfbf29dc424eb8bba5 /src/python/test/test_wasserstein_barycenter.py
parent689563c163c453dea8ca50b4ec6f171a61d28301 (diff)
Remove trailing whitespace
Diffstat (limited to 'src/python/test/test_wasserstein_barycenter.py')
-rwxr-xr-xsrc/python/test/test_wasserstein_barycenter.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/src/python/test/test_wasserstein_barycenter.py b/src/python/test/test_wasserstein_barycenter.py
index f686aef5..f68c748e 100755
--- a/src/python/test/test_wasserstein_barycenter.py
+++ b/src/python/test/test_wasserstein_barycenter.py
@@ -17,7 +17,7 @@ __license__ = "MIT"
def test_lagrangian_barycenter():
-
+
dg1 = np.array([[0.2, 0.5]])
dg2 = np.array([[0.2, 0.7]])
dg3 = np.array([[0.3, 0.6], [0.7, 0.8], [0.2, 0.3]])
@@ -28,12 +28,12 @@ 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]])
dg8 = np.array([[0., 4.], [4, 8]])
-
+
# error crit.
eps = 1e-7
- assert np.linalg.norm(lagrangian_barycenter(pdiagset=[dg1, dg2, dg3, dg4],init=3, verbose=False) - res) < eps
+ assert np.linalg.norm(lagrangian_barycenter(pdiagset=[dg1, dg2, dg3, dg4],init=3, verbose=False) - res) < eps
assert np.array_equal(lagrangian_barycenter(pdiagset=[dg4, dg5, dg6], verbose=False), np.empty(shape=(0,2)))
assert np.linalg.norm(lagrangian_barycenter(pdiagset=[dg7], verbose=False) - dg7) < eps
Y, log = lagrangian_barycenter(pdiagset=[dg4, dg8], verbose=True)