support for staticmethod

obriencj · Jul 6, 2024 · 3ad6ae3 · 3ad6ae3
1 parent febabb9
commit 3ad6ae3
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 2 deletions.
diff --git a/preoccupied/proxytype/mypy.py b/preoccupied/proxytype/mypy.py
@@ -123,6 +123,11 @@ def decorate_proxytype(wrap: TypeInfo, orig: Instance, virt: Instance):
             nsym.node = clone_overloaded(node, wrap, virt)
             wrap.names[name] = nsym
 
+        elif isinstance(node, Decorator):
+            nsym = sym.copy()
+            nsym.node = clone_decorator(node, wrap, virt)
+            wrap.names[name] = nsym
+
         else:
             # ignore others
             pass

diff --git a/setup.cfg b/setup.cfg
@@ -2,7 +2,7 @@
 [metadata]
 
 name = preoccupied.proxytype
-version = 0.3.0
+version = 0.4.0
 description = static typing support for dynamic proxy classes
 
 author = Christopher O'Brien
@@ -83,7 +83,7 @@ download = True
 basepython = python3.9
 
 commands =
-  python -B -I -m nose
+  python -B -I -m nose -s
 
 download = True
 

diff --git a/tests/data/binary.pyi b/tests/data/binary.pyi
@@ -17,6 +17,10 @@ class Normal:
     def getName(self) -> str:
         ...
 
+    @staticmethod
+    def getStatic() -> int:
+        ...
+
 
 RT = TypeVar("RT")
 

diff --git a/tests/data/unary.pyi b/tests/data/unary.pyi
@@ -17,6 +17,10 @@ class Normal:
     def getName(self) -> str:
         ...
 
+    @staticmethod
+    def getStatic() -> int:
+        ...
+
 
 @proxytype(Normal)
 class Delayed:

diff --git a/tests/mypy.py b/tests/mypy.py
@@ -70,10 +70,12 @@ def test_plugin_disabled(self):
         self.assertIn("doSomething", normal.names)
         self.assertIn("doAnother", normal.names)
         self.assertIn("getName", normal.names)
+        self.assertIn("getStatic", normal.names)
 
         self.assertNotIn("doSomething", delayed.names)
         self.assertNotIn("doAnother", delayed.names)
         self.assertIn("getName", delayed.names)
+        self.assertNotIn("getStatic", delayed.names)
 
 
     def test_plugin_enabled(self):
@@ -89,10 +91,12 @@ def test_plugin_enabled(self):
         self.assertIn("doSomething", normal.names)
         self.assertIn("doAnother", normal.names)
         self.assertIn("getName", normal.names)
+        self.assertIn("getStatic", normal.names)
 
         self.assertIn("doSomething", delayed.names)
         self.assertIn("doAnother", delayed.names)
         self.assertIn("getName", delayed.names)
+        self.assertIn("getStatic", delayed.names)
 
 
 class UnaryTest(TestCase):
@@ -115,10 +119,12 @@ def test_plugin_disabled(self):
         self.assertIn("doSomething", normal.names)
         self.assertIn("doAnother", normal.names)
         self.assertIn("getName", normal.names)
+        self.assertIn("getStatic", normal.names)
 
         self.assertNotIn("doSomething", delayed.names)
         self.assertNotIn("doAnother", delayed.names)
         self.assertIn("getName", delayed.names)
+        self.assertNotIn("getStatic", delayed.names)
 
 
     def test_plugin_enabled(self):
@@ -133,10 +139,12 @@ def test_plugin_enabled(self):
         self.assertIn("doSomething", normal.names)
         self.assertIn("doAnother", normal.names)
         self.assertIn("getName", normal.names)
+        self.assertIn("getStatic", normal.names)
 
         self.assertIn("doSomething", delayed.names)
         self.assertIn("doAnother", delayed.names)
         self.assertIn("getName", delayed.names)
+        self.assertIn("getStatic", delayed.names)
 
 
 # The end.