import os.path
import numpy as np
from numpy.testing import assert_allclose
import pytest

from optifik.scheludko import thickness_from_scheludko
from optifik.io import load_spectrum
from optifik.analysis import Data_Smoothed
from optifik.analysis import finds_peak

import yaml


def load():
    FOLDER = os.path.join('tests', 'spectraVictor2', 'order5')

    yaml_file = os.path.join(FOLDER, 'known_value.yaml')
    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.parametrize("spectrum, expected", load())
def test_minmax(spectrum, expected):
    raw_intensities = load_spectrum(spectrum)

    smoothed_intensities, intensities, lambdas = Data_Smoothed(spectrum)

    refractive_index =  1.324188 + 3102.060378 / (lambdas**2)
    prominence = 0.02

    total_extrema, smoothed_intensities, raw_intensities, lambdas, peaks_min, peaks_max = finds_peak(spectrum,
                                                                                                     min_peak_prominence=prominence)

    thickness_scheludko = thickness_from_scheludko(lambdas,
                             raw_intensities,
                             smoothed_intensities,
                             peaks_min,
                             peaks_max,
                             refractive_index)
    result = thickness_scheludko.thickness

    assert_allclose(result, expected, rtol=1e-1)