fix: examples files can import local files.

mind-inria · Jul 1, 2024 · da5fe60 · da5fe60
1 parent 9b2c9f8
commit da5fe60
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/examples/conftest.py b/examples/conftest.py
@@ -12,6 +12,7 @@
 
 """
 
+import sys
 import runpy
 from pathlib import Path
 
@@ -45,6 +46,7 @@ class ScriptItem(pytest.Item):
 
     def runtest(self):
         """Run the script as a test."""
+        sys.path.insert(0, str(self.path.parent))
         runpy.run_path(str(self.path))
 
     def repr_failure(self, excinfo):

diff --git a/examples/example_2D_trajectories.py b/examples/example_2D_trajectories.py
@@ -25,7 +25,7 @@
 import mrinufft as mn
 import mrinufft.trajectories.maths as mntm
 from mrinufft import display_2D_trajectory
-from .utils import show_argument, show_trajectory
+from utils import show_argument, show_trajectory
 
 
 # %%

diff --git a/examples/example_3D_trajectories.py b/examples/example_3D_trajectories.py
@@ -30,7 +30,7 @@
 # Internal
 import mrinufft as mn
 from mrinufft import display_2D_trajectory, display_3D_trajectory
-from .utils import show_argument, show_trajectory
+from utils import show_argument, show_trajectory
 
 
 # %%

diff --git a/examples/example_trajectory_tools.py b/examples/example_trajectory_tools.py
@@ -30,7 +30,7 @@
 import mrinufft.trajectories.tools as tools
 
 from mrinufft.trajectories.utils import KMAX
-from .utils import show_argument, show_trajectory
+from utils import show_argument, show_trajectory
 
 
 # %%