@@ -575,27 +575,30 @@ def test_peek(self):
575575 self .assertEqual (memio .peek (), buf [:1 ])
576576 self .assertEqual (memio .peek (3 ), buf [:3 ])
577577 self .assertEqual (memio .peek (5 ), buf [:5 ])
578- self .assertEqual (memio .peek (0 ), b"" )
578+ self .assertEqual (memio .peek (0 ), buf )
579579 self .assertEqual (memio .peek (len (buf ) + 100 ), buf )
580580 self .assertEqual (memio .peek (- 1 ), buf )
581581 self .assertEqual (memio .tell (), 0 )
582+
582583 memio .read (1 )
583584 self .assertEqual (memio .tell (), 1 )
584585 self .assertEqual (memio .peek (1 ), buf [1 :2 ])
585586 self .assertEqual (memio .peek (), buf [1 :2 ])
586587 self .assertEqual (memio .peek (3 ), buf [1 :4 ])
587588 self .assertEqual (memio .peek (5 ), buf [1 :6 ])
588- self .assertEqual (memio .peek (0 ), b"" )
589+ self .assertEqual (memio .peek (0 ), buf [ 1 :] )
589590 self .assertEqual (memio .peek (len (buf ) + 100 ), buf [1 :])
590591 self .assertEqual (memio .peek (- 1 ), buf [1 :])
591592 self .assertEqual (memio .tell (), 1 )
593+
592594 memio .read ()
593595 self .assertEqual (memio .tell (), len (buf ))
594596 self .assertEqual (memio .peek (1 ), self .EOF )
595597 self .assertEqual (memio .peek (3 ), self .EOF )
596598 self .assertEqual (memio .peek (5 ), self .EOF )
597599 self .assertEqual (memio .peek (0 ), b"" )
598600 self .assertEqual (memio .tell (), len (buf ))
601+
599602 # Peeking works after writing
600603 abc = self .buftype ("abc" )
601604 memio .write (abc )
0 commit comments