Fix paths in test_examples.py

ali-ramadhan · May 9, 2024 · 1d38374 · 1d38374
1 parent e974700
commit 1d38374
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -2,10 +2,15 @@
 
 from pathlib import Path
 
+def run_example(name):
+    filepath = Path(__file__).resolve().parent / ".." / "examples" / f"{name}.py"
+    exec(filepath.read_text())
+    return
+
 def test_sine_wave():
-    exec(Path("../examples/sine_wave.py").read_text())
+    run_example("sine_wave")
     assert Path("sine_wave.gif").is_file()
 
 def test_rotating_circular_sine_wave():
-    exec(Path("../examples/rotating_circular_sine_wave.py").read_text())
+    run_example("rotating_circular_sine_wave")
     assert Path("rotating_circular_sine_wave.mp4").is_file()