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 <[email protected]>
linsomniac · Dec 16, 2019 · 44c5317 · 44c5317
1 parent bad4122
commit 44c5317
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_memcache.py b/tests/test_memcache.py
@@ -4,7 +4,10 @@
 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