def test_make_infections_third_calculate_incidence_less_than_susceptible_return_incidence(self):
number_of_epochs = 10
xsize = ysize = 10
r0 = 1
recovery_period = 1
latency_period = 1
xcoord = ycoord = 0
travel_rate = 0.25
migrate_infections = 5
global_population = 25
count = 0
SEIR_lambda = 1.0 / latency_period
SEIR_gamma = 1.0 / recovery_period
model = Model(number_of_epochs)
region_mock(xsize, ysize, model, r0, recovery_period, latency_period)
patch = Patch(xcoord, ycoord)
patch.new_cases_made = 3
patch.population = 10
patch.num_travel_incases = 0
patch.num_susceptible = 5
patch.make_infections_third_calculate_incidence(travel_rate, migrate_infections,
global_population, count)
self.assertNotEqual(patch.num_susceptible, patch.num_incidence)
def test_add_people_to_schedule(self):
model = Model(10)
region_mock(10, 10, model, 1, 2, 5)
number_of_people = 10
for x in range(0, number_of_people):
model.schedule.agents.add(InfectionAgent(model, x, x))
self.assertEqual(len(model.schedule.agents), 10)
def test_visible_patches_empty(self):
model = Model(10)
region = region_mock(10, 10, model, 1, 2, 5)
for x in range(region.xsize):
for y in range(region.ysize):
region.patches[x][y] = Patch(x, y)
patch = region.patches[2][2]
patch.set_visible_patches(model)
self.assertEqual(len(patch.visible_patches), 0)
def test_number_of_agents_at_patch(self):
x = 0
y = 0
patch = Patch(x, y)
model = Model(10)
region_mock(10, 10, model, 1, 2, 5)
patch.agents.append(InfectionAgent(model, x, y))
patch.agents.append(InfectionAgent(model, x, y))
self.assertEqual(len(patch.agents), 2)
