mirror of
https://github.com/python/cpython.git
synced 2025-08-01 15:43:13 +00:00
Issue #2432: give DictReader the dialect and line_num attributes
advertised in the docs.
This commit is contained in:
parent
331243270d
commit
77354cf5ef
3 changed files with 18 additions and 10 deletions
|
@ -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):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue