Skip to content
Snippets Groups Projects
Commit e8d9ebb6 authored by sandcha's avatar sandcha
Browse files

Initie des tests unitaires de calcul des strates

parent c4e70c2a
Branches
No related tags found
1 merge request!21Ajoute la prod https aux Origins et s'assure que les montants des tendances sont des nombre à virgule
from typing import List
import pytest
from leximpact_dotations_back.dotations_types import Dotation
from leximpact_dotations_back.main_types import StrateImpact
from leximpact_dotations_back.preload import configuration, MODEL_OFDL_BASE
from leximpact_dotations_back.computing.calculate_impact_strates import format_strates_impact, init_strates_demographiques
from leximpact_dotations_back.computing.dotations_simulation import DotationsSimulation
from leximpact_dotations_back.computing.strate_demographique import StrateDemographique
@pytest.fixture
def dotation_simulation() -> DotationsSimulation:
dotations_simulation_base_2024 = DotationsSimulation(
data_directory=configuration['data_directory'],
model=MODEL_OFDL_BASE,
annee=2024
)
return dotations_simulation_base_2024
@pytest.fixture
def strates_demographiques(dotation_simulation) -> List[StrateDemographique]:
strates_year_period = dotation_simulation.model.parameters(2024).population.groupes_demographiques
strates_demographiques_2024_base = init_strates_demographiques(dotation_simulation, strates_year_period)
return strates_demographiques_2024_base
def test_format_strates_impact(dotation_simulation, strates_demographiques):
strates_impact: List[StrateImpact] = format_strates_impact(dotation_simulation, strates_demographiques)
strate_testee = strates_impact[0]
assert strate_testee is not None
assert strate_testee["tendance"] is None
assert strate_testee["dotationsImpacts"] is not None
# toutes les dotations connues de l'application ont la réforme calculée
assert len(strate_testee["dotationsImpacts"]) == len(Dotation), Dotation
for dotationImpact in strate_testee["dotationsImpacts"]:
assert dotationImpact["dotation"] in Dotation
assert dotationImpact["dotationMoyenneParHabitant"] is not None
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment