def test_fixed_plan_agents(self):
    a0 = FixedPlanAgent([s, e, n, w])
    a1 = FixedPlanAgent([s, w, n, e])
    agent_pair = AgentPair(a0, a1)
    env = OvercookedEnv.from_mdp(large_mdp, horizon=10)
    (trajectory, time_taken, _, _) = env.run_agents(agent_pair, include_final_state=True, display=DISPLAY)
    end_state = trajectory[(- 1)][0]
    self.assertEqual(time_taken, 10)
    self.assertEqual(env.mdp.get_standard_start_state().player_positions, end_state.player_positions)
def test_fixed_plan_agents(self):
    a0 = FixedPlanAgent([s, e, n, w])
    a1 = FixedPlanAgent([s, w, n, e])
    agent_pair = AgentPair(a0, a1)
    env = OvercookedEnv.from_mdp(large_mdp, horizon=10)
    (trajectory, time_taken, _, _) = env.run_agents(agent_pair, include_final_state=True, display=DISPLAY)
    end_state = trajectory[(- 1)][0]
    self.assertEqual(time_taken, 10)
    self.assertEqual(env.mdp.get_standard_start_state().player_positions, end_state.player_positions)