import pytest

import scgenerator as sc


def test_dispersion_logic():
    params = sc.Parameters(wavelength=1550e-9, dispersion_parameter=5e-6)
    assert params.compute("beta2_coefficients") == pytest.approx([-6.3772409974749684e-27])

    params = sc.Parameters(wavelength=1550e-9, dispersion_parameter=5e-6, dispersion_slope=25)
    assert params.compute("beta2_coefficients") == pytest.approx(
        [-6.3772409974749684e-27, 5.116448086629504e-41]
    )