summaryrefslogtreecommitdiff
path: root/tests/io
diff options
context:
space:
mode:
authorRami Ali <flowergrass@users.noreply.github.com>2017-01-17 13:12:50 +1100
committerDamien George <damien.p.george@gmail.com>2017-01-17 13:27:02 +1100
commitcba723fc8c9b76150efc95e54996ad12a9a8cfc6 (patch)
treea8f23d19e69fa56a39b0124efb6430f34ee6decc /tests/io
parent96baaa68a4430efd585f2186976cfa473ca73bfc (diff)
tests/io: Improve test coverage of io.BufferedWriter.
Diffstat (limited to 'tests/io')
-rw-r--r--tests/io/buffered_writer.py6
-rw-r--r--tests/io/buffered_writer.py.exp1
2 files changed, 7 insertions, 0 deletions
diff --git a/tests/io/buffered_writer.py b/tests/io/buffered_writer.py
index afeaa839c..bb7b4e8db 100644
--- a/tests/io/buffered_writer.py
+++ b/tests/io/buffered_writer.py
@@ -20,3 +20,9 @@ buf.flush()
print(bts.getvalue())
buf.flush()
print(bts.getvalue())
+
+# special case when alloc is a factor of total buffer length
+bts = io.BytesIO()
+buf = io.BufferedWriter(bts, 1)
+buf.write(b"foo")
+print(bts.getvalue())
diff --git a/tests/io/buffered_writer.py.exp b/tests/io/buffered_writer.py.exp
index 1309487e1..d086935a9 100644
--- a/tests/io/buffered_writer.py.exp
+++ b/tests/io/buffered_writer.py.exp
@@ -2,3 +2,4 @@ b''
b'foobarfo'
b'foobarfoobar'
b'foobarfoobar'
+b'foo'