test(doc2test): Test for eval-rst blocks

twmr · Jul 12, 2022 · 6e6d21a · 6e6d21a
1 parent 015d7bc
commit 6e6d21a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_doc2test.py b/tests/test_doc2test.py
@@ -1,6 +1,7 @@
 import doctest
 import os
 import textwrap
+from typing import Literal
 
 import pytest
 
@@ -87,9 +88,10 @@ def test_indented():
     assert example.lineno == 7
 
 
-def test_cartopy():
+@pytest.mark.parametrize("file_type", ["rst", "md"])
+def test_cartopy(file_type: Literal['rst', 'md']):
     rstpath = os.path.join(
-        os.path.dirname(__file__), "testdata", "using_the_shapereader.rst"
+        os.path.dirname(__file__), "testdata", f"using_the_shapereader.{file_type}"
     )
     with open(rstpath, "r") as fh:
         sections = get_sections(fh.read())