summaryrefslogtreecommitdiff
path: root/python/tests/compat.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/tests/compat.py')
-rw-r--r--python/tests/compat.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/python/tests/compat.py b/python/tests/compat.py
new file mode 100644
index 0000000000000..50138262e9b7b
--- /dev/null
+++ b/python/tests/compat.py
@@ -0,0 +1,8 @@
+try:
+ import unittest2 as unittest
+except ImportError:
+ import unittest
+
+# Python 2.7 - 3.1
+if not hasattr(unittest.TestCase, 'assertRaisesRegex'):
+ unittest.TestCase.assertRaisesRegex = unittest.TestCase.assertRaisesRegexp