12 lines
593 B
Python
12 lines
593 B
Python
import caimira.models
|
|
from caimira.dataclass_utils import nested_replace
|
|
|
|
|
|
def test_exposure_r0(baseline_exposure_model):
|
|
baseline_n3 = nested_replace(
|
|
baseline_exposure_model, {'concentration_model.infected.number': 3}
|
|
)
|
|
# The number of new cases should be greater if there are more infecteds, but
|
|
# the reproduction number should be the same (it is a measure of one infected case).
|
|
assert baseline_n3.expected_new_cases() > baseline_exposure_model.expected_new_cases()
|
|
assert baseline_n3.reproduction_number() == baseline_exposure_model.reproduction_number()
|