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

Handled model generation in test_full_algorithm.py test

Browse source
This commit is contained in:
Luis Aleixo 2022-06-03 09:19:48 +02:00
parent 8601e51adb
commit 0f373a5f3c
1 changed files with 20 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
cara/tests/test_full_algorithm.py
View file

@ -477,7 +477,7 @@ def c_model() -> mc.ConcentrationModel:
host_immunity=0.,
),
evaporation_factor=0.3,
).build_model(SAMPLE_SIZE)
)
@pytest.fixture
@ -496,7 +496,7 @@ def c_model_distr() -> mc.ConcentrationModel:
host_immunity=0.,
),
evaporation_factor=0.3,
).build_model(SAMPLE_SIZE)
)
@pytest.fixture
@ -507,13 +507,13 @@ def sr_models() -> typing.Tuple[mc.ShortRangeModel, ...]:
activity = models.Activity.types['Seated'],
presence = interaction_intervals[0],
distance = 0.854,
).build_model(SAMPLE_SIZE),
),
mc.ShortRangeModel(
expiration = short_range_expiration_distributions['Breathing'],
activity = models.Activity.types['Heavy exercise'],
presence = interaction_intervals[1],
distance = 0.854,
).build_model(SAMPLE_SIZE),
),
)
@ -559,7 +559,7 @@ def expo_sr_model(c_model,sr_models) -> mc.ExposureModel:
activity=models.Activity.types['Seated'],
host_immunity=0.,
),
).build_model(SAMPLE_SIZE)
)
@pytest.fixture
@ -589,13 +589,13 @@ def expo_sr_model_distr(c_model_distr) -> mc.ExposureModel:
activity = activity_distributions['Seated'],
presence = interaction_intervals[0],
distance = short_range_distances,
).build_model(SAMPLE_SIZE),
),
mc.ShortRangeModel(
expiration = short_range_expiration_distributions['Speaking'],
activity = activity_distributions['Seated'],
presence = interaction_intervals[1],
distance = short_range_distances,
).build_model(SAMPLE_SIZE),
),
),
exposed=mc.Population(
number=1,
@ -604,11 +604,11 @@ def expo_sr_model_distr(c_model_distr) -> mc.ExposureModel:
activity=models.Activity.types['Seated'],
host_immunity=0.,
),
).build_model(SAMPLE_SIZE)
)
@pytest.fixture
def simple_expo_sr_model_distr(c_model_distr) -> SimpleExposureModel:
def simple_expo_sr_model_distr() -> SimpleExposureModel:
return SimpleExposureModel(
infected_presence = presence,
viral_load = virus_distributions['SARS_CoV_2_DELTA'
@ -649,7 +649,7 @@ def simple_expo_sr_model_distr(c_model_distr) -> SimpleExposureModel:
)
def test_longrange_concentration(time,c_model,simple_c_model):
npt.assert_allclose(
c_model.concentration(time).mean(),
c_model.build_model(SAMPLE_SIZE).concentration(time).mean(),
simple_c_model.concentration(time), rtol=TOLERANCE
)
@ -661,7 +661,7 @@ def test_longrange_concentration(time,c_model,simple_c_model):
def test_shortrange_concentration(time,c_model,simple_c_model,
sr_models,simple_sr_models):
result_sr_model = np.sum([np.array(
sr_mod.short_range_concentration(c_model,time)).mean()
sr_mod.build_model(SAMPLE_SIZE).short_range_concentration(c_model.build_model(SAMPLE_SIZE),time)).mean()
for sr_mod in sr_models])
result_simple_sr_model = np.sum([np.array(
sr_mod.concentration(simple_c_model,time)).mean()
@ -686,7 +686,7 @@ def test_longrange_exposure(c_model):
sr_models = (),
)
expo_model = mc.ExposureModel(
concentration_model=c_model,
concentration_model=c_model.build_model(SAMPLE_SIZE),
short_range=(),
exposed=mc.Population(
number=1,
@ -721,7 +721,7 @@ def test_longrange_concentration_with_distributions(c_model_distr,time):
BLO_factors = expiration_BLO_factors['Breathing'],
)
npt.assert_allclose(
c_model_distr.concentration(time).mean(),
c_model_distr.build_model(SAMPLE_SIZE).concentration(time).mean(),
simple_expo_model.concentration(time).mean(), rtol=TOLERANCE
)
@ -746,7 +746,7 @@ def test_longrange_exposure_with_distributions(c_model_distr):
sr_models = (),
)
expo_model = mc.ExposureModel(
concentration_model=c_model_distr,
concentration_model=c_model_distr.build_model(SAMPLE_SIZE),
short_range=(),
exposed=mc.Population(
number=1,
@ -775,18 +775,18 @@ def test_longrange_exposure_with_distributions(c_model_distr):
)
def test_concentration_with_shortrange(expo_sr_model,simple_expo_sr_model,time):
npt.assert_allclose(
expo_sr_model.concentration(time).mean(),
expo_sr_model.build_model(SAMPLE_SIZE).concentration(time).mean(),
simple_expo_sr_model.total_concentration(time).mean(), rtol=TOLERANCE
)
def test_exposure_with_shortrange(expo_sr_model,simple_expo_sr_model):
npt.assert_allclose(
expo_sr_model.deposited_exposure().mean(),
expo_sr_model.build_model(SAMPLE_SIZE).deposited_exposure().mean(),
simple_expo_sr_model.dose().mean(), rtol=TOLERANCE
)
npt.assert_allclose(
expo_sr_model.infection_probability().mean(),
expo_sr_model.build_model(SAMPLE_SIZE).infection_probability().mean(),
simple_expo_sr_model.probability_infection().mean(), rtol=TOLERANCE
)
@ -798,7 +798,7 @@ def test_exposure_with_shortrange(expo_sr_model,simple_expo_sr_model):
def test_concentration_with_shortrange_and_distributions(
expo_sr_model_distr,simple_expo_sr_model_distr,time):
npt.assert_allclose(
expo_sr_model_distr.concentration(time).mean(),
expo_sr_model_distr.build_model(SAMPLE_SIZE).concentration(time).mean(),
simple_expo_sr_model_distr.total_concentration(time).mean(),
rtol=TOLERANCE
)
@ -807,11 +807,11 @@ def test_concentration_with_shortrange_and_distributions(
def test_exposure_with_shortrange_and_distributions(expo_sr_model_distr,
simple_expo_sr_model_distr):
npt.assert_allclose(
expo_sr_model_distr.deposited_exposure().mean(),
expo_sr_model_distr.build_model(SAMPLE_SIZE).deposited_exposure().mean(),
simple_expo_sr_model_distr.dose().mean(), rtol=0.05
)
npt.assert_allclose(
expo_sr_model_distr.infection_probability().mean(),
expo_sr_model_distr.build_model(SAMPLE_SIZE).infection_probability().mean(),
simple_expo_sr_model_distr.probability_infection().mean(),
rtol=0.03
)