diff options
| author | Rémi Lapeyre <remi.lapeyre@henki.fr> | 2019-05-10 03:50:11 +0200 |
|---|---|---|
| committer | Victor Stinner <vstinner@redhat.com> | 2019-05-10 03:50:11 +0200 |
| commit | fce5ff1e18b522cf52379934a6560583d840e7f9 (patch) | |
| tree | 89d1c624372e8840a1c843e4e78f16088e32df1a /Lib/test | |
| parent | d237b3f0f61990c972b84c45eb4fe137db51a6a7 (diff) | |
| download | cpython-git-fce5ff1e18b522cf52379934a6560583d840e7f9.tar.gz | |
bpo-27497: Add return value to csv.DictWriter.writeheader (GH-12306)
csv.DictWriter.writeheader() now returns the return value of the
underlying csv.Writer.writerow() method.
Patch contributed by Ashish Nitin Patil.
Diffstat (limited to 'Lib/test')
| -rw-r--r-- | Lib/test/test_csv.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/Lib/test/test_csv.py b/Lib/test/test_csv.py index 7a333139b5..a16d14019f 100644 --- a/Lib/test/test_csv.py +++ b/Lib/test/test_csv.py @@ -608,6 +608,12 @@ class TestQuotedEscapedExcel(TestCsvBase): class TestDictFields(unittest.TestCase): ### "long" means the row is longer than the number of fieldnames ### "short" means there are fewer elements in the row than fieldnames + def test_writeheader_return_value(self): + with TemporaryFile("w+", newline='') as fileobj: + writer = csv.DictWriter(fileobj, fieldnames = ["f1", "f2", "f3"]) + writeheader_return_value = writer.writeheader() + self.assertEqual(writeheader_return_value, 10) + def test_write_simple_dict(self): with TemporaryFile("w+", newline='') as fileobj: writer = csv.DictWriter(fileobj, fieldnames = ["f1", "f2", "f3"]) |