Use unittest2 on Python 2.6

pymysql/tests/__init__.py

@@ -6,5 +6,8 @@ import sys
 from pymysql.tests.thirdparty import *
 
 if __name__ == "__main__":
-    import unittest
+    try:
+        import unittest2 as unittest
+    except ImportError:
+        import unittest
     unittest.main()

pymysql/tests/thirdparty/__init__.py

@@ -1,5 +1,8 @@
 from .test_MySQLdb import *
 
 if __name__ == "__main__":
-    import unittest
+    try:
+        import unittest2 as unittest
+    except ImportError:
+        import unittest
     unittest.main()

pymysql/tests/thirdparty/test_MySQLdb/capabilities.py

@@ -8,7 +8,10 @@
 import sys
 from time import time
 import array
-import unittest
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 PY2 = sys.version_info[0] == 2
 

pymysql/tests/thirdparty/test_MySQLdb/dbapi20.py

@@ -15,7 +15,11 @@ __rcs_id__  = '$Id$'
 __version__ = '$Revision$'[11:-2]
 __author__ = 'Stuart Bishop <zen@shangri-la.dropbear.id.au>'
 
-import unittest
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
+
 import time
 
 # $Log$

pymysql/tests/thirdparty/test_MySQLdb/test_MySQLdb_capabilities.py

@@ -1,6 +1,9 @@
 #!/usr/bin/env python
 from . import capabilities
-import unittest
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 import pymysql
 from pymysql.tests import base
 import warnings

pymysql/tests/thirdparty/test_MySQLdb/test_MySQLdb_nonstandard.py

@@ -1,5 +1,8 @@
 import sys
-import unittest
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 
 import pymysql
 _mysql = pymysql