Use a context manager around memio in test_peek

Author: marcelm

Lib/test/test_memoryio.py

@@ -519,21 +519,20 @@ def test_relative_seek(self):
 
     def test_peek(self):
         buf = self.buftype("1234567890")
-        memio = self.ioclass(buf)
-        self.assertEqual(memio.tell(), 0)
-        self.assertEqual(memio.peek(1), buf[:1])
-        self.assertEqual(memio.peek(1), buf[:1])
-        self.assertEqual(memio.peek(), buf[:1])
-        self.assertEqual(memio.peek(0), buf)
-        self.assertEqual(memio.tell(), 0)
-        memio.read(1)
-        self.assertEqual(memio.peek(1), buf[1:2])
-        self.assertEqual(memio.peek(), buf[1:2])
-        self.assertEqual(memio.peek(0), buf[1:])
-        self.assertEqual(memio.peek(42), buf[1:])
-        memio.read()
-        self.assertEqual(memio.peek(1), self.EOF)
-        memio.close()
+        with self.ioclass(buf) as memio:
+            self.assertEqual(memio.tell(), 0)
+            self.assertEqual(memio.peek(1), buf[:1])
+            self.assertEqual(memio.peek(1), buf[:1])
+            self.assertEqual(memio.peek(), buf[:1])
+            self.assertEqual(memio.peek(0), buf)
+            self.assertEqual(memio.tell(), 0)
+            memio.read(1)
+            self.assertEqual(memio.peek(1), buf[1:2])
+            self.assertEqual(memio.peek(), buf[1:2])
+            self.assertEqual(memio.peek(0), buf[1:])
+            self.assertEqual(memio.peek(42), buf[1:])
+            memio.read()
+            self.assertEqual(memio.peek(1), self.EOF)
         self.assertRaises(ValueError, memio.peek)
 
     def test_unicode(self):