Update test_hmac.py

picnixz · web-flow · commit f93e1ba3e0e3 · 2025-07-21T16:48:45.000+02:00
diff --git a/Lib/test/test_hmac.py b/Lib/test/test_hmac.py
@@ -1505,14 +1505,12 @@ def test_with_fallback(self):
             cache.pop('foo')
 
     @hashlib_helper.requires_builtin_hashdigest("_md5", "md5")
-    @bigmemtest(size=_4G, memuse=2, dry_run=False)
+    @bigmemtest(size=_4G , memuse=2, dry_run=False)
     def test_hmac_digest_overflow_error_no_openssl(self, size):
         hmac = import_fresh_module("hmac", blocked=["_hashlib"])
 
-        UINT32_MAX = (1 << 32) - 1
-        self.assertEqual(UINT32_MAX, size - 1)
-        bigkey = b'K' * UINT32_MAX
-        bigmsg = b'M' * UINT32_MAX
+        bigkey = b'K' * size
+        bigmsg = b'M' * size
 
         with unittest.mock.patch.object(hmac, "_compute_digest_fallback") as f:
             self.assertRaises(OverflowError, hmac.digest, bigkey, b'm', "md5")
@@ -1525,18 +1523,17 @@ def test_hmac_digest_overflow_error_no_builtin(self, size):
         capi = import_helper.import_module("_testcapi")
         hmac = import_fresh_module("hmac", blocked=["_hmac"])
 
-        INT_MAX = capi.INT_MAX
-        self.assertLessEqual(INT_MAX, size)
-        bigkey = b'K' * INT_MAX
-        bigmsg = b'M' * INT_MAX
+        self.assertLess(capi.INT_MAX, size)
+        bigkey = b'K' * size
+        bigmsg = b'M' * size
 
         with unittest.mock.patch.object(hmac, "_compute_digest_fallback") as f:
             self.assertRaises(OverflowError, hmac.digest, bigkey, b'm', "md5")
             self.assertRaises(OverflowError, hmac.digest, b'k', bigmsg, "md5")
         f.assert_not_called()
 
     @hashlib_helper.requires_hashdigest("md5", openssl=True)
-    @bigmemtest(size=_4G + 1, memuse=2, dry_run=False)
+    @bigmemtest(size=_4G, memuse=2, dry_run=False)
     def test_hmac_digest_no_overflow_error_in_fallback(self, size):
         hmac = import_fresh_module("hmac", blocked=["_hashlib", "_hmac"])
 
