From 5da31eb261e4889daf731ec5d88f5a1af3503423 Mon Sep 17 00:00:00 2001 From: Alexandre Vassalotti Date: Tue, 6 May 2008 20:30:41 +0000 Subject: Added tests to check the buffer type returned during reading. --- Lib/test/test_memoryio.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/Lib/test/test_memoryio.py b/Lib/test/test_memoryio.py index 0b09ca4..2f5982f 100644 --- a/Lib/test/test_memoryio.py +++ b/Lib/test/test_memoryio.py @@ -113,6 +113,8 @@ class MemoryTestMixin: memio.seek(0) self.assertEqual(memio.read(-1), buf) memio.seek(0) + self.assertEqual(type(memio.read()), type(buf)) + memio.seek(0) self.assertEqual(memio.read(None), buf) self.assertRaises(TypeError, memio.read, '') memio.close() @@ -142,6 +144,7 @@ class MemoryTestMixin: self.assertEqual(memio.readline(), buf[:-1]) self.assertEqual(memio.readline(), self.EOF) memio.seek(0) + self.assertEqual(type(memio.readline()), type(buf)) self.assertEqual(memio.readline(None), buf) self.assertRaises(TypeError, memio.readline, '') memio.close() @@ -161,6 +164,8 @@ class MemoryTestMixin: memio.seek(0) self.assertEqual(memio.readlines(0), [buf] * 10) memio.seek(0) + self.assertEqual(type(memio.readlines()[0]), type(buf)) + memio.seek(0) self.assertEqual(memio.readlines(None), [buf] * 10) self.assertRaises(TypeError, memio.readlines, '') memio.close() @@ -195,6 +200,7 @@ class MemoryTestMixin: self.assertEqual(memio.getvalue(), buf) memio.read() self.assertEqual(memio.getvalue(), buf) + self.assertEqual(type(memio.getvalue()), type(buf)) memio = self.ioclass(buf * 1000) self.assertEqual(memio.getvalue()[-3:], self.buftype("890")) memio = self.ioclass(buf) -- cgit v0.12