def test_gym_ex():
"""Run the gym ex sequence."""
try:
process = PexpectWrapper( # nosec
[
sys.executable,
str(Path("examples/gym_ex/train.py").resolve()),
"--nb-steps",
"50",
],
env=os.environ.copy(),
maxread=1,
encoding="utf-8",
logfile=sys.stdout,
)
process.expect(["Step 50/50"], timeout=10)
process.wait_to_complete(5)
assert process.returncode == 0, "Test failed"
finally:
process.terminate()
process.wait()
def test_gym_ex(self):
"""Run the gym ex sequence."""
try:
env = os.environ.copy()
env["PYTHONPATH"] = f"{self.t}{env_path_separator()}{env.get('PYTHONPATH', '')}"
process = PexpectWrapper( # nosec
[
sys.executable,
str(Path("examples/gym_ex/train.py").resolve()),
"--nb-steps",
"50",
],
env=env,
maxread=1,
encoding="utf-8",
logfile=sys.stdout,
)
process.expect(["Step 50/50"], timeout=10)
process.wait_to_complete(5)
assert process.returncode == 0, "Test failed"
finally:
process.terminate()
process.wait()
