mirror of https://github.com/python/cpython
84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
"""
|
|
These tests are meant to exercise that requests to create objects bigger
|
|
than what the address space allows are properly met with an OverflowError
|
|
(rather than crash weirdly).
|
|
|
|
Primarily, this means 32-bit builds with at least 2 GB of available memory.
|
|
You need to pass the -M option to regrtest (e.g. "-M 2.1G") for tests to
|
|
be enabled.
|
|
"""
|
|
|
|
from test import support
|
|
from test.support import bigaddrspacetest, MAX_Py_ssize_t
|
|
|
|
import unittest
|
|
import operator
|
|
import sys
|
|
|
|
|
|
class BytesTest(unittest.TestCase):
|
|
|
|
@bigaddrspacetest
|
|
def test_concat(self):
|
|
# Allocate a bytestring that's near the maximum size allowed by
|
|
# the address space, and then try to build a new, larger one through
|
|
# concatenation.
|
|
x = b"x" * (MAX_Py_ssize_t - 128)
|
|
self.assertRaises(OverflowError, operator.add, x, b"x" * 128)
|
|
|
|
@bigaddrspacetest
|
|
def test_optimized_concat(self):
|
|
x = b"x" * (MAX_Py_ssize_t - 128)
|
|
|
|
with self.assertRaises(OverflowError) as cm:
|
|
# this statement uses a fast path in ceval.c
|
|
x = x + b"x" * 128
|
|
|
|
with self.assertRaises(OverflowError) as cm:
|
|
# this statement uses a fast path in ceval.c
|
|
x += b"x" * 128
|
|
|
|
@bigaddrspacetest
|
|
def test_repeat(self):
|
|
x = b"x" * (MAX_Py_ssize_t - 128)
|
|
self.assertRaises(OverflowError, operator.mul, x, 128)
|
|
|
|
|
|
class StrTest(unittest.TestCase):
|
|
|
|
unicodesize = 2 if sys.maxunicode < 65536 else 4
|
|
|
|
@bigaddrspacetest
|
|
def test_concat(self):
|
|
# Create a string half the size that would fill the address space
|
|
x = "x" * (MAX_Py_ssize_t // (2 * self.unicodesize))
|
|
# Unicode objects trigger MemoryError in case an operation that's
|
|
# going to cause a size overflow is executed
|
|
self.assertRaises(MemoryError, operator.add, x, x)
|
|
|
|
@bigaddrspacetest
|
|
def test_optimized_concat(self):
|
|
x = "x" * (MAX_Py_ssize_t // (2 * self.unicodesize))
|
|
|
|
with self.assertRaises(MemoryError) as cm:
|
|
# this statement uses a fast path in ceval.c
|
|
x = x + x
|
|
|
|
with self.assertRaises(MemoryError) as cm:
|
|
# this statement uses a fast path in ceval.c
|
|
x += x
|
|
|
|
@bigaddrspacetest
|
|
def test_repeat(self):
|
|
x = "x" * (MAX_Py_ssize_t // (2 * self.unicodesize))
|
|
self.assertRaises(MemoryError, operator.mul, x, 2)
|
|
|
|
|
|
def test_main():
|
|
support.run_unittest(BytesTest, StrTest)
|
|
|
|
if __name__ == '__main__':
|
|
if len(sys.argv) > 1:
|
|
support.set_memlimit(sys.argv[1])
|
|
test_main()
|