fix tests

Author: picnixz

Lib/test/test_pyexpat.py

@@ -825,7 +825,7 @@ def start_element(name, _):
 
 class AttackProtectionTest(unittest.TestCase):
 
-    def billion_laughs(self, ncols, nrows, text='.'):
+    def billion_laughs(self, ncols, nrows, text='.', indent=' '):
         """Create a billion laugh payload.
 
         Be careful: the number of total items is pow(n, k), thereby
@@ -834,26 +834,28 @@ def billion_laughs(self, ncols, nrows, text='.'):
         body = textwrap.indent('\n'.join(
             f'<!ENTITY row{i + 1} "{f"&row{i};" * ncols}">'
             for i in range(nrows)
-        ), '  ')
+        ), indent)
         return f"""\
 <?xml version="1.0"?>
 <!DOCTYPE doc [
-  <!ENTITY row0 "{text}">
-  <!ELEMENT doc (#PCDATA)>
+{indent}<!ENTITY row0 "{text}">
+{indent}<!ELEMENT doc (#PCDATA)>
 {body}
 ]>
 <doc>&row{nrows};</doc>
 """
 
     def test_set_alloc_tracker_maximum_amplification(self):
-        payload = self.billion_laughs(10, 4)
+        # Run the test with EXPAT_MALLOC_DEBUG=2 to find the max factor.
+        # The max factor depends on the "indent" and should be above 100.0.
+        # For this test, the peak amplification factor is 101.71.
+        payload = self.billion_laughs(1, 2)
 
         p = expat.ParserCreate()
         # Unconditionally enable maximum amplification factor.
         p.SetAllocTrackerActivationThreshold(0)
-        # At runtime, the peak amplification factor is 101.71,
-        # which is above the default threshold (100.0).
-        msg = re.escape("out of memory: line 3, column 15")
+        # At runtime, the peak amplification factor is above the default.
+        msg = r"out of memory: line \d+, column \d+"
         self.assertRaisesRegex(expat.ExpatError, msg, p.Parse, payload)
 
         # # Re-create a parser as the current parser is now in an error state.
@@ -894,7 +896,7 @@ def test_set_alloc_tracker_activation_threshold(self):
         p.SetAllocTrackerActivationThreshold(MIN_ALLOC - 1)
         # Check that we always reach the activation threshold.
         self.assertIsNone(p.SetAllocTrackerMaximumAmplification(1.0))
-        msg = re.escape("out of memory: line 3, column 10")
+        msg = r"out of memory: line \d+, column \d+"
         self.assertRaisesRegex(expat.ExpatError, msg, p.Parse, payload)
 
     def test_set_alloc_tracker_activation_threshold_invalid_args(self):