add files

tests/test_scheludko.py

@@ -0,0 +1,98 @@
+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.analysis import smooth_intensities
+from optifik.io import load_spectrum
+
+
+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
+
+
+    w_start, w_stop = None, None
+    result = thickness_from_scheludko(lambdas,
+                                      smoothed_intensities,
+                                      refractive_index=r_index,
+                                      wavelength_start=w_start,
+                                      wavelength_stop=w_stop,
+                                      interference_order=0,
+                                      plot=False)
+
+
+    assert_allclose(result.thickness, expected, rtol=1e-1)
+
+