mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-22 07:02:19 +00:00
Modify Space.seed
such that the return can be used as seeding values (#1033)
This commit is contained in:
@@ -37,17 +37,16 @@ def test_sequence_inheritance():
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"space, seed, expected_len",
|
||||
"space, seed",
|
||||
[
|
||||
(Tuple([Discrete(5), Discrete(4)]), None, 2),
|
||||
(Tuple([Discrete(5), Discrete(4)]), 123, 3),
|
||||
(Tuple([Discrete(5), Discrete(4)]), (123, 456), 2),
|
||||
(Tuple([Discrete(5), Discrete(4)]), None),
|
||||
(Tuple([Discrete(5), Discrete(4)]), 123),
|
||||
(Tuple([Discrete(5), Discrete(4)]), (123, 456)),
|
||||
(
|
||||
Tuple(
|
||||
(Discrete(5), Tuple((Box(low=0.0, high=1.0, shape=(3,)), Discrete(2))))
|
||||
),
|
||||
(123, (456, 789)),
|
||||
3,
|
||||
),
|
||||
(
|
||||
Tuple(
|
||||
@@ -57,22 +56,21 @@ def test_sequence_inheritance():
|
||||
)
|
||||
),
|
||||
(123, {"position": 456, "velocity": 789}),
|
||||
3,
|
||||
),
|
||||
],
|
||||
)
|
||||
def test_seeds(space, seed, expected_len):
|
||||
seeds = space.seed(seed)
|
||||
assert isinstance(seeds, list) and all(isinstance(elem, int) for elem in seeds)
|
||||
assert len(seeds) == expected_len
|
||||
def test_seeds(space, seed):
|
||||
seeds1 = space.seed(seed)
|
||||
assert isinstance(seeds1, tuple)
|
||||
assert len(seeds1) == len(space)
|
||||
|
||||
sample1 = space.sample()
|
||||
|
||||
seeds2 = space.seed(seed)
|
||||
seeds2 = space.seed(seeds1)
|
||||
sample2 = space.sample()
|
||||
|
||||
data_equivalence(seeds, seeds2)
|
||||
data_equivalence(sample1, sample2)
|
||||
assert data_equivalence(seeds1, seeds2)
|
||||
assert data_equivalence(sample1, sample2)
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
|
Reference in New Issue
Block a user