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

Extending vectorised tests to expected_new_cases

Browse source
This commit is contained in:
Nicolas Mounet 2021-05-04 05:57:21 +02:00
parent c3cf37d3da
commit 99c0e4e1f8
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -49,7 +49,9 @@ def test_exposure_model_ndarray(population, cm, expected_exposure):
)
assert isinstance(model.infection_probability(), np.ndarray)
assert isinstance(model.expected_new_cases(), np.ndarray)
assert model.infection_probability().shape == (2,)
assert model.expected_new_cases().shape == (2,)
@pytest.mark.parametrize("population", populations)
@ -63,6 +65,7 @@ def test_exposure_model_ndarray_and_float_mix(population):
)
assert isinstance(model.infection_probability(), np.ndarray)
assert isinstance(model.expected_new_cases(), np.ndarray)
@pytest.mark.parametrize("population", populations)