def test_csv_data_from_file_with_same_rows_as_preview_limit():
csv = (
"playerID,yearID,teamID,lgID,inseason,G,W,L,rank,plyrMgr\n"
"wrighha01,1871,BS1,NA,1,31,20,10,3,Y\n"
"woodji01,1871,CH1,NA,1,28,19,9,2,Y\n"
"paborch01,1871,CL1,NA,1,29,10,19,8,Y\n"
"lennobi01,1871,FW1,NA,1,14,5,9,8,Y\n"
"deaneha01,1871,FW1,NA,2,5,2,3,8,Y\n"
"fergubo01,1871,NY2,NA,1,33,16,17,5,Y\n"
"mcbridi01,1871,PH1,NA,1,28,21,7,1,Y\n"
"hastisc01,1871,RC1,NA,1,25,4,21,9,Y\n"
"pikeli01,1871,TRO,NA,1,4,1,3,6,Y\n"
)
result = helpers._csv_data_from_file(StringIO.StringIO(csv), preview_limit=10)
assert result["success"] is True
assert len(result["data"]) == 10
assert result["data"][0] == (
"playerID",
"wrighha01",
"woodji01",
"paborch01",
"lennobi01",
"deaneha01",
"fergubo01",
"mcbridi01",
"hastisc01",
"pikeli01",
)
assert result["data"][4] == ("inseason", "1", "1", "1", "1", "2", "1", "1", "1", "1")
assert result["data"][9] == ("plyrMgr", "Y", "Y", "Y", "Y", "Y", "Y", "Y", "Y", "Y")
def test_csv_data_from_file_with_less_rows_than_preview_limit():
csv = (
"playerID,yearID,teamID,lgID,inseason,G,W,L,rank,plyrMgr\n"
"wrighha01,1871,BS1,NA,1,31,20,10,3,Y\n"
"woodji01,1871,CH1,NA,1,28,19,9,2,Y\n"
"paborch01,1871,CL1,NA,1,29,10,19,8,Y\n"
"lennobi01,1871,FW1,NA,1,14,5,9,8,Y\n"
"deaneha01,1871,FW1,NA,2,5,2,3,8,Y\n"
"millsev01,1872,BL1,NA,2,17,8,6,2,Y\n"
)
result = helpers._csv_data_from_file(StringIO.StringIO(csv))
assert result["success"] is True
assert len(result["data"]) == 10
assert result["data"][0] == (
"playerID",
"wrighha01",
"woodji01",
"paborch01",
"lennobi01",
"deaneha01",
"millsev01",
)
assert result["data"][4] == ("inseason", "1", "1", "1", "1", "2", "2")
assert result["data"][9] == ("plyrMgr", "Y", "Y", "Y", "Y", "Y", "Y")
def test_csv_data_from_file_with_same_rows_as_preview_limit():
csv = ('playerID,yearID,teamID,lgID,inseason,G,W,L,rank,plyrMgr\n'
'wrighha01,1871,BS1,NA,1,31,20,10,3,Y\n'
'woodji01,1871,CH1,NA,1,28,19,9,2,Y\n'
'paborch01,1871,CL1,NA,1,29,10,19,8,Y\n'
'lennobi01,1871,FW1,NA,1,14,5,9,8,Y\n'
'deaneha01,1871,FW1,NA,2,5,2,3,8,Y\n'
'fergubo01,1871,NY2,NA,1,33,16,17,5,Y\n'
'mcbridi01,1871,PH1,NA,1,28,21,7,1,Y\n'
'hastisc01,1871,RC1,NA,1,25,4,21,9,Y\n'
'pikeli01,1871,TRO,NA,1,4,1,3,6,Y\n')
result = helpers._csv_data_from_file(StringIO.StringIO(csv),
preview_limit=10)
assert result['success'] is True
assert len(result['data']) == 10
assert result['data'][0] == ('playerID', 'wrighha01', 'woodji01',
'paborch01', 'lennobi01', 'deaneha01',
'fergubo01', 'mcbridi01', 'hastisc01',
'pikeli01')
assert result['data'][4] == ('inseason', '1', '1', '1', '1', '2', '1', '1',
'1', '1')
assert result['data'][9] == ('plyrMgr', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y',
'Y', 'Y')
def test_csv_data_from_file_with_good_csv_file():
'''Test _csv_data_from_file() with a good, 14-line CSV file.
We expect to get only the first 10 rows back, and the rows should have
been transformed from vertical into horizontal rows.
'''
csv = ('playerID,yearID,teamID,lgID,inseason,G,W,L,rank,plyrMgr\n'
'wrighha01,1871,BS1,NA,1,31,20,10,3,Y\n'
'woodji01,1871,CH1,NA,1,28,19,9,2,Y\n'
'paborch01,1871,CL1,NA,1,29,10,19,8,Y\n'
'lennobi01,1871,FW1,NA,1,14,5,9,8,Y\n'
'deaneha01,1871,FW1,NA,2,5,2,3,8,Y\n'
'fergubo01,1871,NY2,NA,1,33,16,17,5,Y\n'
'mcbridi01,1871,PH1,NA,1,28,21,7,1,Y\n'
'hastisc01,1871,RC1,NA,1,25,4,21,9,Y\n'
'pikeli01,1871,TRO,NA,1,4,1,3,6,Y\n'
'cravebi01,1871,TRO,NA,2,25,12,12,6,Y\n'
'youngni99,1871,WS3,NA,1,32,15,15,4,N\n'
'cravebi01,1872,BL1,NA,1,41,27,13,2,Y\n'
'millsev01,1872,BL1,NA,2,17,8,6,2,Y\n')
result = helpers._csv_data_from_file(StringIO.StringIO(csv))
assert result['success'] is True
assert len(result['data']) == 10
assert result['data'][0] == ('playerID', 'wrighha01', 'woodji01',
'paborch01', 'lennobi01', 'deaneha01',
'fergubo01', 'mcbridi01', 'hastisc01',
'pikeli01')
assert result['data'][4] == ('inseason', '1', '1', '1', '1', '2', '1', '1',
'1', '1')
assert result['data'][9] == ('plyrMgr', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y',
'Y', 'Y')
def test_csv_data_from_file_with_less_rows_than_preview_limit():
csv = ('playerID,yearID,teamID,lgID,inseason,G,W,L,rank,plyrMgr\n'
'wrighha01,1871,BS1,NA,1,31,20,10,3,Y\n'
'woodji01,1871,CH1,NA,1,28,19,9,2,Y\n'
'paborch01,1871,CL1,NA,1,29,10,19,8,Y\n'
'lennobi01,1871,FW1,NA,1,14,5,9,8,Y\n'
'deaneha01,1871,FW1,NA,2,5,2,3,8,Y\n'
'millsev01,1872,BL1,NA,2,17,8,6,2,Y\n')
result = helpers._csv_data_from_file(StringIO.StringIO(csv))
assert result['success'] is True
assert len(result['data']) == 10
assert result['data'][0] == ('playerID', 'wrighha01', 'woodji01',
'paborch01', 'lennobi01', 'deaneha01',
'millsev01')
assert result['data'][4] == ('inseason', '1', '1', '1', '1', '2', '2')
assert result['data'][9] == ('plyrMgr', 'Y', 'Y', 'Y', 'Y', 'Y', 'Y')
def test_csv_data_from_file_with_good_csv_file():
"""Test _csv_data_from_file() with a good, 14-line CSV file.
We expect to get only the first 10 rows back, and the rows should have
been transformed from vertical into horizontal rows.
"""
csv = (
"playerID,yearID,teamID,lgID,inseason,G,W,L,rank,plyrMgr\n"
"wrighha01,1871,BS1,NA,1,31,20,10,3,Y\n"
"woodji01,1871,CH1,NA,1,28,19,9,2,Y\n"
"paborch01,1871,CL1,NA,1,29,10,19,8,Y\n"
"lennobi01,1871,FW1,NA,1,14,5,9,8,Y\n"
"deaneha01,1871,FW1,NA,2,5,2,3,8,Y\n"
"fergubo01,1871,NY2,NA,1,33,16,17,5,Y\n"
"mcbridi01,1871,PH1,NA,1,28,21,7,1,Y\n"
"hastisc01,1871,RC1,NA,1,25,4,21,9,Y\n"
"pikeli01,1871,TRO,NA,1,4,1,3,6,Y\n"
"cravebi01,1871,TRO,NA,2,25,12,12,6,Y\n"
"youngni99,1871,WS3,NA,1,32,15,15,4,N\n"
"cravebi01,1872,BL1,NA,1,41,27,13,2,Y\n"
"millsev01,1872,BL1,NA,2,17,8,6,2,Y\n"
)
result = helpers._csv_data_from_file(StringIO.StringIO(csv))
assert result["success"] is True
assert len(result["data"]) == 10
assert result["data"][0] == (
"playerID",
"wrighha01",
"woodji01",
"paborch01",
"lennobi01",
"deaneha01",
"fergubo01",
"mcbridi01",
"hastisc01",
"pikeli01",
)
assert result["data"][4] == ("inseason", "1", "1", "1", "1", "2", "1", "1", "1", "1")
assert result["data"][9] == ("plyrMgr", "Y", "Y", "Y", "Y", "Y", "Y", "Y", "Y", "Y")
