gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-23 23:12:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Type hint fixes and added __all__ dunder (#321)

Browse Source
This commit is contained in:
Howard Huang
2023-02-12 12:42:32 -05:00
committed by GitHub
parent 9bc0bf308d
commit 79ae76ed1e
6 changed files with 23 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/vector/test_sync_vector_env.py
View File

@@ -29,6 +29,7 @@ def test_reset_sync_vector_env():
assert isinstance(env.observation_space, Box)
assert isinstance(observations, np.ndarray)
assert isinstance(infos, dict)
assert observations.dtype == env.observation_space.dtype
assert observations.shape == (8,) + env.single_observation_space.shape
assert observations.shape == env.observation_space.shape
@@ -130,15 +131,18 @@ def test_custom_space_sync_vector_env():
assert isinstance(env.single_action_space, CustomSpace)
assert isinstance(env.action_space, Tuple)
assert isinstance(infos, dict)
actions = ("action-2", "action-3", "action-5", "action-7")
step_observations, rewards, terminateds, truncateds, _ = env.step(actions)
step_observations, rewards, terminateds, truncateds, infos = env.step(actions)
env.close()
assert isinstance(env.single_observation_space, CustomSpace)
assert isinstance(env.observation_space, Tuple)
assert isinstance(infos, dict)
assert isinstance(reset_observations, tuple)
assert reset_observations == ("reset", "reset", "reset", "reset")