From 44c5317773acd0230c2cb9f86e4e6eb6db4e3cb7 Mon Sep 17 00:00:00 2001 From: Tim Orling Date: Sun, 15 Dec 2019 17:33:24 -0800 Subject: 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 --- tests/test_memcache.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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 -- cgit v1.2.1