From 7396b37c6f41c2aa98e40a71b185baa29c3dd2fb Mon Sep 17 00:00:00 2001 From: Matthew Ghosh Date: Wed, 31 Jan 2024 10:31:25 +0000 Subject: [PATCH] Flake8 --- pyEpiabm/pyEpiabm/tests/test_func/test_sim_functional.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pyEpiabm/pyEpiabm/tests/test_func/test_sim_functional.py b/pyEpiabm/pyEpiabm/tests/test_func/test_sim_functional.py index d1fe477ab..544d6ca3c 100644 --- 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