Merge pull request #140 from dstansby/streamtracer-warn

Add user-friendly error when streamtracer not installed
dstansby · Feb 7, 2020 · e2bffa8 · e2bffa8
1 parent 7bf681d
commit e2bffa8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/pfsspy/tracing.py b/pfsspy/tracing.py
@@ -67,7 +67,13 @@ class FortranTracer(Tracer):
     directly on the poles will not go anywhere.
     """
     def __init__(self, max_steps=1000, step_size=0.01):
-        from streamtracer import StreamTracer
+        try:
+            from streamtracer import StreamTracer
+        except ModuleNotFoundError as e:
+            raise RuntimeError(
+                'Using FortranTracer requires the streamtracer module, '
+                'but streamtracer could not be loaded '
+                f'with the following error:\n{e}')
         self.max_steps = max_steps
         self.step_size = step_size
         self.tracer = StreamTracer(max_steps, step_size)

diff --git a/setup.cfg b/setup.cfg
@@ -16,5 +16,6 @@ filterwarnings =
     error
     ignore::ImportWarning
     ignore:numpy.ufunc size changed:RuntimeWarning
+    ignore:.*decorator is deprecated.*:DeprecationWarning
 markers =
   remote_data