Issues #23363, #23364, #23365, #23366: Fixed itertools overflow tests.

Used PyMem_New to check overflow.
This commit is contained in:
Serhiy Storchaka 2015-02-03 01:34:09 +02:00
parent 1572944499
commit dee948b359
2 changed files with 11 additions and 27 deletions

View file

@ -260,7 +260,7 @@ class TestBasicOps(unittest.TestCase):
@support.bigaddrspacetest
def test_combinations_overflow(self):
with self.assertRaises(OverflowError):
with self.assertRaises((OverflowError, MemoryError)):
combinations("AA", 2**29)
# Test implementation detail: tuple re-use
@ -346,7 +346,7 @@ class TestBasicOps(unittest.TestCase):
@support.bigaddrspacetest
def test_combinations_with_replacement_overflow(self):
with self.assertRaises(OverflowError):
with self.assertRaises((OverflowError, MemoryError)):
combinations_with_replacement("AA", 2**30)
# Test implementation detail: tuple re-use
@ -420,10 +420,8 @@ class TestBasicOps(unittest.TestCase):
@support.bigaddrspacetest
def test_permutations_overflow(self):
with self.assertRaises(OverflowError):
with self.assertRaises((OverflowError, MemoryError)):
permutations("A", 2**30)
with self.assertRaises(OverflowError):
permutations("A", 2, 2**30)
@support.impl_detail("tuple resuse is CPython specific")
def test_permutations_tuple_reuse(self):
@ -939,8 +937,8 @@ class TestBasicOps(unittest.TestCase):
@support.bigaddrspacetest
def test_product_overflow(self):
with self.assertRaises(OverflowError):
product(["a"]*(2**16), repeat=2**16)
with self.assertRaises((OverflowError, MemoryError)):
product(*(['ab']*2**5), repeat=2**25)
@support.impl_detail("tuple reuse is specific to CPython")
def test_product_tuple_reuse(self):