From 08e5c0a91bc9b4afcd375109c08a14bcb0b1bd51 Mon Sep 17 00:00:00 2001 From: Vivien Seguy Date: Mon, 9 Jul 2018 16:58:00 +0900 Subject: add test free support barycenter algorithm + cleaning --- test/test_ot.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'test/test_ot.py') diff --git a/test/test_ot.py b/test/test_ot.py index dafc03f..45e777a 100644 --- a/test/test_ot.py +++ b/test/test_ot.py @@ -137,13 +137,13 @@ def test_lp_barycenter(): def test_free_support_barycenter(): - measures_locations = [np.array([-1.]).reshape((1,1)), np.array([1.]).reshape((1,1))] + measures_locations = [np.array([-1.]).reshape((1, 1)), np.array([1.]).reshape((1, 1))] measures_weights = [np.array([1.]), np.array([1.])] - X_init = np.array([-12.]).reshape((1,1)) + X_init = np.array([-12.]).reshape((1, 1)) # obvious barycenter location between two diracs - bar_locations = np.array([0.]).reshape((1,1)) + bar_locations = np.array([0.]).reshape((1, 1)) X = ot.lp.free_support_barycenter(measures_locations, measures_weights, X_init) -- cgit v1.2.3