Add test with using partial as custom wrapper

python · Bobronium · Sep 27, 2024 · Sep 27, 2024 · Sep 27, 2024 · Sep 27, 2024
commit 42f9dc3f8f5e5ce1502d3e7b6059f34908249d79
@@ -13,6 +13,7 @@
 import weakref
 import traceback
 import unittest
+from functools import partial
 from functools import update_wrapper
 from unittest.mock import Mock
 from typing import ClassVar, Any, List, Union, Tuple, Dict, Generic, TypeVar, Optional, Protocol, DefaultDict
@@ -5120,6 +5121,26 @@ def foo(cls):

        self.assertEqual(A().foo, "bar")

+    def test_custom_nested_descriptor_with_partial(self):
+        class CustomDescriptor:
+            def __init__(self, f):
+                self._wrapper = partial(f, value="bar")
+
+            def __get__(self, instance, owner):
+                return self._wrapper(instance)
+
+        class B:
+            def foo(self, value):
+                return value
+
+        @dataclass(slots=True)
+        class A(B):
+            @CustomDescriptor
+            def foo(self, value):
+                return super().foo(value)
+
+        self.assertEqual(A().foo, "bar")
+
    def test_custom_too_nested_descriptor(self):
        class UnnecessaryNestedWrapper:
            def __init__(self, wrapper):