Adding unittest2 for Python <= 2.6

1 files changed, 7 insertions, 1 deletions

diff --git a/tests.py b/tests.py
index 5661b8c..386275e 100644
--- a/tests.py
+++ b/tests.py
@@ -25,7 +25,8 @@ from jsonrpclib.SimpleJSONRPCServer import SimpleJSONRPCServer
 from jsonrpclib.SimpleJSONRPCServer import SimpleJSONRPCRequestHandler
 import socket
 import tempfile
-import unittest
+import sys
+
 import os
 try:
     import json
@@ -33,6 +34,11 @@ except ImportError:
     import simplejson as json
 from threading import Thread
 
+if sys.version_info < (2, 7):
+    import unittest2 as unittest
+else:
+    import unittest
+
 PORTS = range(8000, 8999)