diff options
Diffstat (limited to 'src/python/test/test_betti_curve_representations.py')
| -rwxr-xr-x | src/python/test/test_betti_curve_representations.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/python/test/test_betti_curve_representations.py b/src/python/test/test_betti_curve_representations.py index 779f6d6e..1c905b82 100755 --- a/src/python/test/test_betti_curve_representations.py +++ b/src/python/test/test_betti_curve_representations.py @@ -30,8 +30,9 @@ def test_betti_curve_is_irregular_betti_curve_followed_by_interpolation(): success = success and (bettis2 == bettis).all() for i in range(0, m): - grid = np.linspace(pds[i].min(), pds[i].max() + 1, res) + grid = np.linspace(pds[i][np.isfinite(pds[i])].min(), pds[i][np.isfinite(pds[i])].max() + 1, res) bc_gridded = BettiCurve2(grid) + bc_gridded.fit([]) bettis_gridded = bc_gridded(pds[i]) interp = scipy.interpolate.interp1d(bc.grid_, bettis[i, :], kind="previous", fill_value="extrapolate") |