From 7901b48a1f89b9bfa9d111ae3725400b466a9baa Mon Sep 17 00:00:00 2001 From: Serhiy Storchaka Date: Mon, 30 Mar 2015 09:09:54 +0300 Subject: Issue #23171: csv.Writer.writerow() now supports arbitrary iterables. --- Lib/test/test_csv.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'Lib/test/test_csv.py') 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): -- cgit v1.2.1