Make mandel_main to work without Cython

Panda142 · Sep 16, 2020 · 208cc38 · 208cc38
1 parent 5ae6f0d
commit 208cc38
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/demos/cython/mandel_main.py b/demos/cython/mandel_main.py
@@ -1,8 +1,11 @@
-from mandel import compute_mandel as compute_mandel_py
-from mandel_cyt import compute_mandel as compute_mandel_cyt
 import numpy as np
 import matplotlib.pyplot as plt
 import sys
+from mandel import compute_mandel as compute_mandel_py
+try:
+    from mandel_cyt import compute_mandel as compute_mandel_cyt
+except ImportError:
+    pass
 
 def plot_mandel(mandel):
     plt.imshow(mandel)
@@ -20,7 +23,10 @@ def main(version='py'):
         mandel_func = compute_mandel_py
     elif version == 'cyt': 
         print("Using Cython")
-        mandel_func = compute_mandel_cyt
+        try:
+            mandel_func = compute_mandel_cyt
+        except NameError as ex:
+            raise RuntimeError("Cython extension missing") from ex
     else:
         raise RuntimeError("Unknown version")
 

diff --git a/demos/performance/mandel.py b/demos/performance/mandel.py
@@ -0,0 +1 @@
+../cython/mandel.py
diff --git a/demos/performance/mandel_main.py b/demos/performance/mandel_main.py
@@ -0,0 +1 @@
+../cython/mandel_main.py