Kaydet (Commit) 5da31eb2 authored tarafından Alexandre Vassalotti's avatar Alexandre Vassalotti

Added tests to check the buffer type returned during reading.

üst 77250f4d
...@@ -113,6 +113,8 @@ class MemoryTestMixin: ...@@ -113,6 +113,8 @@ class MemoryTestMixin:
memio.seek(0) memio.seek(0)
self.assertEqual(memio.read(-1), buf) self.assertEqual(memio.read(-1), buf)
memio.seek(0) memio.seek(0)
self.assertEqual(type(memio.read()), type(buf))
memio.seek(0)
self.assertEqual(memio.read(None), buf) self.assertEqual(memio.read(None), buf)
self.assertRaises(TypeError, memio.read, '') self.assertRaises(TypeError, memio.read, '')
memio.close() memio.close()
...@@ -142,6 +144,7 @@ class MemoryTestMixin: ...@@ -142,6 +144,7 @@ class MemoryTestMixin:
self.assertEqual(memio.readline(), buf[:-1]) self.assertEqual(memio.readline(), buf[:-1])
self.assertEqual(memio.readline(), self.EOF) self.assertEqual(memio.readline(), self.EOF)
memio.seek(0) memio.seek(0)
self.assertEqual(type(memio.readline()), type(buf))
self.assertEqual(memio.readline(None), buf) self.assertEqual(memio.readline(None), buf)
self.assertRaises(TypeError, memio.readline, '') self.assertRaises(TypeError, memio.readline, '')
memio.close() memio.close()
...@@ -161,6 +164,8 @@ class MemoryTestMixin: ...@@ -161,6 +164,8 @@ class MemoryTestMixin:
memio.seek(0) memio.seek(0)
self.assertEqual(memio.readlines(0), [buf] * 10) self.assertEqual(memio.readlines(0), [buf] * 10)
memio.seek(0) memio.seek(0)
self.assertEqual(type(memio.readlines()[0]), type(buf))
memio.seek(0)
self.assertEqual(memio.readlines(None), [buf] * 10) self.assertEqual(memio.readlines(None), [buf] * 10)
self.assertRaises(TypeError, memio.readlines, '') self.assertRaises(TypeError, memio.readlines, '')
memio.close() memio.close()
...@@ -195,6 +200,7 @@ class MemoryTestMixin: ...@@ -195,6 +200,7 @@ class MemoryTestMixin:
self.assertEqual(memio.getvalue(), buf) self.assertEqual(memio.getvalue(), buf)
memio.read() memio.read()
self.assertEqual(memio.getvalue(), buf) self.assertEqual(memio.getvalue(), buf)
self.assertEqual(type(memio.getvalue()), type(buf))
memio = self.ioclass(buf * 1000) memio = self.ioclass(buf * 1000)
self.assertEqual(memio.getvalue()[-3:], self.buftype("890")) self.assertEqual(memio.getvalue()[-3:], self.buftype("890"))
memio = self.ioclass(buf) memio = self.ioclass(buf)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment