diff options
| -rw-r--r-- | tests/test_memcache.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/test_memcache.py b/tests/test_memcache.py index 40b6524..621ac5c 100644 --- a/tests/test_memcache.py +++ b/tests/test_memcache.py @@ -4,7 +4,10 @@ from __future__ import print_function import unittest import zlib -import mock +try: + import unittest.mock as mock +except ImportError: + import mock from memcache import Client, _Host, SERVER_MAX_KEY_LENGTH, SERVER_MAX_VALUE_LENGTH # noqa: H301 from .utils import captured_stderr |