Merge pull request #54 from robotadam/fix-unittests

Swap unittest2 for stdlib unittest so test work on modern python versions

Author: ShreyaThapa

dwollav2/test/__init__.py

@@ -1,13 +1,13 @@
 import os
-import unittest2
+import unittest
 
 
 def all():
     path = os.path.dirname(os.path.realpath(__file__))
-    return unittest2.defaultTestLoader.discover(path)
+    return unittest.defaultTestLoader.discover(path)
 
 
 def resources():
     path = os.path.dirname(os.path.realpath(__file__))
-    return unittest2.defaultTestLoader.discover(
+    return unittest.defaultTestLoader.discover(
         os.path.join(path, "resources"))

dwollav2/test/test_auth.py

@@ -1,4 +1,4 @@
-import unittest2
+import unittest
 import responses
 from mock import Mock
 try:
@@ -9,7 +9,7 @@
 import dwollav2
 
 
-class AuthShould(unittest2.TestCase):
+class AuthShould(unittest.TestCase):
     client = dwollav2.Client(id='id', secret='secret', on_grant=Mock())
     redirect_uri = 'redirect uri'
     scope = 'scope'

dwollav2/test/test_client.py

@@ -1,10 +1,10 @@
-import unittest2
+import unittest
 import responses
 
 import dwollav2
 
 
-class ClientShould(unittest2.TestCase):
+class ClientShould(unittest.TestCase):
     id = 'client-id'
     secret = 'client-secret'
 

dwollav2/test/test_error.py

@@ -1,10 +1,10 @@
-import unittest2
+import unittest
 import requests
 
 import dwollav2
 
 
-class ErrorShould(unittest2.TestCase):
+class ErrorShould(unittest.TestCase):
     def test_maps_string_to_generic_error(self):
         error = 'foo'
         with self.assertRaises(dwollav2.Error) as ecm:

dwollav2/test/test_response.py

@@ -1,10 +1,10 @@
-import unittest2
+import unittest
 import requests
 
 import dwollav2
 
 
-class ResponseShould(unittest2.TestCase):
+class ResponseShould(unittest.TestCase):
     def test_raises_error_if_over_400(self):
         res = requests.Response()
         res.status_code = 400

dwollav2/test/test_token.py

@@ -1,10 +1,10 @@
-import unittest2
+import unittest
 import responses
 
 import dwollav2
 
 
-class TokenShould(unittest2.TestCase):
+class TokenShould(unittest.TestCase):
     client = dwollav2.Client(id='id', secret='secret')
     access_token = 'access token'
     refresh_token = 'refresh token'