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

single test for infectious_dose_vectorisation fix #184

Browse source
This commit is contained in:
Luis Aleixo 2021-08-04 11:43:58 +02:00
parent 160dcf6a60
commit 415ade0185
1 changed files with 2 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
cara/tests/models/test_exposure_model.py
View file

@ -168,17 +168,7 @@ def test_exposure_model_integral_accuracy(exposed_time_interval,
model = ExposureModel(conc_model, population, fraction_deposited=1.)
np.testing.assert_allclose(model.exposure(), expected_exposure)
@pytest.mark.parametrize(
"override_params", [
{'infectious_dose': np.array([50, 20, 30])},
]
)
def test_infectious_dose_vectorisation(override_params):
defaults = {
'infectious_dose': 50
}
defaults.update(override_params)
def test_infectious_dose_vectorisation():
infPopulation = models.InfectedPopulation(
number=1,
presence=halftime,
@ -186,7 +176,7 @@ def test_infectious_dose_vectorisation(override_params):
activity=models.Activity.types['Standing'],
virus = models.SARSCoV2(
viral_load_in_sputum=1e9,
infectious_dose=defaults['infectious_dose'],
infectious_dose=np.array([50, 20, 30]),
),
expiration=models.Expiration.types['Talking']
)