summaryrefslogtreecommitdiff
path: root/src/python/test/test_wasserstein_distance.py
diff options
context:
space:
mode:
authortlacombe <lacombe1993@gmail.com>2020-03-10 18:08:15 +0100
committertlacombe <lacombe1993@gmail.com>2020-03-10 18:08:15 +0100
commit6c369a6aa566dfcb8cdb501d0c39eafb32219669 (patch)
tree2947dbd4025ec4600c5b2cf88fd23d1a3a3630a8 /src/python/test/test_wasserstein_distance.py
parentfc4e10863d103ee6bc22863f48548fe246a3ddd6 (diff)
fix typo in test_wasserstein_distance
Diffstat (limited to 'src/python/test/test_wasserstein_distance.py')
-rwxr-xr-xsrc/python/test/test_wasserstein_distance.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/python/test/test_wasserstein_distance.py b/src/python/test/test_wasserstein_distance.py
index ca9a4a61..f92208c0 100755
--- a/src/python/test/test_wasserstein_distance.py
+++ b/src/python/test/test_wasserstein_distance.py
@@ -69,7 +69,7 @@ def _basic_wasserstein(wasserstein_distance, delta, test_infinity=True, test_mat
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]]))
match = wasserstein_distance(diag1, diag2, matching=True, internal_p=2., order=2.)[1]
- assert np.array_equal(match, np.array_equal([[0, 0], [1, 1], [2, -1]]))
+ assert np.array_equal(match, np.array([[0, 0], [1, 1], [2, -1]]))