summaryrefslogtreecommitdiff
path: root/Lib/test/test_io.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/test_io.py')
-rw-r--r--Lib/test/test_io.py11
1 files changed, 11 insertions, 0 deletions
diff --git a/Lib/test/test_io.py b/Lib/test/test_io.py
index 688f6dc200..737dfab36b 100644
--- a/Lib/test/test_io.py
+++ b/Lib/test/test_io.py
@@ -2,6 +2,7 @@
import sys
import time
+import array
import unittest
from itertools import chain
from test import test_support
@@ -235,6 +236,16 @@ class IOTest(unittest.TestCase):
self.assertEqual(f.read(), b"xxx")
f.close()
+ def test_array_writes(self):
+ a = array.array('i', range(10))
+ n = len(buffer(a))
+ f = io.open(test_support.TESTFN, "wb", 0)
+ self.assertEqual(f.write(a), n)
+ f.close()
+ f = io.open(test_support.TESTFN, "wb")
+ self.assertEqual(f.write(a), n)
+ f.close()
+
class MemorySeekTestMixin: