feat: see methods from the bound function (#193)

* feat: see methods from the bound function Signed-off-by: nstarman <[email protected]> * test: add test for the dispatcher. Signed-off-by: nstarman <[email protected]> --------- Signed-off-by: nstarman <[email protected]>
beartype · Nov 15, 2024 · 463acd9 · 463acd9
1 parent ba3a974
commit 463acd9
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/plum/function.py b/plum/function.py
@@ -515,3 +515,8 @@ def wrapped_method(*args, **kw_args):
         # extended, so unwrapping is likely never desired.
 
         return wrapped_method
+
+    @property
+    def methods(self) -> List[Signature]:
+        """list[:class:`.signature.Signature`]: All available methods."""
+        return self._f.methods
diff --git a/tests/test_dispatcher.py b/tests/test_dispatcher.py
@@ -40,6 +40,20 @@ def g(x: float):
     assert dispatch.classes[b]["g"].methods[0].signature == Sig(float, precedence=1)
 
 
+def test_bound_function_attributes():
+    """Test the attributes on a bound function."""
+    dispatch = Dispatcher()
+
+    class A:
+        @dispatch
+        def f(self, x: int):
+            pass
+
+    # The 'methods' should point to those of the underlying object.
+    assert A().f.methods is A.f.methods
+    assert A().f.methods is A().f._f.methods
+
+
 def test_dispatch_multi():
     dispatch = Dispatcher()