mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-01 06:07:08 +00:00
24 lines
806 B
Python
24 lines
806 B
Python
"""Tests that all shortened imports for wrappers all work."""
|
|
|
|
import pytest
|
|
|
|
import gymnasium
|
|
from gymnasium.experimental.wrappers import (
|
|
_wrapper_to_class, # pyright: ignore[reportPrivateUsage]
|
|
)
|
|
from gymnasium.experimental.wrappers import __all__
|
|
|
|
|
|
def test_all_wrapper_shorten():
|
|
"""Test that all wrappers in `__alL__` are contained within the `_wrapper_to_class` conversion."""
|
|
all_wrappers = set(__all__)
|
|
all_wrappers.remove("vector")
|
|
assert all_wrappers == set(_wrapper_to_class.keys())
|
|
|
|
|
|
@pytest.mark.parametrize("wrapper_name", __all__)
|
|
def test_all_wrappers_shortened(wrapper_name):
|
|
"""Check that each element of the `__all__` wrappers can be loaded."""
|
|
if wrapper_name != "vector":
|
|
assert getattr(gymnasium.experimental.wrappers, wrapper_name) is not None
|