wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix bug with exec

This commit is contained in:
Cadene 2024-04-17 18:56:59 +00:00
parent b0f5120cee
commit 7c11568279
1 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/test_examples.py
View File

@ -1,4 +1,5 @@
from pathlib import Path from pathlib import Path
import subprocess
def _find_and_replace(text: str, finds_and_replaces: list[tuple[str, str]]) -> str: def _find_and_replace(text: str, finds_and_replaces: list[tuple[str, str]]) -> str:
@ -8,23 +9,19 @@ def _find_and_replace(text: str, finds_and_replaces: list[tuple[str, str]]) -> s
return text return text
def _run_script(path):
subprocess.run(['python', path], check=True)
def test_example_1(): def test_example_1():
path = "examples/1_load_hugging_face_dataset.py" path = "examples/1_load_hugging_face_dataset.py"
_run_script(path)
with open(path, "r") as file:
file_contents = file.read()
exec(file_contents)
assert Path("outputs/examples/1_load_hugging_face_dataset/episode_5.mp4").exists() assert Path("outputs/examples/1_load_hugging_face_dataset/episode_5.mp4").exists()
def test_example_2(): def test_example_2():
path = "examples/2_load_lerobot_dataset.py" path = "examples/2_load_lerobot_dataset.py"
_run_script(path)
with open(path, "r") as file:
file_contents = file.read()
exec(file_contents)
assert Path("outputs/examples/2_load_lerobot_dataset/episode_5.mp4").exists() assert Path("outputs/examples/2_load_lerobot_dataset/episode_5.mp4").exists()