diff options
author | tlacombe <lacombe1993@gmail.com> | 2020-03-11 10:41:53 +0100 |
---|---|---|
committer | tlacombe <lacombe1993@gmail.com> | 2020-03-11 10:41:53 +0100 |
commit | a17a09a2c58bba79e897d0ba00aada05da556967 (patch) | |
tree | 826a65ba670f95aed9eca7298058ac8549286fd5 /src/python/test | |
parent | c9d6e27495c8927d736d593afb0450360b46ccc9 (diff) |
clean test_wasserstein from useless np.array
Diffstat (limited to 'src/python/test')
-rwxr-xr-x | src/python/test/test_wasserstein_distance.py | 10 |
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]]) |