include peak detection in scheludko
This commit is contained in:
parent
56aac01151
commit
ccd1f2c1f0
7 changed files with 100 additions and 82 deletions
|
@ -6,7 +6,6 @@ import pytest
|
|||
from optifik.scheludko import thickness_from_scheludko
|
||||
from optifik.io import load_spectrum
|
||||
from optifik.analysis import smooth_intensities
|
||||
from optifik.analysis import finds_peak
|
||||
|
||||
import yaml
|
||||
|
||||
|
@ -22,22 +21,18 @@ def load():
|
|||
|
||||
|
||||
@pytest.mark.parametrize("spectrum_path, expected", load())
|
||||
def test_minmax(spectrum_path, expected):
|
||||
def test_SV2o5(spectrum_path, expected):
|
||||
lambdas, raw_intensities = load_spectrum(spectrum_path, lambda_min=450)
|
||||
smoothed_intensities = smooth_intensities(raw_intensities)
|
||||
|
||||
refractive_index = 1.324188 + 3102.060378 / (lambdas**2)
|
||||
prominence = 0.02
|
||||
|
||||
total_extrema, peaks_min, peaks_max = finds_peak(lambdas, smoothed_intensities,
|
||||
min_peak_prominence=prominence,
|
||||
plot=False)
|
||||
|
||||
thickness_scheludko = thickness_from_scheludko(lambdas,
|
||||
smoothed_intensities,
|
||||
peaks_min,
|
||||
peaks_max,
|
||||
refractive_index)
|
||||
thickness_scheludko = thickness_from_scheludko(lambdas, smoothed_intensities,
|
||||
refractive_index=refractive_index,
|
||||
min_peak_prominence=prominence,
|
||||
plot=False)
|
||||
result = thickness_scheludko.thickness
|
||||
|
||||
assert_allclose(result, expected, rtol=1e-1)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue