do better at describing function implementations

1 files changed, 15 insertions, 0 deletions

diff --git a/neb/__init__.py b/neb/__init__.py
index 8ac8aeb..5602ebe 100644
--- a/neb/__init__.py
+++ b/neb/__init__.py
@@ -166,6 +166,21 @@ class MultiFunction(Callable):
         super().__init__(name, None, None)
         self.impls = {}
 
+    def __str__(self):
+        builtin_count = len(list(x for x in self.impls.values() if isinstance(x, Builtin)))
+        userfunc_count = len(list(x for x in self.impls.values() if isinstance(x, UserFunction)))
+        b = ""
+        uf = ""
+        if builtin_count != 0:
+            b = f"{builtin_count} builtin"
+        if userfunc_count != 0:
+            uf = f"{userfunc_count} user defined"
+        if b != "" and uf != "":
+            desc = f"{b}, {uf}"
+        else:
+            desc = b if b != "" else uf
+        return f"function {self.name} ({desc})"
+
     def describe(self):
         return [i.describe() for i in self.impls.values()]