diff options
author | Tim Orling <ticotimo@gmail.com> | 2019-12-15 17:33:24 -0800 |
---|---|---|
committer | Tim Orling <ticotimo@gmail.com> | 2019-12-15 17:33:24 -0800 |
commit | 44c5317773acd0230c2cb9f86e4e6eb6db4e3cb7 (patch) | |
tree | b14bde848b78b4b50b32b3bd1703a17c8b6237bc | |
parent | bad41222379102e3f18f6f2f7be3ee608de6fbff (diff) | |
download | python-memcached-44c5317773acd0230c2cb9f86e4e6eb6db4e3cb7.tar.gz |
test_memcache.py: try import unittest.mock
Rather than always require the separate mock module,
try to use unittest.mock if it is available
Signed-off-by: Tim Orling <ticotimo@gmail.com>
-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 |