no need to use anything else than unittest2 when Py2.6

Author: Bartek Ogryczak

pymysqlreplication/tests/__init__.py

@@ -5,5 +5,8 @@
 from pymysqlreplication.tests.test_data_objects import *
 
 if __name__ == "__main__":
-    import unittest
+    if sys.version_info < (2, 7):
+        import unittest2 as unittest
+    else:
+        import unittest
     unittest.main()

pymysqlreplication/tests/base.py

@@ -1,18 +1,17 @@
 # -*- coding: utf-8 -*-
 
 import pymysql
-import unittest
 import copy
 from pymysqlreplication import BinLogStreamReader
 import os
 import sys
 
-(major, minor, _, _, _) = sys.version_info
-if (major, minor) < (2, 7):
-    import unittest2
-    base = unittest2.TestCase
+if sys.version_info < (2, 7):
+    import unittest2 as unittest
 else:
-    base = unittest.TestCase
+    import unittest
+
+base = unittest.TestCase
 
 
 class PyMySQLReplicationTestCase(base):

pymysqlreplication/tests/test_basic.py

@@ -1,7 +1,10 @@
 # -*- coding: utf-8 -*-
-import unittest
-
 import time
+import sys
+if sys.version_info < (2, 7):
+    import unittest2 as unittest
+else:
+    import unittest
 
 from pymysqlreplication.tests import base
 from pymysqlreplication import BinLogStreamReader

pymysqlreplication/tests/test_data_objects.py

@@ -1,4 +1,8 @@
-import unittest
+import sys
+if sys.version_info < (2, 7):
+    import unittest2 as unittest
+else:
+    import unittest
 
 from pymysqlreplication.column import Column
 from pymysqlreplication.table import Table

pymysqlreplication/tests/test_data_type.py

@@ -2,7 +2,11 @@
 
 import copy
 import platform
-import unittest
+import sys
+if sys.version_info < (2, 7):
+    import unittest2 as unittest
+else:
+    import unittest
 
 from decimal import Decimal
 

setup.py

@@ -11,8 +11,7 @@
 install_requires = ['pymysql']
 
 # add unittest2 to install_requires for python < 2.7
-major, minor, _, _, _ = sys.version_info
-if (major, minor) < (2, 7):
+if sys.version_info < (2, 7):
     install_requires.append("unittest2")