19 lines
422 B
Python
19 lines
422 B
Python
|
import pytest
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"path",
|
||
|
|
[
|
||
|
|
"examples/1_visualize_dataset.py",
|
||
|
|
"examples/2_evaluate_pretrained_policy.py",
|
||
|
|
"examples/3_train_policy.py",
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_example(path):
|
||
|
|
|
||
|
|
with open(path, 'r') as file:
|
||
|
|
file_contents = file.read()
|
||
|
|
exec(file_contents)
|
||
|
|
|
||
|
|
assert Path("outputs/visualize_dataset/example/episode_0.mp4").exists()
|