Merge pull request #172 from moto-timo/try-import-unittest-mock

test_memcache.py: try import unittest.mock
linsomniac · Apr 15, 2023 · 8ded8c3 · 8ded8c3
2 parents d5fb419 + 44c5317
commit 8ded8c3
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