Backwards compatibility fix for 3.8 and earlier.

testing-cabal · Jan 9, 2023 · 5219601 · 5219601
1 parent 18046f1
commit 5219601
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/mock/mock.py b/mock/mock.py
@@ -1070,9 +1070,16 @@ def _calls_repr(self, prefix="Calls"):
         return f"\n{prefix}: {safe_repr(self.mock_calls)}."
 
 
+try:
+    removeprefix = str.removeprefix
+except AttributeError:
+    # Py 3.8 and earlier:
+    def removeprefix(name, prefix):
+        return name[len(prefix):]
+
 # Denylist for forbidden attribute names in safe mode
 _ATTRIB_DENY_LIST = frozenset({
-    name.removeprefix("assert_")
+    removeprefix(name, "assert_")
     for name in dir(NonCallableMock)
     if name.startswith("assert_")
 })