Skip to content

Commit

Permalink
Adjust classmethod test for Python 3.9+.
Browse files Browse the repository at this point in the history
  • Loading branch information
GrahamDumpleton committed Aug 3, 2021
1 parent 76596ca commit 945a980
Showing 1 changed file with 19 additions and 4 deletions.
23 changes: 19 additions & 4 deletions tests/test_outer_classmethod.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
from __future__ import print_function

import sys
import unittest
import inspect
import imp
Expand All @@ -8,6 +9,8 @@

from compat import PY2, PY3, exec_

PYXY = tuple(sys.version_info[:2])

DECORATORS_CODE = """
import wrapt
Expand Down Expand Up @@ -133,11 +136,17 @@ def test_class_call_function(self):

@wrapt.decorator
def _decorator(wrapped, instance, args, kwargs):
self.assertEqual(instance, None)
self.assertEqual(args, (Class,)+_args)
if PYXY < (3, 9):
self.assertEqual(instance, None)
self.assertEqual(args, (Class,)+_args)
else:
self.assertEqual(instance, Class)
self.assertEqual(args, _args)

self.assertEqual(kwargs, _kwargs)
self.assertEqual(wrapped.__module__, _function.__module__)
self.assertEqual(wrapped.__name__, _function.__name__)

return wrapped(*args, **kwargs)

@_decorator
Expand Down Expand Up @@ -167,11 +176,17 @@ def test_instance_call_function(self):

@wrapt.decorator
def _decorator(wrapped, instance, args, kwargs):
self.assertEqual(instance, None)
self.assertEqual(args, (Class,)+_args)
if PYXY < (3, 9):
self.assertEqual(instance, None)
self.assertEqual(args, (Class,)+_args)
else:
self.assertEqual(instance, Class)
self.assertEqual(args, _args)

self.assertEqual(kwargs, _kwargs)
self.assertEqual(wrapped.__module__, _function.__module__)
self.assertEqual(wrapped.__name__, _function.__name__)

return wrapped(*args, **kwargs)

@_decorator
Expand Down

0 comments on commit 945a980

Please sign in to comment.