add test free support barycenter algorithm + cleaning

Author: Vivien Seguy

examples/plot_free_support_barycenter.py

@@ -48,9 +48,9 @@
 # Compute free support barycenter
 # -------------
 
-k = 10 # number of Diracs of the barycenter
-X_init = np.random.normal(0., 1., (k, d)) # initial Dirac locations
-b = np.ones((k,)) / k # weights of the barycenter (it will not be optimized, only the locations are optimized)
+k = 10  # number of Diracs of the barycenter
+X_init = np.random.normal(0., 1., (k, d))  # initial Dirac locations
+b = np.ones((k,)) / k  # weights of the barycenter (it will not be optimized, only the locations are optimized)
 
 X = ot.lp.free_support_barycenter(measures_locations, measures_weights, X_init, 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)