diff --git a/tests/cache/tests.py b/tests/cache/tests.py index bf1543ce8a3..44524b07076 100644 --- a/tests/cache/tests.py +++ b/tests/cache/tests.py @@ -31,10 +31,14 @@ from django.utils import timezone from django.utils import translation from django.utils.cache import (patch_vary_headers, get_cache_key, learn_cache_key, patch_cache_control, patch_response_headers) -from django.utils.six.moves import cPickle as pickle from django.utils.encoding import force_text from django.views.decorators.cache import cache_page +try: # Use the same idiom as in cache backends + from django.utils.six.moves import cPickle as pickle +except ImportError: + import pickle + from .models import Poll, expensive_calculation