Adds a __repr__ to graph.
[pyutils.git] / src / pyutils / function_utils.py
index ecfd4c32c9b4e8372749da2ad9b4a21f83795513..8b2cd3790fa499baa1e010cf570e18ffe1c284f0 100644 (file)
@@ -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()