Wasserstein defined as the cost itself (do not return transportation matrix)

1 files changed, 1 insertions, 5 deletions

diff --git a/test/test_ot.py b/test/test_ot.py
index 48423e7..3c4ac11 100644
--- a/test/test_ot.py
+++ b/test/test_ot.py
@@ -98,15 +98,11 @@ def test_wass_1d():
     G, log = ot.emd([], [], M, log=True)
     wass = log["cost"]
 
-    G_1d, log = ot.wasserstein_1d(u, v, [], [], p=2., log=True)
-    wass1d = log["cost"]
+    wass1d = ot.wasserstein_1d(u, v, [], [], p=2.)
 
     # check loss is similar
     np.testing.assert_allclose(np.sqrt(wass), wass1d)
 
-    # check G is similar
-    np.testing.assert_allclose(G, G_1d)
-
 
 def test_emd_empty():
     # test emd and emd2 for simple identity