Flake8

SABS-R3-Epidemiology · Jan 31, 2024 · 7396b37 · 7396b37
1 parent d8100f9
commit 7396b37
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pyEpiabm/pyEpiabm/tests/test_func/test_sim_functional.py b/pyEpiabm/pyEpiabm/tests/test_func/test_sim_functional.py
@@ -173,11 +173,12 @@ def test_waning_compartments(self, *mocks):
 
             for cell in pop.cells:
                 cell_data = cell.compartment_counter.retrieve()
-                for status in [InfectionStatus.Recovered, InfectionStatus.Dead]:
+                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.pop_params["population_size"])
 
     def test_no_infection(self, *mocks):
         """Basic functional test to ensure noone is infected when there are