doctest uses inspect.isfunction() to detect callable objects on which to detect docstrings. Unfortunately, this prevents running doctests on functions which have been decorated to return other types of callables (for example numba's @jit decorator). In the attached example file, the wrapped "bar" function does not have its doctest executed.

It would be useful for doctest to be more flexible here, although I'm not sure what the exact criterion should be.

Would using callable() instead of inspect.isfunction() be ok?

Ezio Melotti added the comment:

Would using callable() instead of inspect.isfunction() be ok?

I'm not sure, because it would also select classes. I guess we need
something a bit smarter.

How about using this?

diff -r 1e74350dd056 Lib/doctest.py
--- a/Lib/doctest.py    Tue Jun 17 22:27:46 2014 -0500
+++ b/Lib/doctest.py    Fri Jun 20 11:08:00 2014 +0300
@@ -984,7 +984,8 @@
             for valname, val in obj.__dict__.items():
                 valname = '%s.%s' % (name, valname)
                 # Recurse to functions & classes.
-                if ((inspect.isroutine(val) or inspect.isclass(val)) and
+
+                if ((inspect.isroutine(inspect.unwrap(val)) or inspect.isclass(val)) and
                     self._from_module(module, val)):
                     self._find(tests, val, valname, module, source_lines,
                                globs, seen)

This seems to work for the given example and if the decorator uses update_wrapper or @wraps.

I'm not sure, because it would also select classes.

Is there any reason to preclude classes? They could reasonably have docstrings that contain doctests.

> I'm not sure, because it would also select classes.

Is there any reason to preclude classes? They could reasonably have docstrings that contain doctests.

I don't know. I just didn't want to change doctest behaviour too much,
but if people more knowledgeable than me about it feel it's ok, then all
the better.

Class doctests are already supported separately, see https://docs.python.org/3/library/doctest.html#which-docstrings-are-examined

Here's a test patch which uses inspect.unwrap. Unfortunately, I can't test with numba, so I don't know if it works for that, but any decorated function which uses functools.update_wrapper or wraps should be detected by doctest.

New changeset d22ca7496c54 by Yury Selivanov in branch 'default':
Issue bpo-21740: Support wrapped callables in pydoc. Patch by Claudiu Popa.
https://hg.python.org/cpython/rev/d22ca7496c54

Thank you for the patch, Claudiu!

New changeset 12ef799a9a51 by Zachary Ware in branch 'default':
Issue bpo-21740: Fix module name in NEWS entry.
https://hg.python.org/cpython/rev/12ef799a9a51