clean test_wasserstein from useless np.array

1 files changed, 5 insertions, 5 deletions

diff --git a/src/python/test/test_wasserstein_distance.py b/src/python/test/test_wasserstein_distance.py
index f92208c0..0d70e11a 100755
--- a/src/python/test/test_wasserstein_distance.py
+++ b/src/python/test/test_wasserstein_distance.py
@@ -61,15 +61,15 @@ def _basic_wasserstein(wasserstein_distance, delta, test_infinity=True, test_mat
 
     if test_matching:
         match = wasserstein_distance(emptydiag, emptydiag, matching=True, internal_p=1., order=2)[1]
-        assert np.array_equal(match, np.array([]))
+        assert np.array_equal(match, [])
         match = wasserstein_distance(emptydiag, emptydiag, matching=True, internal_p=np.inf, order=2.24)[1]
-        assert np.array_equal(match, np.array([]))
+        assert np.array_equal(match, [])
         match = wasserstein_distance(emptydiag, diag2, matching=True, internal_p=np.inf, order=2.)[1]
-        assert np.array_equal(match , np.array([[-1, 0], [-1, 1]]))
+        assert np.array_equal(match , [[-1, 0], [-1, 1]])
         match = wasserstein_distance(diag2, emptydiag, matching=True, internal_p=np.inf, order=2.24)[1]
-        assert np.array_equal(match , np.array([[0, -1], [1, -1]]))
+        assert np.array_equal(match , [[0, -1], [1, -1]])
         match = wasserstein_distance(diag1, diag2, matching=True, internal_p=2., order=2.)[1]
-        assert np.array_equal(match, np.array([[0, 0], [1, 1], [2, -1]]))
+        assert np.array_equal(match, [[0, 0], [1, 1], [2, -1]])