tests: Add unittest aliases to Python 2.7

2020-10-09 04:36:57 -04:00
parent 11b1f1c14e
commit cef0b48993
3 changed files with 23 additions and 0 deletions
--- a/tests/test_config.py
+++ b/tests/test_config.py
@@ -31,6 +31,15 @@ from yamllint import config


 class SimpleConfigTestCase(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        super(SimpleConfigTestCase, cls).setUpClass()
+
+        # https://docs.python.org/3/library/unittest.html#deprecated-aliases
+        if sys.version_info < (3, 2):
+            cls.assertRegex = cls.assertRegexpMatches
+            cls.assertRaisesRegex = cls.assertRaisesRegexp
+
    def test_parse_config(self):
        new = config.YamlLintConfig('rules:\n'
                                    '  colons:\n'