33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
|
# Copyright (c) 2014, Menno Smits
|
||
|
|
# Released subject to the New BSD License
|
||
|
|
# Please see http://en.wikipedia.org/wiki/BSD_licenses
|
||
|
|
|
||
|
|
from __future__ import unicode_literals
|
||
|
|
|
||
|
|
|
||
|
|
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()
|
||
|
|
|
||
|
|
|
||
|
|
def patch_TestCase():
|
||
|
|
TestCase = unittest.TestCase
|
||
|
|
# Older versions of unittest2 don't have
|
||
|
|
# TestCase.assertRaisesRegex and newer version raises warnings
|
||
|
|
# when you use assertRaisesRegexp. This helps deal with the
|
||
|
|
# mismatch.
|
||
|
|
if not hasattr(TestCase, 'assertRaisesRegex'):
|
||
|
|
TestCase.assertRaisesRegex = TestCase.assertRaisesRegexp
|
||
|
|
|
||
|
|
patch_TestCase()
|