Skip to content

Commit

Permalink
solve flake8 issues
Browse files Browse the repository at this point in the history
  • Loading branch information
HenrietteCapel committed Sep 26, 2023
1 parent f9a3693 commit 7c64f87
Showing 1 changed file with 4 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -128,11 +128,12 @@ def test_spatial_place_closure(self):
'place_closure']['closure_place_type'] = [1, 2, 3, 4, 5, 6]
result_closure_foi = SpatialInfection.spatial_foi(
self.cell, self.cell, self.infector, self.infectee, self.time)
self.assertEqual(result_susc*closure_spatial_params*result_inf*closure_spatial_params,
result_closure_foi)
self.assertEqual(result_susc * closure_spatial_params * result_inf *
closure_spatial_params, result_closure_foi)

# Place closure foi, place of only infectee closed
pe.Parameters.instance().intervention_params['place_closure']['closure_place_type'] = [1, 2, 3]
pe.Parameters.instance().intervention_params['place_closure'][
'closure_place_type'] = [1, 2, 3]
result_closure_foi = SpatialInfection.spatial_foi(
self.cell, self.cell, self.infector, self.infectee, self.time)
self.assertEqual(result_susc*result_inf*closure_spatial_params,
Expand Down

0 comments on commit 7c64f87

Please sign in to comment.