fr/tmp_optifik
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add files

Browse source
This commit is contained in:
François Boulogne 2025-06-20 11:12:53 +02:00
parent 5fe2ef7a67
commit 17b53f1c4c
6 changed files with 319 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

136
tests/test_basic.py
View file

@ -1,136 +0,0 @@
import os.path
import numpy as np
from numpy.testing import assert_allclose
import pytest
from optifik.fft import thickness_from_fft
from optifik.minmax import thickness_from_minmax
from optifik.scheludko import thickness_from_scheludko
from optifik.scheludko import get_default_start_stop_wavelengths
from optifik.scheludko import thickness_for_order0
from optifik.analysis import smooth_intensities
from optifik.io import load_spectrum
def test_FFT():
FOLDER = os.path.join('tests', 'basic')
FILE_NAME = '003582.xy'
expected = 3524.51
spectrum_path = os.path.join(FOLDER, FILE_NAME)
lambdas, raw_intensities = load_spectrum(spectrum_path, lambda_min=450)
smoothed_intensities = smooth_intensities(raw_intensities)
r_index = 1.324188 + 3102.060378 / (lambdas**2)
thickness_FFT = thickness_from_fft(lambdas,
smoothed_intensities,
refractive_index=r_index)
result = thickness_FFT.thickness
assert_allclose(result, expected, rtol=1e-1)
def test_minmax_ransac():
FOLDER = os.path.join('tests', 'basic')
FILE_NAME = '000004310.xy'
expected = 1338.35
spectrum_path = os.path.join(FOLDER, FILE_NAME)
lambdas, raw_intensities = load_spectrum(spectrum_path, lambda_min=450)
smoothed_intensities = smooth_intensities(raw_intensities)
r_index = 1.324188 + 3102.060378 / (lambdas**2)
prominence = 0.02
result = thickness_from_minmax(lambdas,
smoothed_intensities,
refractive_index=r_index,
min_peak_prominence=prominence,
method='ransac',
plot=False)
assert_allclose(result.thickness, expected, rtol=1e-1)
def test_scheludko_4peaks():
FOLDER = os.path.join('tests', 'basic')
FILE_NAME = '000005253.xy'
expected = 777.07
spectrum_path = os.path.join(FOLDER, FILE_NAME)
lambdas, raw_intensities = load_spectrum(spectrum_path, lambda_min=450)
smoothed_intensities = smooth_intensities(raw_intensities)
r_index = 1.324188 + 3102.060378 / (lambdas**2)
prominence = 0.02
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=None,
plot=False)
assert_allclose(result.thickness, expected, rtol=1e-1)
def test_scheludko_2peaks():
FOLDER = os.path.join('tests', 'basic')
FILE_NAME = '000006544.xy'
expected = 495.69
spectrum_path = os.path.join(FOLDER, FILE_NAME)
lambdas, raw_intensities = load_spectrum(spectrum_path, lambda_min=450)
smoothed_intensities = smooth_intensities(raw_intensities)
r_index = 1.324188 + 3102.060378 / (lambdas**2)
prominence = 0.03
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=None,
plot=False)
assert_allclose(result.thickness, expected, rtol=1e-1)
def test_order0():
FOLDER = os.path.join('tests', 'basic')
FILE_NAME = '000018918.xy'
expected = 115.33
spectrum_path = os.path.join(FOLDER, FILE_NAME)
lambdas, raw_intensities = load_spectrum(spectrum_path, lambda_min=450)
smoothed_intensities = smooth_intensities(raw_intensities)
r_index = 1.324188 + 3102.060378 / (lambdas**2)
prominence = 0.03
result = thickness_for_order0(lambdas, smoothed_intensities,
refractive_index=r_index,
min_peak_prominence=prominence,
plot=False)
assert_allclose(result.thickness, expected, rtol=1e-1)