diff options
| author | Georg Brandl <georg@python.org> | 2008-03-21 20:01:51 (GMT) |
|---|---|---|
| committer | Georg Brandl <georg@python.org> | 2008-03-21 20:01:51 (GMT) |
| commit | 77354cf5ef9644121a28041216591762628d8b65 (patch) | |
| tree | b0fc669e46257a85f58988cd93180cfe5cc71ef8 /Lib/test/test_csv.py | |
| parent | 331243270dc91cfccd31b071abb8d2fcf84a7984 (diff) | |
| download | cpython-77354cf5ef9644121a28041216591762628d8b65.zip cpython-77354cf5ef9644121a28041216591762628d8b65.tar.gz cpython-77354cf5ef9644121a28041216591762628d8b65.tar.bz2 | |
Issue #2432: give DictReader the dialect and line_num attributes
advertised in the docs.
Diffstat (limited to 'Lib/test/test_csv.py')
| -rw-r--r-- | Lib/test/test_csv.py | 22 |
1 files changed, 12 insertions, 10 deletions
diff --git a/Lib/test/test_csv.py b/Lib/test/test_csv.py index 4954383..8a89f4c 100644 --- a/Lib/test/test_csv.py +++ b/Lib/test/test_csv.py @@ -269,16 +269,18 @@ class Test_Csv(unittest.TestCase): csv.field_size_limit(limit) def test_read_linenum(self): - r = csv.reader(['line,1', 'line,2', 'line,3']) - self.assertEqual(r.line_num, 0) - r.next() - self.assertEqual(r.line_num, 1) - r.next() - self.assertEqual(r.line_num, 2) - r.next() - self.assertEqual(r.line_num, 3) - self.assertRaises(StopIteration, r.next) - self.assertEqual(r.line_num, 3) + for r in (csv.reader(['line,1', 'line,2', 'line,3']), + csv.DictReader(['line,1', 'line,2', 'line,3'], + fieldnames=['a', 'b', 'c'])): + self.assertEqual(r.line_num, 0) + r.next() + self.assertEqual(r.line_num, 1) + r.next() + self.assertEqual(r.line_num, 2) + r.next() + self.assertEqual(r.line_num, 3) + self.assertRaises(StopIteration, r.next) + self.assertEqual(r.line_num, 3) class TestDialectRegistry(unittest.TestCase): def test_registry_badargs(self): |