Skip to content

Commit d6691b8

Browse files
committed
Use a context manager around memio in test_peek
1 parent d214089 commit d6691b8

1 file changed

Lines changed: 14 additions & 15 deletions

File tree

Lib/test/test_io/test_memoryio.py

Lines changed: 14 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)