Test peek(3) and peek(5)

Author: marcelm

Lib/test/test_memoryio.py

@@ -523,6 +523,8 @@ def test_peek(self):
             self.assertEqual(memio.tell(), 0)
             self.assertEqual(memio.peek(1), buf[:1])
             self.assertEqual(memio.peek(1), buf[:1])
+            self.assertEqual(memio.peek(3), buf[:3])
+            self.assertEqual(memio.peek(5), buf[:5])
             self.assertEqual(memio.peek(), buf[:1])
             self.assertEqual(memio.peek(0), b"")
             self.assertEqual(memio.peek(len(buf) + 100), buf)
@@ -532,6 +534,8 @@ def test_peek(self):
             self.assertEqual(memio.tell(), 1)
             self.assertEqual(memio.peek(1), buf[1:2])
             self.assertEqual(memio.peek(1), buf[1:2])
+            self.assertEqual(memio.peek(3), buf[1:4])
+            self.assertEqual(memio.peek(5), buf[1:6])
             self.assertEqual(memio.peek(), buf[1:2])
             self.assertEqual(memio.peek(0), b"")
             self.assertEqual(memio.peek(len(buf) + 100), buf[1:])
@@ -540,6 +544,8 @@ def test_peek(self):
             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