"""
doctest does not discover tests in joblib-@cache'd functions.

Tests do get discovered in this module docstring:

>>> uncached("module docstring")
'Testing uncached: module docstring'
>>> cached("module docstring")
'Testing cached: module docstring'

However, the doctest in the cached() function below is ignored.
"""

from joblib import Memory
mem = Memory("test-joblib-doctest", verbose=0)

def uncached(x):
    """
    An uncached function.

    >>> uncached("function docstring")
    'Testing uncached: function docstring'
    """
    return "Testing uncached: {}".format(x)

@mem.cache
def cached(x):
    """
    A cached function.

    >>> cached("function docstring")
    'Testing cached: function docstring'
    """
    return "Testing cached: {}".format(x)

if __name__ == "__main__":
    import doctest
    doctest.testmod()