Added tests to check the buffer type returned during reading.

alexandre.vassalotti · alexandre.vassalotti · commit 7cd0d4f2e2c8 · 2008-05-06T20:30:41.000Z
git-svn-id: http://svn.python.org/projects/python/branches/py3k@62779 6015fed2-1504-0410-9fe1-9d1591cc4771
diff --git a/Lib/test/test_memoryio.py b/Lib/test/test_memoryio.py
@@ -113,6 +113,8 @@ def test_read(self):
         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 @@ def test_readline(self):
         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 @@ def test_readlines(self):
         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 @@ def test_getvalue(self):
         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)
