mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-05 15:31:44 +00:00
83 lines
3.0 KiB
Python
83 lines
3.0 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
import gym
|
||
|
|
from gym.envs.box2d import BipedalWalker
|
||
|
|
from gym.envs.box2d.lunar_lander import demo_heuristic_lander
|
||
|
|
from gym.envs.toy_text.frozen_lake import generate_random_map
|
||
|
|
|
||
|
|
|
||
|
|
def test_lunar_lander_heuristics():
|
||
|
|
lunar_lander = gym.make("LunarLander-v2", disable_env_checker=True)
|
||
|
|
total_reward = demo_heuristic_lander(lunar_lander, seed=1)
|
||
|
|
assert total_reward > 100
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("seed", range(5))
|
||
|
|
def test_bipedal_walker_hardcore_creation(seed: int):
|
||
|
|
"""Test BipedalWalker hardcore creation.
|
||
|
|
|
||
|
|
BipedalWalker with `hardcore=True` should have ladders
|
||
|
|
stumps and pitfalls. A convenient way to identify if ladders,
|
||
|
|
stumps and pitfall are created is checking whether the terrain
|
||
|
|
has that particular terrain color.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
seed (int): environment seed
|
||
|
|
"""
|
||
|
|
HC_TERRAINS_COLOR1 = (255, 255, 255)
|
||
|
|
HC_TERRAINS_COLOR2 = (153, 153, 153)
|
||
|
|
|
||
|
|
env = gym.make("BipedalWalker-v3", disable_env_checker=True).unwrapped
|
||
|
|
hc_env = gym.make("BipedalWalkerHardcore-v3", disable_env_checker=True).unwrapped
|
||
|
|
assert isinstance(env, BipedalWalker) and isinstance(hc_env, BipedalWalker)
|
||
|
|
assert env.hardcore is False and hc_env.hardcore is True
|
||
|
|
|
||
|
|
env.reset(seed=seed)
|
||
|
|
hc_env.reset(seed=seed)
|
||
|
|
|
||
|
|
for terrain in env.terrain:
|
||
|
|
assert terrain.color1 != HC_TERRAINS_COLOR1
|
||
|
|
assert terrain.color2 != HC_TERRAINS_COLOR2
|
||
|
|
|
||
|
|
hc_terrains_color1_count = 0
|
||
|
|
hc_terrains_color2_count = 0
|
||
|
|
for terrain in hc_env.terrain:
|
||
|
|
if terrain.color1 == HC_TERRAINS_COLOR1:
|
||
|
|
hc_terrains_color1_count += 1
|
||
|
|
if terrain.color2 == HC_TERRAINS_COLOR2:
|
||
|
|
hc_terrains_color2_count += 1
|
||
|
|
|
||
|
|
assert hc_terrains_color1_count > 0
|
||
|
|
assert hc_terrains_color2_count > 0
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("map_size", [5, 10, 16])
|
||
|
|
def test_frozenlake_dfs_map_generation(map_size: int):
|
||
|
|
"""Frozenlake has the ability to generate random maps.
|
||
|
|
|
||
|
|
This function checks that the random maps will always be possible to solve for sizes 5, 10, 16,
|
||
|
|
currently only 8x8 maps can be generated.
|
||
|
|
"""
|
||
|
|
new_frozenlake = generate_random_map(map_size)
|
||
|
|
assert len(new_frozenlake) == map_size
|
||
|
|
assert len(new_frozenlake[0]) == map_size
|
||
|
|
|
||
|
|
# Runs a depth first search through the map to find the path.
|
||
|
|
directions = [(1, 0), (0, 1), (-1, 0), (0, -1)]
|
||
|
|
frontier, discovered = [], set()
|
||
|
|
frontier.append((0, 0))
|
||
|
|
while frontier:
|
||
|
|
row, col = frontier.pop()
|
||
|
|
if (row, col) not in discovered:
|
||
|
|
discovered.add((row, col))
|
||
|
|
|
||
|
|
for row_direction, col_direction in directions:
|
||
|
|
new_row = row + row_direction
|
||
|
|
new_col = col + col_direction
|
||
|
|
if 0 <= new_row < map_size and 0 <= new_col < map_size:
|
||
|
|
if new_frozenlake[new_row][new_col] == "G":
|
||
|
|
return # Successful, a route through the map was found
|
||
|
|
if new_frozenlake[new_row][new_col] not in "#H":
|
||
|
|
frontier.append((new_row, new_col))
|
||
|
|
raise AssertionError("No path through the frozenlake was found.")
|