mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-09-02 18:36:16 +00:00
37 lines
819 B
Python
37 lines
819 B
Python
|
import contextlib
|
||
|
|
import glob
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
import tempfile
|
||
|
|
|
||
|
|
import gym
|
||
|
|
from gym.monitoring import monitor
|
||
|
|
|
||
|
|
class FakeEnv(gym.Env):
|
||
|
|
def _render(self, close=True):
|
||
|
|
raise RuntimeError('Raising')
|
||
|
|
|
||
|
|
@contextlib.contextmanager
|
||
|
|
def tempdir():
|
||
|
|
temp = tempfile.mkdtemp()
|
||
|
|
yield temp
|
||
|
|
shutil.rmtree(temp)
|
||
|
|
|
||
|
|
def test_monitor_filename():
|
||
|
|
with tempdir() as temp:
|
||
|
|
env = gym.make('Acrobot-v0')
|
||
|
|
env.monitor.start(temp)
|
||
|
|
env.monitor.close()
|
||
|
|
|
||
|
|
manifests = glob.glob(os.path.join(temp, '*.manifest.*'))
|
||
|
|
assert len(manifests) == 1
|
||
|
|
|
||
|
|
def test_close_monitor():
|
||
|
|
with tempdir() as temp:
|
||
|
|
env = FakeEnv()
|
||
|
|
env.monitor.start(temp)
|
||
|
|
env.monitor.close()
|
||
|
|
|
||
|
|
manifests = monitor.detect_training_manifests(temp)
|
||
|
|
assert len(manifests) == 1
|