Fix attrib decorator (fixes bpython#553)

sebastinas · sebastinas · commit d9aec6769076 · 2015-08-02T21:12:46.000+02:00
Signed-off-by: Sebastian Ramacher &lt;sebastian+dev@ramacher.at&gt;
diff --git a/bpython/test/test_args.py b/bpython/test/test_args.py
@@ -14,8 +14,10 @@
 try:
     from nose.plugins.attrib import attr
 except ImportError:
-    def attr(func, *args, **kwargs):
-        return func
+    def attr(*args, **kwargs):
+        def identity(func):
+            return func
+        return identity
 
 
 @attr(speed='slow')
diff --git a/bpython/test/test_crashers.py b/bpython/test/test_crashers.py
@@ -27,8 +27,10 @@ class TrialTestCase(object):
 try:
     from nose.plugins.attrib import attr
 except ImportError:
-    def attr(func, *args, **kwargs):
-        return func
+    def attr(*args, **kwargs):
+        def identity(func):
+            return func
+        return identity
 
 TEST_CONFIG = os.path.join(os.path.dirname(__file__), "test.config")
 
