TST: fix the xdoctest Parser drop-in example test

Author: ev-br

scipy_doctest/tests/test_runner.py

@@ -125,6 +125,15 @@ def __init__(self, config):
     def parse(self, string, name='<string>'):
         return self.xd.parse(string)
 
+    def get_examples(self, string, name='<string>'):
+        """
+        Similar to doctest.DocTestParser.get_examples, only
+        account for the fact that individual examples
+        are instances of DoctestPart not doctest.Example
+        """
+        return [x for x in self.parse(string, name)
+                if isinstance(x, xdoctest.doctest_part.DoctestPart)]
+
 
 class TestParserDropIn:
     """ Test an alternative DoctestParser
@@ -139,9 +148,18 @@ def test_vanilla_parser(self):
 
     @pytest.mark.skipif(not HAVE_XDOCTEST, reason="needs xdoctest")
     def test_xdoctest_parser(self):
+        # Note that the # of examples differ from DTParser:
+        # - xdoctest groups doctest lines with no 'want' output into a single
+        #   example.
+        # - "examples" here are DoctestPart instances, which _almost_ quack
+        #    like `doctest.Example` but not completely.
         config = DTConfig(ParserKlass=XDParser)
         runner = DebugDTRunner(config=config)
         tests = DTFinder(config=config).find(module_cases.func3)
 
         assert len(tests) == 1
-        assert len(tests[0].examples) == 3
+        assert len(tests[0].examples) == 2
+        assert (tests[0].examples[0].source ==
+                'import numpy as np\na = np.array([1, 2, 3, 4]) / 3'
+        )
+        assert tests[0].examples[1].source == 'print(a)'