Skip to content
Snippets Groups Projects
test_variable.py 2.49 KiB
from source.variable import *


def test_variable_initialization():
    var = VariableAtmospheric.AIR_TEMPERATURE
    assert var.title == "temperature de l'air"
    assert var.modelisation is not None
    assert var.format_variable == "%.3f"
    assert isinstance(var.values, np.ndarray)
    assert isinstance(var.times, np.ndarray)


def test_variable_to_list():
    variables = VariableAtmospheric.to_list()
    assert len(variables) == 4
    assert all(isinstance(var, Variable) for var in variables)


def test_variable_frliq3d_initialization():
    var = VariableFrLiq3D.GARONNE_FLOW
    assert var.title == "débit de la garonne"
    assert var.modelisation is not None
    assert var.format_variable == "%.3f"
    assert isinstance(var.values, np.ndarray)
    assert isinstance(var.times, np.ndarray)


def test_variable_frliq3d_to_list():
    variables = VariableFrLiq3D.to_list()
    assert len(variables) == 23
    assert all(isinstance(var, Variable) for var in variables)


def test_variable_frliq2d_initialization():
    var = VariableFrLiq2D.GARONNE_FLOW
    assert var.title == "débit de la garonne"
    assert var.modelisation is not None
    assert var.format_variable == "%.3f"
    assert isinstance(var.values, np.ndarray)
    assert isinstance(var.times, np.ndarray)


def test_variable_frliq2d_to_list():
    variables = VariableFrLiq2D.to_list()
    assert len(variables) == 5
    assert all(isinstance(var, Variable) for var in variables)


def test_variable_wave_initialization():
    var = VariableWave.WAVE_HEIGHT
    assert var.title == "hauteur de la vague"
    assert var.modelisation is None
    assert var.format_variable == "%.3f"
    assert isinstance(var.values, np.ndarray)
    assert isinstance(var.times, np.ndarray)


def test_variable_wave_to_list():
    variables = VariableWave.to_list()
    assert len(variables) == 5
    assert all(isinstance(var, Variable) for var in variables)


def test_extrapolate_sediment_dordogne():
    assert VariableFrLiq3D.DORDOGNE_SEDIMENT1 in EXTRAPOLATE_SEDIMENT_DORDOGNE
    assert VariableFrLiq3D.DORDOGNE_SEDIMENT2 in EXTRAPOLATE_SEDIMENT_DORDOGNE


def test_extrapolate_sediment_garonne():
    assert VariableFrLiq3D.GARONNE_SEDIMENT1 in EXTRAPOLATE_SEDIMENT_GARONNE
    assert VariableFrLiq3D.GARONNE_SEDIMENT2 in EXTRAPOLATE_SEDIMENT_GARONNE


def test_variable_str_method():
    var = VariableAtmospheric.AIR_TEMPERATURE
    assert str(var) == "temperature de l'air"


def test_variable_int_method():
    var = VariableAtmospheric.AIR_TEMPERATURE
    assert int(var) == 0