summaryrefslogtreecommitdiff
path: root/Lib/test/test_csv.py
diff options
context:
space:
mode:
authorSerhiy Storchaka <storchaka@gmail.com>2015-03-30 09:09:54 +0300
committerSerhiy Storchaka <storchaka@gmail.com>2015-03-30 09:09:54 +0300
commit7901b48a1f89b9bfa9d111ae3725400b466a9baa (patch)
tree4cebbcaeeb7b1f29f5440c7762f55bf3a57c0c54 /Lib/test/test_csv.py
parenta695f83f0de060a77352174be8a5c6f6500ab98a (diff)
downloadcpython-git-7901b48a1f89b9bfa9d111ae3725400b466a9baa.tar.gz
Issue #23171: csv.Writer.writerow() now supports arbitrary iterables.
Diffstat (limited to 'Lib/test/test_csv.py')
-rw-r--r--Lib/test/test_csv.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/Lib/test/test_csv.py b/Lib/test/test_csv.py
index 41ef790eb2..7be3cc3c06 100644
--- a/Lib/test/test_csv.py
+++ b/Lib/test/test_csv.py
@@ -186,6 +186,14 @@ class Test_Csv(unittest.TestCase):
self._write_test(['a',1,'p,q'], 'a,1,p\\,q',
escapechar='\\', quoting = csv.QUOTE_NONE)
+ def test_write_iterable(self):
+ self._write_test(iter(['a', 1, 'p,q']), 'a,1,"p,q"')
+ self._write_test(iter(['a', 1, None]), 'a,1,')
+ self._write_test(iter([]), '')
+ self._write_test(iter([None]), '""')
+ self._write_error_test(csv.Error, iter([None]), quoting=csv.QUOTE_NONE)
+ self._write_test(iter([None, None]), ',')
+
def test_writerows(self):
class BrokenFile:
def write(self, buf):