diff --git a/pymemcache/test/test_client.py b/pymemcache/test/test_client.py index 8b323ded..f0aac442 100644 --- a/pymemcache/test/test_client.py +++ b/pymemcache/test/test_client.py @@ -1,5 +1,4 @@ # Copyright 2012 Pinterest.com -# -*- coding: utf-8 -*- # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -20,8 +19,8 @@ import functools import json import os -import mock import platform +from unittest import mock import re import socket import sys diff --git a/pymemcache/test/test_client_hash.py b/pymemcache/test/test_client_hash.py index ad3f2dac..ceab7052 100644 --- a/pymemcache/test/test_client_hash.py +++ b/pymemcache/test/test_client_hash.py @@ -7,7 +7,7 @@ import unittest import os import pytest -import mock +from unittest import mock import socket diff --git a/pymemcache/test/test_client_retry.py b/pymemcache/test/test_client_retry.py index 230a941..de64a46 100644 --- a/pymemcache/test/test_client_retry.py +++ b/pymemcache/test/test_client_retry.py @@ -2,8 +2,8 @@ import functools import unittest +from unittest import mock -import mock import pytest from .test_client import ClientTestMixin, MockSocket