Skip to content

Commit 93d4dd1

Browse files
committed
bpo-32255: Add tests for csv.writer
1 parent f547e72 commit 93d4dd1

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

Lib/test/test_csv.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,25 @@ def write(self, buf):
211211
fileobj.seek(0)
212212
self.assertEqual(fileobj.read(), "a,b\r\nc,d\r\n")
213213

214+
def test_writerows_with_none(self):
215+
with TemporaryFile("w+", newline='') as fileobj:
216+
writer = csv.writer(fileobj)
217+
writer.writerows([['a',None],[None,'d']])
218+
fileobj.seek(0)
219+
self.assertEqual(fileobj.read(), "a,\r\n,d\r\n")
220+
221+
with TemporaryFile("w+", newline='') as fileobj:
222+
writer = csv.writer(fileobj)
223+
writer.writerows([[None],['a']])
224+
fileobj.seek(0)
225+
self.assertEqual(fileobj.read(), '""\r\na\r\n')
226+
227+
with TemporaryFile("w+", newline='') as fileobj:
228+
writer = csv.writer(fileobj)
229+
writer.writerows([['a'],[None]])
230+
fileobj.seek(0)
231+
self.assertEqual(fileobj.read(), 'a\r\n""\r\n')
232+
214233
@support.cpython_only
215234
def test_writerows_legacy_strings(self):
216235
import _testcapi

0 commit comments

Comments
 (0)