Avoid deprecation warning due to renamed assertion method in Py3.2+

dysosmus · Dec 3, 2015 · b1e3eaf · b1e3eaf
1 parent 4c0ef6e
commit b1e3eaf
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/dulwich/tests/test_repository.py b/dulwich/tests/test_repository.py
@@ -125,8 +125,13 @@ def test_getitem_unicode(self):
         for k, contained in test_keys:
             self.assertEqual(k in r, contained)
 
+        # Avoid deprecation warning under Py3.2+
+        if hasattr(self, 'assertRaisesRegex'):
+            assertRaisesRegexp = self.assertRaisesRegex
+        else:
+            assertRaisesRegexp = self.assertRaisesRegexp
         for k, _ in test_keys:
-            self.assertRaisesRegexp(
+            assertRaisesRegexp(
                 TypeError, "'name' must be bytestring, not int",
                 r.__getitem__, 12
             )