mirror of
				https://github.com/python/cpython.git
				synced 2025-10-26 11:14:33 +00:00 
			
		
		
		
	 3cbc23aa22
			
		
	
	
		3cbc23aa22
		
	
	
	
	
		
			
			Since PEP 393 in Python 3.3, this value is always 0x10ffff, the maximum codepoint in Unicode; there's no longer such a thing as a UCS-2 build of Python, which couldn't properly represent some characters. There are a couple of spots left where we still condition on the value of this constant. Take them out.
		
			
				
	
	
		
			101 lines
		
	
	
	
		
			2.9 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			101 lines
		
	
	
	
		
			2.9 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 GiB 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.
 | |
|         try:
 | |
|             x = b"x" * (MAX_Py_ssize_t - 128)
 | |
|             self.assertRaises(OverflowError, operator.add, x, b"x" * 128)
 | |
|         finally:
 | |
|             x = None
 | |
| 
 | |
|     @bigaddrspacetest
 | |
|     def test_optimized_concat(self):
 | |
|         try:
 | |
|             x = b"x" * (MAX_Py_ssize_t - 128)
 | |
| 
 | |
|             with self.assertRaises(OverflowError) as cm:
 | |
|                 # this statement used a fast path in ceval.c
 | |
|                 x = x + b"x" * 128
 | |
| 
 | |
|             with self.assertRaises(OverflowError) as cm:
 | |
|                 # this statement used a fast path in ceval.c
 | |
|                 x +=  b"x" * 128
 | |
|         finally:
 | |
|             x = None
 | |
| 
 | |
|     @bigaddrspacetest
 | |
|     def test_repeat(self):
 | |
|         try:
 | |
|             x = b"x" * (MAX_Py_ssize_t - 128)
 | |
|             self.assertRaises(OverflowError, operator.mul, x, 128)
 | |
|         finally:
 | |
|             x = None
 | |
| 
 | |
| 
 | |
| class StrTest(unittest.TestCase):
 | |
| 
 | |
|     unicodesize = 4
 | |
| 
 | |
|     @bigaddrspacetest
 | |
|     def test_concat(self):
 | |
|         try:
 | |
|             # Create a string that would fill almost the address space
 | |
|             x = "x" * int(MAX_Py_ssize_t // (1.1 * 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)
 | |
|         finally:
 | |
|             x = None
 | |
| 
 | |
|     @bigaddrspacetest
 | |
|     def test_optimized_concat(self):
 | |
|         try:
 | |
|             x = "x" * int(MAX_Py_ssize_t // (1.1 * 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
 | |
|         finally:
 | |
|             x = None
 | |
| 
 | |
|     @bigaddrspacetest
 | |
|     def test_repeat(self):
 | |
|         try:
 | |
|             x = "x" * int(MAX_Py_ssize_t // (1.1 * self.unicodesize))
 | |
|             self.assertRaises(MemoryError, operator.mul, x, 2)
 | |
|         finally:
 | |
|             x = None
 | |
| 
 | |
| 
 | |
| def test_main():
 | |
|     support.run_unittest(BytesTest, StrTest)
 | |
| 
 | |
| if __name__ == '__main__':
 | |
|     if len(sys.argv) > 1:
 | |
|         support.set_memlimit(sys.argv[1])
 | |
|     test_main()
 |