def test_clear(self):
state = Apocalypse(5, 5)
state.add_human(2, 3)
state.add_human(1, 2)
self.assertEqual(state.num_humans(), 2)
# test the clear method
state.clear()
self.assertEqual(state.num_humans(), 0, "test clear")
def test_compute_distance_field(self):
state = Apocalypse(5, 5)
state.add_human(2, 3)
state.add_human(1, 2)
self.assertEqual(state.num_humans(), 2)
human_distance_field = state.compute_distance_field(zombie_apocalypse_bfs.HUMAN)
expected_field = [[3, 2, 1, 2, 3], [2, 1, 0, 1, 2], [3, 2, 1, 0, 1], [4, 3, 2, 1, 2], [5, 4, 3, 2, 3]]
self.assertEqual(human_distance_field, expected_field, "test distance_field")
# some print statements for debugging purposes.
print("human_distance_field:")
for row in human_distance_field:
print(row)
print()
print("human_list:", state._human_list)
def test_num_humans(self):
state = Apocalypse(5, 5)
state.add_human(2, 3)
self.assertEqual(state.num_humans(), 1, "test num humans and add humans")
