From 496c02dd237c1bf99d21d37325c9b89b65aada4a Mon Sep 17 00:00:00 2001
From: Pierre de Buyl <pierre.debuyl@meteo.be>
Date: Tue, 16 Nov 2021 11:58:24 +0100
Subject: Ignore matplotlib objects output lines

Contains code inspired by
https://github.com/wooyek/pytest-doctest-ellipsis-markers
(MIT license)
---
 doc/conftest.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'doc/conftest.py')

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)
-- 
cgit v1.2.1