diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/composite_pk/test_filter.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/tests/composite_pk/test_filter.py b/tests/composite_pk/test_filter.py index 03037d4d82..b88d02731c 100644 --- a/tests/composite_pk/test_filter.py +++ b/tests/composite_pk/test_filter.py @@ -14,6 +14,7 @@ from django.db.models import ( from django.db.models.functions import Cast from django.db.models.lookups import Exact from django.test import TestCase, skipUnlessDBFeature +from django.utils.version import PY311 from .models import Comment, Tenant, User @@ -556,4 +557,10 @@ class CompositePKFilterTupleLookupFallbackTests(CompositePKFilterTests): feature_patch = patch.object( connection.features, "supports_tuple_lookups", False ) - self.enterContext(feature_patch) + if PY311: + self.enterContext(feature_patch) + else: + # unittest.TestCase.enterContext() was added in Python 3.11. + from django.test.testcases import _enter_context + + _enter_context(feature_patch, self.addCleanup) |