ENH: allow per-doctest requirements in pytest_extra_requires

Author: ev-br

scipy_doctest/impl.py

@@ -93,10 +93,11 @@ class DTConfig:
         Each key is a doctest name to skip, and the corresponding value is
         a string. If not empty, the string value is used as the skip reason.
     pytest_extra_requires : dict
-        Paths to conditionally ignore unless requirements are met.
-        The format is ``{path/or/glob/pattern: requirement}``, where the values are
-        PEP 508 dependency specifiers. If a requirement is not met, the behavior
-        is equivalent to using the ``--ignore=...`` command line switch.
+        Paths or functions to conditionally ignore unless requirements are met.
+        The format is ``{path/or/glob/pattern: requirement, full.func.name: requiremet}``,
+        where the values are PEP 508 dependency specifiers. If a requirement is not met,
+        the behavior is equivalent to using the ``--ignore=...`` command line switch for
+        paths, and to using a `pytest_extra_skip` for function names.
     CheckerKlass : object, optional
         The class for the Checker object. Must mimic the ``DTChecker`` API:
         subclass the `doctest.OutputChecker` and make the constructor signature

scipy_doctest/plugin.py

@@ -5,9 +5,6 @@
 import warnings
 import doctest
 
-from importlib.metadata import version as get_version, PackageNotFoundError
-from packaging.requirements import Requirement
-
 import pytest
 import _pytest
 from _pytest import doctest as pydoctest, outcomes
@@ -16,7 +13,7 @@
 
 from .impl import DTParser, DebugDTRunner
 from .conftest import dt_config
-from .util import np_errstate, matplotlib_make_nongui, temp_cwd
+from .util import np_errstate, matplotlib_make_nongui, temp_cwd, is_req_satisfied
 from .frontend import find_doctests
 
 
@@ -93,12 +90,7 @@ def pytest_ignore_collect(collection_path, config):
 
     for entry, req_str in config.dt_config.pytest_extra_requires.items():
         if fnmatch_ex(entry, collection_path):
-            # check the requirement
-            req = Requirement(req_str)
-            try:
-                return not (get_version(req.name) in req.specifier)
-            except PackageNotFoundError:
-                return True
+            return not is_req_satisfied(req_str)
 
 
 def is_private(item):
@@ -139,6 +131,13 @@ def _maybe_add_markers(item, config):
             pytest.mark.xfail(reason=reason)
         )
 
+    extra_requires = dt_config.pytest_extra_requires
+    if req_str := extra_requires.get(item.name, None):
+        if not is_req_satisfied(req_str):
+            item.add_marker(
+                pytest.mark.skip(reason=f"requires {req_str}")
+            )
+
 
 def pytest_collection_modifyitems(config, items):
     """

scipy_doctest/util.py

@@ -10,6 +10,8 @@
 import inspect
 from contextlib import contextmanager
 
+from importlib.metadata import version as get_version, PackageNotFoundError
+from packaging.requirements import Requirement
 
 @contextmanager
 def matplotlib_make_nongui():
@@ -255,6 +257,16 @@ def get_public_objects(module, skiplist=None):
     return (items, names), failures
 
 
+def is_req_satisfied(req_str):
+    """ Check if a PEP 508-compliant requirement is satisfied or not.
+    """
+    req = Requirement(req_str)
+    try:
+        return get_version(req.name) in req.specifier
+    except PackageNotFoundError:
+        return False
+
+
 # XXX: not used ATM
 modules = []
 def generate_log(module, test):