mdevaluate/test/test_utils.py
2024-01-10 12:41:18 +01:00

34 lines
921 B
Python

from copy import copy
import pytest
import numpy as np
from mdevaluate import utils
@pytest.fixture
def logdata(request):
xdata = np.logspace(-1, 3, 50)
ydata = np.exp(- (xdata)**0.7)
return xdata, ydata
def test_filon_fourier_transformation(logdata):
xdata, ydata = logdata
xdata_zero = copy(xdata)
xdata_zero[0] = 0
_, filon = utils.filon_fourier_transformation(xdata_zero, ydata)
assert not np.isnan(filon).any(), 'There are NaN values in the filon result!'
freqs = np.logspace(-4, 1)
filon_freqs, filon_imag = utils.filon_fourier_transformation(
xdata, xdata, frequencies=freqs, derivative='linear', imag=True
)
assert (freqs == filon_freqs).all()
freqs, filon_real = utils.filon_fourier_transformation(
xdata, xdata, frequencies=freqs, derivative='linear', imag=False
)
assert np.isclose(filon_imag.real, filon_real).all()