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)
