diff --git a/devito/__init__.py b/devito/__init__.py index 214a0ee8f24..5aeb46837fb 100644 --- a/devito/__init__.py +++ b/devito/__init__.py @@ -1,7 +1,10 @@ import atexit from itertools import product +import os from . import _version +import numpy as np + # Import the global `configuration` dict from devito.parameters import * # noqa @@ -160,6 +163,13 @@ def mode_performance(): configuration['opt-options']['blockinner'] = True +if "PYTEST_VERSION" in os.environ: + # Avoid change in repr break docstring tests + # Only sets it here for testing + # https://numpy.org/devdocs/release/2.0.0-notes.html#representation-of-numpy-scalars-changed # noqa + np.set_printoptions(legacy="1.25") + + # Ensure the SymPy caches are purged at exit # For whatever reason, if we don't do this the garbage collector won't its # job properly and thus we may end up missing some custom __del__'s