summaryrefslogtreecommitdiff
path: root/test_six.py
diff options
context:
space:
mode:
Diffstat (limited to 'test_six.py')
-rw-r--r--test_six.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/test_six.py b/test_six.py
index 9e39444..371ee81 100644
--- a/test_six.py
+++ b/test_six.py
@@ -636,6 +636,24 @@ def test_with_metaclass():
assert issubclass(X, Base2)
+def test_wraps():
+ def f(g):
+ @six.wraps(g)
+ def w():
+ return 42
+ return w
+ def k():
+ pass
+ original_k = k
+ k = f(f(k))
+ assert hasattr(k, '__wrapped__')
+ k = k.__wrapped__
+ assert hasattr(k, '__wrapped__')
+ k = k.__wrapped__
+ assert k is original_k
+ assert not hasattr(k, '__wrapped__')
+
+
def test_add_metaclass():
class Meta(type):
pass