dedebenui/scgenerator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test wl masking

Browse Source
This commit is contained in:
dedebenui
2024-03-07 09:37:13 +01:00
parent 706c35b892
commit 559c6dfddd
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_spectra.py
View File

@@ -3,6 +3,7 @@ import pytest
from scgenerator.math import abs2, wspace from scgenerator.math import abs2, wspace
from scgenerator.physics.units import m_rads from scgenerator.physics.units import m_rads
from scgenerator.physics.pulse import find_lobe_limits
from scgenerator.spectra import Spectrum from scgenerator.spectra import Spectrum
@@ -90,10 +91,10 @@ def test_energy():
def test_mask(): def test_mask():
wl = 800e-9 wl = 800e-9
bw = 50e-9 bw = 50e-9
t = np.linspace(-1e-12, 1e-12, 4096) n = 8192
t = np.linspace(-1e-12, 1e-12, n)
w = wspace(t) + m_rads(wl) w = wspace(t) + m_rads(wl)
spec = Spectrum(np.ones(4096) + 0j, w, t) spec = Spectrum(np.ones(n) + 0j, w, t)
mased = spec.mask_wl(wl, bw) mased = spec.mask_wl(wl, bw)
above = mased.wl_int / mased.wl_int.max() - 0.5 >= 0 l, r = find_lobe_limits(mased.wl_disp, mased.wl_int)[1]
diff = spec.wl_disp[above].max() - spec.wl_disp[above].min() assert r - l == pytest.approx(bw, rel=1e-2, abs=1e-10)
assert diff == bw