Adds a __repr__ to graph.

[pyutils.git] / src / pyutils / function_utils.py

diff --git a/src/pyutils/function_utils.py b/src/pyutils/function_utils.py
index a8ab0c74cfc50cb53b1e5b1b3f4dc49a8c9fc51e..8b2cd3790fa499baa1e010cf570e18ffe1c284f0 100644 (file)
--- 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."""
 
@@ -9,24 +9,38 @@ from typing import Callable
 
 def function_identifier(f: Callable) -> str:
     """
-    Given a callable function, return a string that identifies it.
-    Usually that string is just __module__:__name__ but there's a
+    Given a named `Callable`, return a string that identifies it.
+    Usually that string is just "__module__:__name__" but there's a
     corner case: when __module__ is __main__ (i.e. the callable is
     defined in the same module as __main__).  In this case,
     f.__module__ returns "__main__" instead of the file that it is
-    defined in.  Work around this using pathlib.Path (see below).
+    defined in.  Work around this using `pathlib.Path`.
+
+    Args:
+        f: a Callable
+
+    Returns:
+        A unique identifier for that callable in the format
+        module:function that avoids the pseudo-module '__main__'
 
     >>> function_identifier(function_identifier)
     'function_utils:function_identifier'
+
     """
 
-    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()