summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_array.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/test_array.py')
-rwxr-xr-xLib/test/test_array.py9
1 files changed, 5 insertions, 4 deletions
diff --git a/Lib/test/test_array.py b/Lib/test/test_array.py
index 34a8f79..2efbc7d 100755
--- a/Lib/test/test_array.py
+++ b/Lib/test/test_array.py
@@ -1011,20 +1011,21 @@ class DoubleTest(FPTest):
minitemsize = 8
def test_alloc_overflow(self):
+ from sys import maxsize
a = array.array('d', [-1]*65536)
try:
- a *= 65536
+ a *= maxsize//65536 + 1
except MemoryError:
pass
else:
- self.fail("a *= 2**16 didn't raise MemoryError")
+ self.fail("Array of size > maxsize created - MemoryError expected")
b = array.array('d', [ 2.71828183, 3.14159265, -1])
try:
- b * 1431655766
+ b * (maxsize//3 + 1)
except MemoryError:
pass
else:
- self.fail("a * 1431655766 didn't raise MemoryError")
+ self.fail("Array of size > maxsize created - MemoryError expected")
tests.append(DoubleTest)