diff --git a/tests/test_shift_inversion_dask.py b/tests/test_shift_inversion_dask.py index e5fa9b5b4cb6c6d3a883cb80daac6c180c08c0ee..f32568331a4ad6dabf2eb8c3f33ea84b74dc7f02 100644 --- a/tests/test_shift_inversion_dask.py +++ b/tests/test_shift_inversion_dask.py @@ -4,7 +4,10 @@ import numpy as np import pytest from lisainstrument.dynamic_delay_dask import numpyfy_dask_multi -from lisainstrument.shift_inversion_dask import make_shift_inverse_lagrange_dask +from lisainstrument.shift_inversion_dask import ( + make_shift_inverse_dsp_dask, + make_shift_inverse_lagrange_dask, +) from lisainstrument.shift_inversion_numpy import make_shift_inverse_lagrange_numpy @@ -44,7 +47,18 @@ def test_shift_inversion_dask(): ) op_na = numpyfy_dask_multi(op_da, chunks) + op_dsp_da = make_shift_inverse_dsp_dask( + order=order, + fsample=fsample, + max_abs_shift=a_mod * 1.01, + max_iter=max_it, + tolerance=tol, + ) + op_dsp_na = numpyfy_dask_multi(op_dsp_da, chunks) + ai_np = op_np(dxi) ai_na = op_na(dxi) + ai_dsp_na = op_dsp_na(dxi) assert ai_np == pytest.approx(ai_na, abs=tol, rel=0) + assert ai_dsp_na == pytest.approx(ai_na, abs=tol, rel=0)