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")