X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=src%2Fpyutils%2Ffunction_utils.py;h=8b2cd3790fa499baa1e010cf570e18ffe1c284f0;hb=HEAD;hp=ecfd4c32c9b4e8372749da2ad9b4a21f83795513;hpb=b38920f24d1ac948958480c540bc4b8436186765;p=pyutils.git diff --git a/src/pyutils/function_utils.py b/src/pyutils/function_utils.py index ecfd4c3..8b2cd37 100644 --- a/src/pyutils/function_utils.py +++ b/src/pyutils/function_utils.py @@ -1,6 +1,6 @@ #!/usr/bin/env python3 -# © Copyright 2021-2022, Scott Gasch +# © Copyright 2021-2023, Scott Gasch """Helper methods dealing with functions.""" @@ -28,13 +28,19 @@ def function_identifier(f: Callable) -> str: """ - if f.__module__ == '__main__': + if f.__module__ == "__main__": from pathlib import Path import __main__ module = __main__.__file__ module = Path(module).stem - return f'{module}:{f.__name__}' + return f"{module}:{f.__name__}" else: - return f'{f.__module__}:{f.__name__}' + return f"{f.__module__}:{f.__name__}" + + +if __name__ == "__main__": + import doctest + + doctest.testmod()