84 lines
3.5 KiB
Python
84 lines
3.5 KiB
Python
import os.path
|
|
import numpy as np
|
|
from numpy.testing import assert_allclose
|
|
import pytest
|
|
|
|
from optifik.scheludko import thickness_from_scheludko
|
|
from optifik.scheludko import get_default_start_stop_wavelengths
|
|
from optifik.io import load_spectrum
|
|
from optifik.analysis import smooth_intensities
|
|
|
|
import yaml
|
|
|
|
|
|
def load(filename):
|
|
FOLDER = os.path.join('tests', 'spectraVictor2', 'order0')
|
|
|
|
yaml_file = os.path.join(FOLDER, filename)
|
|
with open(yaml_file, "r") as yaml_file:
|
|
thickness_dict = yaml.safe_load(yaml_file)
|
|
data = [(os.path.join(FOLDER, fn), val) for fn, val in thickness_dict.items()]
|
|
return data
|
|
|
|
|
|
#@pytest.mark.skip('...')
|
|
@pytest.mark.parametrize("spectrum_path, expected", load('known_value.yaml'))
|
|
def test_SV2o0_small_tol(spectrum_path, expected):
|
|
lambdas, raw_intensities = load_spectrum(spectrum_path, wavelength_min=450)
|
|
|
|
File_I_min = 'tests/spectre_trou/000043641.xy'
|
|
_, intensities_void = load_spectrum(File_I_min, wavelength_min=450)
|
|
|
|
smoothed_intensities = smooth_intensities(raw_intensities)
|
|
|
|
r_index = 1.324188 + 3102.060378 / (lambdas**2)
|
|
prominence = 0.020
|
|
|
|
w_start, w_stop = None, None
|
|
#w_start, w_stop = get_default_start_stop_wavelengths(lambdas,
|
|
# smoothed_intensities,
|
|
# refractive_index=r_index,
|
|
# min_peak_prominence=prominence,
|
|
# plot=False)
|
|
|
|
result = thickness_from_scheludko(lambdas,
|
|
smoothed_intensities,
|
|
refractive_index=r_index,
|
|
wavelength_start=w_start,
|
|
wavelength_stop=w_stop,
|
|
interference_order=0,
|
|
intensities_void=intensities_void,
|
|
plot=False)
|
|
|
|
assert_allclose(result.thickness, expected, rtol=1e-1)
|
|
|
|
@pytest.mark.parametrize("spectrum_path, expected", load('known_value_large_tol.yaml'))
|
|
def test_SV2o0_large_tol(spectrum_path, expected):
|
|
lambdas, raw_intensities = load_spectrum(spectrum_path, wavelength_min=450)
|
|
|
|
File_I_min = 'tests/spectre_trou/000043641.xy'
|
|
_, intensities_void = load_spectrum(File_I_min, wavelength_min=450)
|
|
|
|
smoothed_intensities = smooth_intensities(raw_intensities)
|
|
|
|
r_index = 1.324188 + 3102.060378 / (lambdas**2)
|
|
prominence = 0.020
|
|
|
|
w_start, w_stop = None, None
|
|
# w_start, w_stop = get_default_start_stop_wavelengths(lambdas,
|
|
# smoothed_intensities,
|
|
# refractive_index=r_index,
|
|
# min_peak_prominence=prominence,
|
|
# plot=False)
|
|
|
|
result = thickness_from_scheludko(lambdas,
|
|
smoothed_intensities,
|
|
refractive_index=r_index,
|
|
wavelength_start=w_start,
|
|
wavelength_stop=w_stop,
|
|
interference_order=0,
|
|
intensities_void=intensities_void,
|
|
plot=False)
|
|
|
|
assert_allclose(result.thickness, expected, rtol=2.5e-1)
|
|
|