make relative issue work for python 3

fangohr · Mar 18, 2016 · 09c831b · 09c831b
1 parent 6b83bde
commit 09c831b
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/joommf/test_examples.py b/joommf/test_examples.py
@@ -1,8 +1,17 @@
+import os
+import sys
+
+# putting 'joommf' into search path
+sys.path.insert(0, os.path.abspath('..'))
+print(sys.path)
+
+
 def test_run_min_example():
-    import minimisation_example
-    minimisation_example.main()
+    import joommf.minimisation_example
+    joommf.minimisation_example.main()
+
 
 
 def test_run_dyn_example():
-    import dynamics_example
-    dynamics_example.main()
+    import joommf.dynamics_example
+    joommf.dynamics_example.main()