Only import mock in tests/utils.py

sigmavirus24 · sigmavirus24 · commit f04cc80620ae · 2014-05-04T13:49:46.000-05:00
diff --git a/tests/utils.py b/tests/utils.py
@@ -9,10 +9,12 @@
 
 import requests
 import github3
+
 try:
-   from unittest.mock import patch
+    from unittest import mock
 except ImportError:
-   from mock import patch
+    import mock
+
 from io import BytesIO
 from requests.structures import CaseInsensitiveDict
 
@@ -39,7 +41,7 @@ def setUp(self):
             self.g.login(token=os.environ['GH_AUTH'])
         self.args = ()
         self.conf = {'allow_redirects': True}
-        self.mock = patch.object(requests.sessions.Session, 'request')
+        self.mock = mock.patch.object(requests.sessions.Session, 'request')
         self.request = self.mock.start()
 
     def tearDown(self):
