diff options
| -rw-r--r-- | Lib/test/test_io.py | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/Lib/test/test_io.py b/Lib/test/test_io.py index a5d87b9d1c..d10f52a552 100644 --- a/Lib/test/test_io.py +++ b/Lib/test/test_io.py @@ -2233,6 +2233,28 @@ class TextIOWrapperTest(unittest.TestCase): reads += c self.assertEqual(reads, "A"*127+"\nB") + def test_writelines(self): + l = ['ab', 'cd', 'ef'] + buf = self.BytesIO() + txt = self.TextIOWrapper(buf) + txt.writelines(l) + txt.flush() + self.assertEqual(buf.getvalue(), b'abcdef') + + def test_writelines_userlist(self): + l = UserList(['ab', 'cd', 'ef']) + buf = self.BytesIO() + txt = self.TextIOWrapper(buf) + txt.writelines(l) + txt.flush() + self.assertEqual(buf.getvalue(), b'abcdef') + + def test_writelines_error(self): + txt = self.TextIOWrapper(self.BytesIO()) + self.assertRaises(TypeError, txt.writelines, [1, 2, 3]) + self.assertRaises(TypeError, txt.writelines, None) + self.assertRaises(TypeError, txt.writelines, b'abc') + def test_issue1395_1(self): txt = self.TextIOWrapper(self.BytesIO(self.testdata), encoding="ascii") |