Test peek(3) and peek(5)

Author: marcelm

Lib/test/test_io/test_memoryio.py

@@ -573,6 +573,8 @@ def test_peek(self):
             self.assertEqual(memio.peek(1), buf[:1])
             self.assertEqual(memio.peek(1), buf[:1])
             self.assertEqual(memio.peek(), buf[:1])
+            self.assertEqual(memio.peek(3), buf[:3])
+            self.assertEqual(memio.peek(5), buf[:5])
             self.assertEqual(memio.peek(0), b"")
             self.assertEqual(memio.peek(len(buf) + 100), buf)
             self.assertEqual(memio.peek(-1), buf)
@@ -581,13 +583,17 @@ def test_peek(self):
             self.assertEqual(memio.tell(), 1)
             self.assertEqual(memio.peek(1), buf[1:2])
             self.assertEqual(memio.peek(), buf[1:2])
+            self.assertEqual(memio.peek(3), buf[1:4])
+            self.assertEqual(memio.peek(5), buf[1:6])
             self.assertEqual(memio.peek(0), b"")
             self.assertEqual(memio.peek(len(buf) + 100), buf[1:])
             self.assertEqual(memio.peek(-1), buf[1:])
             self.assertEqual(memio.tell(), 1)
             memio.read()
             self.assertEqual(memio.tell(), len(buf))
             self.assertEqual(memio.peek(1), self.EOF)
+            self.assertEqual(memio.peek(3), self.EOF)
+            self.assertEqual(memio.peek(5), self.EOF)
             self.assertEqual(memio.peek(0), b"")
             self.assertEqual(memio.tell(), len(buf))
             # Peeking works after writing