import sys


if sys.version_info < (2, 7):
    from tests import TestCase  # NOQA
else:
    from unittest import TestCase  # NOQA

import sys


if sys.version_info < (2, 7):
    from unittest2 import TestCase  # NOQA
else:
    from unittest import TestCase  # NOQA

Use right test case for py 2.6
Use right test case for py 2.6
