From 5ee37b0e7966cd7181f3436a6758b19871d2597f Mon Sep 17 00:00:00 2001
From: Pierre de Buyl <pdebuyl@pdebuyl.be>
Date: Wed, 17 Nov 2021 10:08:57 +0100
Subject: Use empty lines for mpl_toolkits.mplot3d objects

---
 doc/conftest.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'doc/conftest.py')

diff --git a/doc/conftest.py b/doc/conftest.py
index 80a29487c..cfd65c1aa 100644
--- a/doc/conftest.py
+++ b/doc/conftest.py
@@ -13,10 +13,13 @@ matplotlib.use('agg', force=True)
 # https://github.com/wooyek/pytest-doctest-ellipsis-markers (MIT license)
 OutputChecker = doctest.OutputChecker
 
+empty_line_markers = ['<matplotlib.', '<mpl_toolkits.mplot3d.']
 class SkipMatplotlibOutputChecker(doctest.OutputChecker):
     def check_output(self, want, got, optionflags):
-        if '<matplotlib.' in got:
-            got = ''
+        for marker in empty_line_markers:
+            if marker in got:
+                got = ''
+                break
         return OutputChecker.check_output(self, want, got, optionflags)
 
 doctest.OutputChecker = SkipMatplotlibOutputChecker
-- 
cgit v1.2.1