<s> import sys <EOL> import UnitTest <EOL> import csv <EOL> class CsvModuleTest ( UnitTest . UnitTest ) : <EOL> def test_reader ( self ) : <EOL> lines = [ <EOL> '''<STR_LIT>''' , <EOL> '''<STR_LIT>''' , <EOL> '''<STR_LIT>''' , <EOL> '''<STR_LIT>''' , <EOL> '''<STR_LIT>''' , <EOL> '<STR_LIT:\n>' , <EOL> '<STR_LIT>' , <EOL> '''<STR_LIT>''' , <EOL> '''<STR_LIT>''' , <EOL> '<STR_LIT>' , <EOL> ] <EOL> expected = [ <EOL> [ '<STR_LIT:1>' , '<STR_LIT>' , '<STR_LIT:3>' ] , <EOL> [ '<STR_LIT>' , '<STR_LIT:a>' , '<STR_LIT>' ] , <EOL> [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> [ '<STR_LIT:s>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> [ '<STR_LIT>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> [ ] , <EOL> [ ] , <EOL> [ '<STR_LIT:1>' , '<STR_LIT>' , '<STR_LIT:3>' ] , <EOL> [ '<STR_LIT:s>' , '<STR_LIT>' , '<STR_LIT>' ] , <EOL> [ ] , <EOL> ] <EOL> reader = csv . reader ( lines ) <EOL> rows = [ ] <EOL> idx = - <NUM_LIT:1> <EOL> for row in reader : <EOL> idx += <NUM_LIT:1> <EOL> self . assertEqual ( row , expected [ idx ] , <EOL> "<STR_LIT>" % ( idx , row , expected [ idx ] ) ) <EOL> rows . append ( row ) <EOL> lines = [ <EOL> '''<STR_LIT>''' , <EOL> ] <EOL> expected = [ <EOL> [ '<STR_LIT:1>' , '<STR_LIT:2>' , '<STR_LIT:3>' ] , <EOL> ] <EOL> reader = csv . reader ( lines ) <EOL> rows = [ ] <EOL> idx = - <NUM_LIT:1> <EOL> for row in reader : <EOL> idx += <NUM_LIT:1> <EOL> self . assertEqual ( row , expected [ idx ] , <EOL> "<STR_LIT>" % ( idx , row , expected [ idx ] ) ) <EOL> rows . append ( row ) </s>
