pystencils/runhelper/db.py

@@ -35,10 +35,12 @@ class Database:
         ...     assert next(db.filter_params(params))['params'] == params # get data set, keys are 'params', 'results'
         ...                                                               # and 'env'
         ...     # get a pandas object with all results matching a query
-        ...     db.to_pandas({'dx': 1.5}, remove_prefix=True)  # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
-                             dx       method     error
-        pk
-        ...                 1.5  finite_diff  0.000001
+        ...     df = db.to_pandas({'dx': 1.5}, remove_prefix=True)
+        ...     # order columns alphabetically (just for doctest output)
+        ...     df.reindex(sorted(df.columns), axis=1)  # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
+                                 dx     error       method
+                pk
+                ...             1.5  0.000001  finite_diff
     """
 
     class SimulationResult(blitzdb.Document):

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)

setup.py

@@ -3,7 +3,6 @@ import io
 import os
 import sys
 from contextlib import redirect_stdout
-from distutils.extension import Extension
 from importlib import import_module
 
 from setuptools import find_packages, setup
@@ -59,6 +58,7 @@ def readme():
 
 
 def cython_extensions(*extensions):
+    from distutils.extension import Extension
     ext = '.pyx' if USE_CYTHON else '.c'
     result = [Extension(e, [e.replace('.', '/') + ext]) for e in extensions]
     if USE_CYTHON: