unittest or unittest2

test.py

import sys

if sys.version_info >= (2,7):
    from unittest import *
else:
    try:
        from unittest2 import *
    except ImportError:
        print('To run tests in python 2.6 you need to install\
 the unitest2 package')
        exit(0)