@@ -568,21 +568,20 @@ def test_issue141311(self):
568568
569569 def test_peek (self ):
570570 buf = self .buftype ("1234567890" )
571- memio = self .ioclass (buf )
572- self .assertEqual (memio .tell (), 0 )
573- self .assertEqual (memio .peek (1 ), buf [:1 ])
574- self .assertEqual (memio .peek (1 ), buf [:1 ])
575- self .assertEqual (memio .peek (), buf [:1 ])
576- self .assertEqual (memio .peek (0 ), buf )
577- self .assertEqual (memio .tell (), 0 )
578- memio .read (1 )
579- self .assertEqual (memio .peek (1 ), buf [1 :2 ])
580- self .assertEqual (memio .peek (), buf [1 :2 ])
581- self .assertEqual (memio .peek (0 ), buf [1 :])
582- self .assertEqual (memio .peek (42 ), buf [1 :])
583- memio .read ()
584- self .assertEqual (memio .peek (1 ), self .EOF )
585- memio .close ()
571+ with self .ioclass (buf ) as memio :
572+ self .assertEqual (memio .tell (), 0 )
573+ self .assertEqual (memio .peek (1 ), buf [:1 ])
574+ self .assertEqual (memio .peek (1 ), buf [:1 ])
575+ self .assertEqual (memio .peek (), buf [:1 ])
576+ self .assertEqual (memio .peek (0 ), buf )
577+ self .assertEqual (memio .tell (), 0 )
578+ memio .read (1 )
579+ self .assertEqual (memio .peek (1 ), buf [1 :2 ])
580+ self .assertEqual (memio .peek (), buf [1 :2 ])
581+ self .assertEqual (memio .peek (0 ), buf [1 :])
582+ self .assertEqual (memio .peek (42 ), buf [1 :])
583+ memio .read ()
584+ self .assertEqual (memio .peek (1 ), self .EOF )
586585 self .assertRaises (ValueError , memio .peek )
587586
588587 def test_unicode (self ):
0 commit comments