Updated test

djgroen · Dec 1, 2023 · 3271919 · 3271919
1 parent e685dd1
commit 3271919
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tests/test_disease.py b/tests/test_disease.py
@@ -31,7 +31,7 @@ def test_disease_initialization(disease: Disease):
         repr(disease) == "Disease(infection_rate=0.05, "
         "incubation_period=5.0, mild_recovery_period=10.0, "
         "recovery_period=15.0, mortality_period=20.0, "
-        "period_to_hospitalisation=25.0, immunity_duration=30.0)"
+        "period_to_hospitalisation=25.0, immunity_duration=30.0, immunity_fraction=0.5)"
     )
 
 

diff --git a/tests/test_household.py b/tests/test_household.py
@@ -106,6 +106,8 @@ def test_evolve(mock_probability, sample_house, sample_ages):
 
     household = Household(sample_house, sample_ages, size=3)
     household.agents[0].status = "infectious"
+    household.agents[1].status = "susceptible"
+    household.agents[2].status = "susceptible"
     household.evolve(eco, disease)
 
     assert mock_probability.call_count == 2