Unittest2 support for Py2.6

setup.py

@@ -1,8 +1,13 @@
 #!/usr/bin/env python
 
+import sys
 from setuptools import setup
 from ts3 import __version__
 
+test_require = ['mock']
+if sys.version < '2.7':
+      test_require.append('unittest2')
+
 setup(
     name="python-ts3",
     version=__version__,
@@ -14,7 +19,7 @@ setup(
     packages=['ts3'],
     scripts=['examples/gents3privkey.py'],
     test_suite='ts3.test.suite',
-
+    test_require=test_require,
     classifiers=[
         'License :: OSI Approved :: BSD License',
         'Topic :: Internet',

ts3/test.py

@@ -1,4 +1,8 @@
-import unittest
+
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
 import socket
 import threading
 import time