diff options
author | tlacombe <lacombe1993@gmail.com> | 2021-04-12 10:47:18 +0200 |
---|---|---|
committer | tlacombe <lacombe1993@gmail.com> | 2021-04-12 10:47:18 +0200 |
commit | 01bd9eef85b0d93eb1629f1a0c5a28a359e4e7b9 (patch) | |
tree | ac1ab5e2fd59007b0840ccc724fb52a81ae7517c /src/python | |
parent | 89bb3d11064de40f2b4fda958aa2e2e8cfa5b489 (diff) |
change name _offdiag to _finite_part in test file
Diffstat (limited to 'src/python')
-rwxr-xr-x | src/python/test/test_wasserstein_distance.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/python/test/test_wasserstein_distance.py b/src/python/test/test_wasserstein_distance.py index 6701c7ba..12bf71df 100755 --- a/src/python/test/test_wasserstein_distance.py +++ b/src/python/test/test_wasserstein_distance.py @@ -9,7 +9,7 @@ - YYYY/MM Author: Description of the modification """ -from gudhi.wasserstein.wasserstein import _proj_on_diag, _offdiag, _handle_essential_parts, _get_essential_parts +from gudhi.wasserstein.wasserstein import _proj_on_diag, _finite_part, _handle_essential_parts, _get_essential_parts from gudhi.wasserstein import wasserstein_distance as pot from gudhi.hera import wasserstein_distance as hera import numpy as np @@ -28,10 +28,10 @@ def test_proj_on_diag(): assert np.array_equal(_proj_on_diag(empty), empty) -def test_offdiag(): +def test_finite_part(): diag = np.array([[0, 1], [3, 5], [2, np.inf], [3, np.inf], [-np.inf, 8], [-np.inf, 12], [-np.inf, -np.inf], [np.inf, np.inf], [-np.inf, np.inf], [-np.inf, np.inf]]) - assert np.array_equal(_offdiag(diag, enable_autodiff=False), [[0, 1], [3, 5]]) + assert np.array_equal(_finite_part(diag, enable_autodiff=False), [[0, 1], [3, 5]]) def test_handle_essential_parts(): |