Guard for 3D axis equal (not supported in backend used for unittests)

ddbe84da · Martin Bauer · 73eb95e9 · ddbe84da
Commit ddbe84da authored Aug 30, 2019 by Martin Bauer
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

pystencils/stencil.py pystencils/stencil.py +4 -1

No files found.
--- a/pystencils/stencil.py
+++ b/pystencils/stencil.py
@@ -419,7 +419,10 @@ def plot_3d(stencil, figure=None, axes=None, data=None, textsize='8'):
        if figure is None:
            figure = plt.figure()
        axes = figure.gca(projection='3d')
-        axes.set_aspect("equal")
+        try:
+            axes.set_aspect("equal")
+        except NotImplementedError:
+            pass

    if data is None:
        data = [None] * len(stencil)