File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed
Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff 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'\xe4 bcdefg' )
7880 # Check that trying to write bytes does not truncate the file.
7981 self .assertRaises (TypeError , p .write_text , b'somebytes' , encoding = 'utf-8' )
You can’t perform that action at this time.
0 commit comments