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

Add explicit error messages when unflatten discrete and multidiscrete fail (#267)

Browse Source
This commit is contained in:
Pierre Mardon
2023-01-18 18:32:54 +01:00
committed by GitHub
parent bb368fe75f
commit 6ba886abce
2 changed files with 26 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/spaces/test_utils.py
View File

@@ -135,3 +135,15 @@ def test_flatten_roundtripping(space):
for original, roundtripped in zip(samples, unflattened_samples):
assert data_equivalence(original, roundtripped)
def test_unflatten_discrete_error():
value = np.array([0])
with pytest.raises(ValueError):
utils.unflatten(gym.spaces.Discrete(1), value)
def test_unflatten_multidiscrete_error():
value = np.array([0, 0])
with pytest.raises(ValueError):
utils.unflatten(gym.spaces.MultiDiscrete([1, 1]), value)