tests: Add test for io.BufferedWriter.

2 files changed, 26 insertions, 0 deletions

diff --git a/tests/io/buffered_writer.py b/tests/io/buffered_writer.py
new file mode 100644
index 000000000..0fe8a7730
--- /dev/null
+++ b/tests/io/buffered_writer.py
@@ -0,0 +1,22 @@
+import _io as io
+
+try:
+    io.BytesIO
+    io.BufferedWriter
+except AttributeError:
+    import sys
+    print('SKIP')
+    sys.exit()
+
+bts = io.BytesIO()
+buf = io.BufferedWriter(bts, 8)
+
+buf.write(b"foobar")
+print(bts.getvalue())
+buf.write(b"foobar")
+# CPython has different flushing policy, so value below is different
+print(bts.getvalue())
+buf.flush()
+print(bts.getvalue())
+buf.flush()
+print(bts.getvalue())
diff --git a/tests/io/buffered_writer.py.exp b/tests/io/buffered_writer.py.exp
new file mode 100644
index 000000000..1309487e1
--- /dev/null
+++ b/tests/io/buffered_writer.py.exp
@@ -0,0 +1,4 @@
+b''
+b'foobarfo'
+b'foobarfoobar'
+b'foobarfoobar'