Provide assert(Not)Regex for Python 2.7

salilab · May 2, 2020 · df68f15 · df68f15
1 parent 2ce8719
commit df68f15
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/python/saliweb/test/__init__.py b/python/saliweb/test/__init__.py
@@ -88,8 +88,8 @@ def assertAlmostEqual(self, first, second, places=None, msg=None,
 # Provide assert(Not)Regex for Python 2 users (assertRegexpMatches is
 # deprecated in Python 3)
 if not hasattr(unittest.TestCase, 'assertRegex'):
-    assertRegex = unittest.TestCase.assertRegexpMatches
-    assertNotRegex = unittest.TestCase.assertNotRegexpMatches
+    unittest.TestCase.assertRegex = unittest.TestCase.assertRegexpMatches
+    unittest.TestCase.assertNotRegex = unittest.TestCase.assertNotRegexpMatches
 
 
 class RunInDir(object):

diff --git a/test/build/testutil.py b/test/build/testutil.py
@@ -94,8 +94,8 @@ def assertAlmostEqual(self, first, second, places=None, msg=None,
 # Provide assert(Not)Regex for Python 2 users (assertRegexpMatches is
 # deprecated in Python 3)
 if not hasattr(unittest.TestCase, 'assertRegex'):
-    assertRegex = unittest.TestCase.assertRegexpMatches
-    assertNotRegex = unittest.TestCase.assertNotRegexpMatches
+    unittest.TestCase.assertRegex = unittest.TestCase.assertRegexpMatches
+    unittest.TestCase.assertNotRegex = unittest.TestCase.assertNotRegexpMatches
 
 
 def run_catch_warnings(method, *args, **keys):