Python 2.6 compatibility

3 files changed, 6 insertions, 8 deletions

diff --git a/pygithub3/tests/services/test_core.py b/pygithub3/tests/services/test_core.py
index 612b5ea..8ee1c6e 100644
--- a/pygithub3/tests/services/test_core.py
+++ b/pygithub3/tests/services/test_core.py
@@ -1,11 +1,11 @@
 #!/usr/bin/env python
 # -*- encoding: utf-8 -*-
 
-from unittest import TestCase
-from mock import patch
-
 import requests
 
+from mock import patch
+
+from pygithub3.tests.utils.core import TestCase
 from pygithub3.services.base import Service, MimeTypeMixin
 from pygithub3.core.result import Result
 from pygithub3.tests.utils.base import DummyRequest, mock_response
@@ -68,7 +68,7 @@ class TestMimeType(TestCase):
 
     def test_WITH_mimetype(self, request_method):
         request_method.return_value = mock_response()
-        self.ms.set_html_mimetype()
+        self.ms.set_html()
         self.ms.dummy_request()
         request_method.assert_called_with('get', _('dummyrequest'),
             headers={'Accept': 'application/vnd.github.%s.html+json' %
diff --git a/pygithub3/tests/services/test_repos.py b/pygithub3/tests/services/test_repos.py
index 865c5df..1993340 100644
--- a/pygithub3/tests/services/test_repos.py
+++ b/pygithub3/tests/services/test_repos.py
@@ -1,11 +1,10 @@
 #!/usr/bin/env python
 # -*- encoding: utf-8 -*-
 
-from unittest import TestCase
-
 import requests
 from mock import patch, Mock
 
+from pygithub3.tests.utils.core import TestCase
 from pygithub3.services.repos import (Repos, Collaborators, Commits, Downloads,
                                       Forks, Keys, Watchers)
 from pygithub3.resources.base import json
diff --git a/pygithub3/tests/services/test_users.py b/pygithub3/tests/services/test_users.py
index 049197b..f777d6d 100644
--- a/pygithub3/tests/services/test_users.py
+++ b/pygithub3/tests/services/test_users.py
@@ -1,11 +1,10 @@
 #!/usr/bin/env python
 # -*- encoding: utf-8 -*-
 
-from unittest import TestCase
-
 import requests
 from mock import patch, Mock
 
+from pygithub3.tests.utils.core import TestCase
 from pygithub3.core.client import Client
 from pygithub3.services.users import User, Emails, Followers, Keys
 from pygithub3.exceptions import ValidationError