move doctest setup routines to a location actually checked by pytest

madmom/conftest.py

@@ -0,0 +1,17 @@
+import numpy as np
+import pytest
+
+# set and restore numpy's print options for doctests
+_NP_PRINT_OPTIONS = np.get_printoptions()
+
+
+@pytest.fixture(autouse=True)
+def setup_package(request):
+    """Set up the environment for doctests (when run through pytest)."""
+    np.set_printoptions(precision=5, edgeitems=2, suppress=True)
+
+    def fin():
+        """Restore the environment after doctests (when run through pytest)."""
+        np.set_printoptions(**_NP_PRINT_OPTIONS)
+
+    request.addfinalizer(fin)