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 @@ -59,15 +59,17 @@ def test_open_wb(self):
5959
6060 def test_write_bytes (self ):
6161 p = self .root / 'fileA'
62- p .write_bytes (b'abcdefg' )
63- self .assertEqual (self .ground .readbytes (p ), b'abcdefg' )
62+ data = b'abcdefg'
63+ self .assertEqual (len (data ), p .write_bytes (data ))
64+ self .assertEqual (self .ground .readbytes (p ), data )
6465 # Check that trying to write str does not truncate the file.
6566 self .assertRaises (TypeError , p .write_bytes , 'somestr' )
66- self .assertEqual (self .ground .readbytes (p ), b'abcdefg' )
67+ self .assertEqual (self .ground .readbytes (p ), data )
6768
6869 def test_write_text (self ):
6970 p = self .root / 'fileA'
70- p .write_text ('äbcdefg' , encoding = 'latin-1' )
71+ data = 'äbcdefg'
72+ self .assertEqual (len (data ), p .write_text (data , encoding = 'latin-1' ))
7173 self .assertEqual (self .ground .readbytes (p ), b'\xe4 bcdefg' )
7274 # Check that trying to write bytes does not truncate the file.
7375 self .assertRaises (TypeError , p .write_text , b'somebytes' , encoding = 'utf-8' )
You can’t perform that action at this time.
0 commit comments