20 lines
476 B
Python
20 lines
476 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)
|
|
|
|
if path == "examples/1_visualize_dataset.py":
|
|
assert Path("outputs/visualize_dataset/example/episode_0.mp4").exists()
|