pingu_98/cara
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added test for expected new cases

Browse source
This commit is contained in:
Luis Aleixo 2024-05-21 11:04:25 +02:00
parent 10146f9a18
commit e61700aaad
1 changed files with 26 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
caimira/tests/apps/calculator/test_report_generator.py
View file

@ -7,7 +7,9 @@ import numpy as np
import pytest
from caimira.apps.calculator import make_app
from caimira.apps.calculator.report_generator import ReportGenerator, readable_minutes
from caimira.apps.calculator.model_generator import VirusFormData
from caimira.apps.calculator.report_generator import (ReportGenerator, readable_minutes, calculate_report_data,
manufacture_alternative_scenarios, interesting_times, comparison_report)
import caimira.apps.calculator.report_generator as rep_gen
@ -90,3 +92,26 @@ def test_interesting_times_w_temp(exposure_model_w_outside_temp_changes):
5., 5.4, 5.8, 6.2, 6.6, 7., 7.4, 7.8, 8.
]
np.testing.assert_allclose(result, expected)
def test_expected_new_cases(baseline_form_with_sr: VirusFormData):
model = baseline_form_with_sr.build_model()
executor_factory = partial(
concurrent.futures.ThreadPoolExecutor, 1,
)
# Short- and Long-range contributions
report_data = calculate_report_data(baseline_form_with_sr, model, executor_factory)
sr_lr_expected_new_cases = report_data['expected_new_cases']
sr_lr_prob_inf = report_data['prob_inf']/100
# Long-range contributions alone
scenario_sample_times = interesting_times(model)
alternative_scenarios = manufacture_alternative_scenarios(baseline_form_with_sr)
alternative_statistics = comparison_report(
baseline_form_with_sr, report_data, alternative_scenarios, scenario_sample_times, executor_factory=executor_factory,
)
lr_expected_new_cases = alternative_statistics['stats']['Base scenario without short-range interactions']['expected_new_cases']
np.testing.assert_almost_equal(sr_lr_expected_new_cases, lr_expected_new_cases + sr_lr_prob_inf * baseline_form_with_sr.short_range_total_people, 2)