mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-01 07:33:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

accept issue 3436

Browse source
This commit is contained in:
Skip Montanaro 2008-08-08 22:52:51 +00:00
parent b04d4853a7
commit a032bf41f6
4 changed files with 67 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

33
Lib/test/test_csv.py
View file

@ -611,11 +611,43 @@ class TestDictFields(unittest.TestCase):
fileobj.write("f1,f2,f3\r\n1,2,abc\r\n")
fileobj.seek(0)
reader = csv.DictReader(fileobj)
self.assertEqual(reader.fieldnames, ["f1", "f2", "f3"])
self.assertEqual(reader.next(), {"f1": '1', "f2": '2', "f3": 'abc'})
finally:
fileobj.close()
os.unlink(name)
# Two test cases to make sure existing ways of implicitly setting
# fieldnames continue to work. Both arise from discussion in issue3436.
def test_read_dict_fieldnames_from_file(self):
fd, name = tempfile.mkstemp()
f = os.fdopen(fd, "w+b")
try:
f.write("f1,f2,f3\r\n1,2,abc\r\n")
f.seek(0)
reader = csv.DictReader(f, fieldnames=csv.reader(f).next())
self.assertEqual(reader.fieldnames, ["f1", "f2", "f3"])
self.assertEqual(reader.next(), {"f1": '1', "f2": '2', "f3": 'abc'})
finally:
f.close()
os.unlink(name)
def test_read_dict_fieldnames_chain(self):
import itertools
fd, name = tempfile.mkstemp()
f = os.fdopen(fd, "w+b")
try:
f.write("f1,f2,f3\r\n1,2,abc\r\n")
f.seek(0)
reader = csv.DictReader(f)
first = next(reader)
for row in itertools.chain([first], reader):
self.assertEqual(reader.fieldnames, ["f1", "f2", "f3"])
self.assertEqual(row, {"f1": '1', "f2": '2', "f3": 'abc'})
finally:
f.close()
os.unlink(name)
def test_read_long(self):
fd, name = tempfile.mkstemp()
fileobj = os.fdopen(fd, "w+b")
@ -651,6 +683,7 @@ class TestDictFields(unittest.TestCase):
fileobj.write("f1,f2\r\n1,2,abc,4,5,6\r\n")
fileobj.seek(0)
reader = csv.DictReader(fileobj, restkey="_rest")
self.assertEqual(reader.fieldnames, ["f1", "f2"])
self.assertEqual(reader.next(), {"f1": '1', "f2": '2',
"_rest": ["abc", "4", "5", "6"]})
finally: