summaryrefslogtreecommitdiff
path: root/Lib/test
diff options
context:
space:
mode:
authorGiampaolo Rodola' <g.rodola@gmail.com>2014-04-29 02:04:14 +0200
committerGiampaolo Rodola' <g.rodola@gmail.com>2014-04-29 02:04:14 +0200
commit7971286b6e5a022103b5ee646086686cd37a1e7d (patch)
tree1aee04c6a10acd222a6ded34b1d0e7a92edc621f /Lib/test
parent8af740c46a89f46d6a2f59869847724a5eaf3625 (diff)
parent05da1ef40a4285ee859836d67bced63504dd89b2 (diff)
downloadcpython-git-7971286b6e5a022103b5ee646086686cd37a1e7d.tar.gz
merge heads
Diffstat (limited to 'Lib/test')
-rw-r--r--Lib/test/test_csv.py10
1 files changed, 10 insertions, 0 deletions
diff --git a/Lib/test/test_csv.py b/Lib/test/test_csv.py
index 7e2485f954..7c31ac71f6 100644
--- a/Lib/test/test_csv.py
+++ b/Lib/test/test_csv.py
@@ -575,6 +575,16 @@ class TestDictFields(unittest.TestCase):
fileobj.readline() # header
self.assertEqual(fileobj.read(), "10,,abc\r\n")
+ def test_write_multiple_dict_rows(self):
+ fileobj = StringIO()
+ writer = csv.DictWriter(fileobj, fieldnames=["f1", "f2", "f3"])
+ writer.writeheader()
+ self.assertEqual(fileobj.getvalue(), "f1,f2,f3\r\n")
+ writer.writerows([{"f1": 1, "f2": "abc", "f3": "f"},
+ {"f1": 2, "f2": 5, "f3": "xyz"}])
+ self.assertEqual(fileobj.getvalue(),
+ "f1,f2,f3\r\n1,abc,f\r\n2,5,xyz\r\n")
+
def test_write_no_fields(self):
fileobj = StringIO()
self.assertRaises(TypeError, csv.DictWriter, fileobj)