Merge

SABS-R3-Epidemiology · Jan 31, 2024 · d8100f9 · d8100f9
1 parent 27b8ea8
commit d8100f9
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/pyEpiabm/pyEpiabm/tests/test_func/test_sim_functional.py b/pyEpiabm/pyEpiabm/tests/test_func/test_sim_functional.py
@@ -168,16 +168,16 @@ def test_waning_compartments(self, *mocks):
                                                    self.sim_params,
                                                    self.file_params)
 
-        # Test not all individuals have entered Recovered or Dead at end
-        recov_dead_state_count = 0
+            # Test not all individuals have entered Recovered or Dead at end
+            recov_dead_state_count = 0
 
-        for cell in pop.cells:
-            cell_data = cell.compartment_counter.retrieve()
-            for status in [InfectionStatus.Recovered, InfectionStatus.Dead]:
-                recov_dead_state_count += cell_data[status]
+            for cell in pop.cells:
+                cell_data = cell.compartment_counter.retrieve()
+                for status in [InfectionStatus.Recovered, InfectionStatus.Dead]:
+                    recov_dead_state_count += cell_data[status]
 
-        self.assertNotEqual(np.sum(recov_dead_state_count),
-                         self.pop_params["population_size"])
+            self.assertNotEqual(np.sum(recov_dead_state_count),
+                             self.pop_params["population_size"])
 
     def test_no_infection(self, *mocks):
         """Basic functional test to ensure noone is infected when there are