mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-14 10:48:45 +00:00
26 lines
669 B
Python
26 lines
669 B
Python
|
import pickle
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from gym import envs
|
||
|
|
from gym.envs.tests.spec_list import skip_mujoco, SKIP_MUJOCO_WARNING_MESSAGE
|
||
|
|
|
||
|
|
|
||
|
|
ENVIRONMENT_IDS = (
|
||
|
|
'HandManipulateEgg-v0',
|
||
|
|
'HandManipulatePen-v0',
|
||
|
|
'HandManipulateBlock-v0',
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.skipif(skip_mujoco, reason=SKIP_MUJOCO_WARNING_MESSAGE)
|
||
|
|
@pytest.mark.parametrize("environment_id", ENVIRONMENT_IDS)
|
||
|
|
def test_serialize_deserialize(environment_id):
|
||
|
|
env1 = envs.make(environment_id, target_position='fixed')
|
||
|
|
env1.reset()
|
||
|
|
env2 = pickle.loads(pickle.dumps(env1))
|
||
|
|
|
||
|
|
assert env1.target_position == env2.target_position, (
|
||
|
|
env1.target_position, env2.target_position)
|