root/imapclient/test/util.py

# Copyright (c) 2012, Menno Smits
# Released subject to the New BSD License
# Please see http://en.wikipedia.org/wiki/BSD_licenses

def find_unittest2():
    import unittest
    if hasattr(unittest, 'skip') and hasattr(unittest, 'loader'):
        return unittest    # unittest from stdlib is unittest2, use that
    try:
        import unittest2   # try for a separately installed unittest2 package
    except ImportError:
        raise ImportError('unittest2 not installed and unittest in standard library is not unittest2')
    else:
        return unittest2

unittest = find_unittest2()