def test_8_by_8_random_agent(self): agent = RandomAgent(problem_id=0) self.assertEqual(agent.problem_id, 0) self.assertEqual(agent.is_stochastic(), True) self.assertEqual(agent.env.ncol, 8) self.assertEqual(agent.env.nrow, 8) agent = RandomAgent(problem_id=1) self.assertEqual(agent.problem_id, 1) self.assertEqual(agent.is_stochastic(), True) self.assertEqual(agent.env.ncol, 8) self.assertEqual(agent.env.nrow, 8)
def test_4_by_4_random_agent(self): agent = RandomAgent(problem_id=1, map_name_base="4x4-base") self.assertEqual(agent.problem_id, 1) self.assertEqual(agent.is_stochastic(), True) self.assertEqual(agent.env.ncol, 4) self.assertEqual(agent.env.nrow, 4)