Skip to content

Commit cb6a662

Browse files
b9788213sobolevn
andauthored
gh-143866: Verify return value of pathlib.write_{bytes,text} methods in tests (#143870)
Co-authored-by: sobolevn <mail@sobolevn.me>
1 parent d8ab1c7 commit cb6a662

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

Lib/test/test_pathlib/test_write.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -65,15 +65,17 @@ def test_open_wb(self):
6565

6666
def test_write_bytes(self):
6767
p = self.root / 'fileA'
68-
p.write_bytes(b'abcdefg')
69-
self.assertEqual(self.ground.readbytes(p), b'abcdefg')
68+
data = b'abcdefg'
69+
self.assertEqual(len(data), p.write_bytes(data))
70+
self.assertEqual(self.ground.readbytes(p), data)
7071
# Check that trying to write str does not truncate the file.
7172
self.assertRaises(TypeError, p.write_bytes, 'somestr')
72-
self.assertEqual(self.ground.readbytes(p), b'abcdefg')
73+
self.assertEqual(self.ground.readbytes(p), data)
7374

7475
def test_write_text(self):
7576
p = self.root / 'fileA'
76-
p.write_text('äbcdefg', encoding='latin-1')
77+
data = 'äbcdefg'
78+
self.assertEqual(len(data), p.write_text(data, encoding='latin-1'))
7779
self.assertEqual(self.ground.readbytes(p), b'\xe4bcdefg')
7880
# Check that trying to write bytes does not truncate the file.
7981
self.assertRaises(TypeError, p.write_text, b'somebytes', encoding='utf-8')

0 commit comments

Comments
 (0)