|
| 1 | +import sys |
1 | 2 | import textwrap |
2 | 3 | import unittest |
3 | 4 | from test import support |
@@ -234,26 +235,41 @@ def test_decodeescape(self): |
234 | 235 |
|
235 | 236 | def test_resize(self): |
236 | 237 | """Test _PyBytes_Resize()""" |
237 | | - resize = _testcapi.bytes_resize |
| 238 | + _resize = _testcapi.bytes_resize |
| 239 | + |
| 240 | + def resize(obj, size, new): |
| 241 | + result = _resize(obj, size, new) |
| 242 | + if 1 <= len(result): |
| 243 | + if new or size != len(obj): |
| 244 | + # gh-156995: Make sure that the result is a fresh object. |
| 245 | + # Previously, _PyBytes_Resize(&obj, 1) returned a singleton |
| 246 | + # if _PyObject_IsUniquelyReferenced() is false. |
| 247 | + self.assertEqual(sys.getrefcount(result), 1) |
| 248 | + self.assertFalse(sys._is_immortal(result)) |
| 249 | + else: |
| 250 | + # check that the result is the empty bytes string singleton |
| 251 | + self.assertTrue(sys._is_immortal(result)) |
| 252 | + return result |
238 | 253 |
|
239 | 254 | for new in True, False: |
240 | | - self.assertEqual(resize(b'abc', 0, new), b'') |
241 | | - self.assertEqual(resize(b'abc', 1, new), b'a') |
242 | | - self.assertEqual(resize(b'abc', 2, new), b'ab') |
243 | | - self.assertEqual(resize(b'abc', 3, new), b'abc') |
244 | | - b = resize(b'abc', 4, new) |
245 | | - self.assertEqual(len(b), 4) |
246 | | - self.assertEqual(b[:3], b'abc') |
247 | | - |
248 | | - self.assertEqual(resize(b'a', 0, new), b'') |
249 | | - self.assertEqual(resize(b'a', 1, new), b'a') |
250 | | - b = resize(b'a', 2, new) |
251 | | - self.assertEqual(len(b), 2) |
252 | | - self.assertEqual(b[:1], b'a') |
253 | | - |
254 | | - self.assertEqual(resize(b'', 0, new), b'') |
255 | | - self.assertEqual(len(resize(b'', 1, new)), 1) |
256 | | - self.assertEqual(len(resize(b'', 2, new)), 2) |
| 255 | + with self.subTest(new=new): |
| 256 | + self.assertEqual(resize(b'abc', 0, new), b'') |
| 257 | + self.assertEqual(resize(b'abc', 1, new), b'a') |
| 258 | + self.assertEqual(resize(b'abc', 2, new), b'ab') |
| 259 | + self.assertEqual(resize(b'abc', 3, new), b'abc') |
| 260 | + b = resize(b'abc', 4, new) |
| 261 | + self.assertEqual(len(b), 4) |
| 262 | + self.assertEqual(b[:3], b'abc') |
| 263 | + |
| 264 | + self.assertEqual(resize(b'a', 0, new), b'') |
| 265 | + self.assertEqual(resize(b'a', 1, new), b'a') |
| 266 | + b = resize(b'a', 2, new) |
| 267 | + self.assertEqual(len(b), 2) |
| 268 | + self.assertEqual(b[:1], b'a') |
| 269 | + |
| 270 | + self.assertEqual(resize(b'', 0, new), b'') |
| 271 | + self.assertEqual(len(resize(b'', 1, new)), 1) |
| 272 | + self.assertEqual(len(resize(b'', 2, new)), 2) |
257 | 273 |
|
258 | 274 | self.assertRaises(SystemError, resize, b'abc', -1, False) |
259 | 275 | self.assertRaises(SystemError, resize, bytearray(b'abc'), 3, False) |
|
0 commit comments