def test_human_and_zombie_and_obstacle_generators(self):
state = Apocalypse(5, 5, obstacle_list = [(1,2), (3,4), (1,3)])
state.add_human(2, 3)
state.add_human(0, 3)
state.add_zombie(1, 3)
state.add_zombie(3, 3)
# test human generator
human_generator = state.humans()
self.assertEqual(next(human_generator), (2,3), "test human generator")
self.assertEqual(next(human_generator), (0,3), "test human generator")
# test zombie generator
zombie_generator = state.zombies()
self.assertEqual(next(zombie_generator), (1,3), "test zombie generator")
self.assertEqual(next(zombie_generator), (3,3), "test zombie generator")
# test obstacle generator
obstacle_generator = state.obstacle()
self.assertEqual(next(obstacle_generator), (1,2), "test obstacle generator")
self.assertEqual(next(obstacle_generator), (3,4), "test obstacle generator")
self.assertEqual(next(obstacle_generator), (1,3), "test obstacle generator")
def test_num_zombies(self):
state = Apocalypse(5, 5)
state.add_zombie(2, 3)
self.assertEqual(state.num_zombies(), 1, "test num zombies and add zombies")
