summaryrefslogtreecommitdiff
path: root/doc/conftest.py
diff options
context:
space:
mode:
Diffstat (limited to 'doc/conftest.py')
-rw-r--r--doc/conftest.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/doc/conftest.py b/doc/conftest.py
index c091c6c5e..80a29487c 100644
--- a/doc/conftest.py
+++ b/doc/conftest.py
@@ -4,9 +4,23 @@ Pytest configuration and fixtures for the Numpy test suite.
import pytest
import numpy
import matplotlib
+import doctest
matplotlib.use('agg', force=True)
+# Ignore matplotlib output such as `<matplotlib.image.AxesImage at
+# 0x7f956908c280>`. doctest monkeypatching inspired by
+# https://github.com/wooyek/pytest-doctest-ellipsis-markers (MIT license)
+OutputChecker = doctest.OutputChecker
+
+class SkipMatplotlibOutputChecker(doctest.OutputChecker):
+ def check_output(self, want, got, optionflags):
+ if '<matplotlib.' in got:
+ got = ''
+ return OutputChecker.check_output(self, want, got, optionflags)
+
+doctest.OutputChecker = SkipMatplotlibOutputChecker
+
@pytest.fixture(autouse=True)
def add_np(doctest_namespace):
numpy.random.seed(1)