Merged in harlowja/six (pull request #32)

Add a wraps helper

1 files changed, 18 insertions, 0 deletions

diff --git a/test_six.py b/test_six.py
index 766da9e..a520a83 100644
--- a/test_six.py
+++ b/test_six.py
@@ -637,6 +637,24 @@ def test_with_metaclass():
     assert X.__mro__ == (X, Base, Base2, object)
 
 
+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