Skip to content

Commit b8e23da

Browse files
authored
gh-155742: Test PyBytesWriter_FinishWithSize() with negative size (#155783)
Add test_finish_with_size() to test_capi.test_bytes.
1 parent 079dd5e commit b8e23da

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

Lib/test/test_capi/test_bytes.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -315,10 +315,16 @@ def test_create(self):
315315
self.assertEqual(writer.get_size(), 3)
316316
self.assertEqual(writer.finish(), self.result_type(b'abc'))
317317

318+
def test_finish_with_size(self):
319+
# Test PyBytesWriter_FinishWithSize()
318320
writer = self.create_writer(10, b'abc')
319321
self.assertEqual(writer.get_size(), 10)
320322
self.assertEqual(writer.finish_with_size(3), self.result_type(b'abc'))
321323

324+
writer = self.create_writer(3, b'abc')
325+
with self.assertRaises(SystemError):
326+
writer.finish_with_size(-3)
327+
322328
def test_write_bytes(self):
323329
# Test PyBytesWriter_WriteBytes()
324330
writer = self.create_writer()

0 commit comments

Comments
 (0)