From db14caf05be8b52c06c602b00fe94daba58558fa Mon Sep 17 00:00:00 2001
From: Wolfgang Kastaun <wolfgang.kastaun@aei.mpg.de>
Date: Mon, 3 Feb 2025 17:37:14 +0100
Subject: [PATCH] Unit test: compare shift inversion using lagrange
interpolator from dsp
---
tests/test_shift_inversion_dask.py | 16 +++++++++++++++-
1 file changed, 15 insertions(+), 1 deletion(-)
diff --git a/tests/test_shift_inversion_dask.py b/tests/test_shift_inversion_dask.py
index e5fa9b5..f325683 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)
--
GitLab