Update references to mock in Iterator tests

sigmavirus24 · sigmavirus24 · commit 965f0bd0d42f · 2014-05-04T14:00:30.000-05:00
diff --git a/tests/test_structs.py b/tests/test_structs.py
@@ -1,10 +1,6 @@
 import github3
 from github3.structs import GitHubIterator
-from tests.utils import BaseCase
-try:
-   from unittest.mock import patch
-except ImportError:
-   from mock import patch
+from tests.utils import BaseCase, mock
 
 
 class TestGitHubIterator(BaseCase):
@@ -74,7 +70,7 @@ def test_count_reaches_0(self):
         self.mock_assertions()
 
     def test_refresh(self):
-        with patch.object(GitHubIterator, '__iter__') as i:
+        with mock.patch.object(GitHubIterator, '__iter__') as i:
             self.i.refresh()
             i.__iter__.assert_called()
 
