def test_is_agent_next_to_wall(example, expected_result):
env = custom_frozen_lake.FrozenLakeEnv()
env._agent_position = example
result = env._get_walls_next_to_agent()
assert result == expected_result
def test_is_agent_in_hole(example, expected_result):
env = custom_frozen_lake.FrozenLakeEnv()
env._agent_position = example
result = env._is_agent_in_hole()
assert result == expected_result
def test_set_agent_position(example, expected_result):
env = custom_frozen_lake.FrozenLakeEnv()
env._set_agent_position(example)
assert env._agent_position == expected_result
def test_step(example, initial_position, expected_result):
env = custom_frozen_lake.FrozenLakeEnv()
env._agent_position = initial_position
result = env.step(example)
assert result == expected_result
def test_get_goal_observation(example, expected_result):
env = custom_frozen_lake.FrozenLakeEnv(goal_position=example)
result = env._get_goal_observation()
assert result == expected_result