* sumprod: L2742, L2752, L2772, L2775, L2779, L2783

skirpichev · skirpichev · commit 4e25c35e300f · 2023-09-28T07:24:24.000+03:00
diff --git a/Lib/test/test_math.py b/Lib/test/test_math.py
@@ -1299,6 +1299,12 @@ def testSumProd(self):
         self.assertRaises(ValueError, sumprod, [10, 20], [30])
         self.assertRaises(ValueError, sumprod, [10], [20, 30])
 
+        # Overflows
+        self.assertEqual(sumprod([10**20], [1]), 10**20)
+        self.assertEqual(sumprod([1], [10**20]), 10**20)
+        self.assertEqual(sumprod([10**10], [10**10]), 10**20)
+        self.assertEqual(sumprod([10**7]*10**5, [10**7]*10**5), 10**19)
+
         # Error in iterator
         def raise_after(n):
             for i in range(n):
@@ -1309,6 +1315,11 @@ def raise_after(n):
         with self.assertRaises(RuntimeError):
             sumprod(raise_after(5), range(10))
 
+        from test.test_iter import BasicIterClass
+
+        self.assertEqual(sumprod(BasicIterClass(1), [1]), 0)
+        self.assertEqual(sumprod([1], BasicIterClass(1)), 0)
+
         # Error in multiplication
         class BadMultiply:
             def __mul__(self, other):
