def CSVReader_tests():
def callback(msg, fail):
return
print "+-----------------------------+"
print "| CSVReader Tests |"
print "+-----------------------------+"
csv = CSVReader('testdevice.csv', callback)
assert (csv.getColumn('Not existing') == None)
print "[+] Successfully handled unexisting column name for getColumn"
assert (csv.getColumn('Name'))
print "[+] Successfully fetches column"
assert (type(csv.getColumn('Name')) == list)
print "[+] Correct return type for getColumn"
assert (csv.getColumns(['Not existing']) == None)
print "[+] Successfully handled unexisting column for getColumns"
assert (csv.getColumns(['Name', 'IP']))
print "[+] Successfully fetches columns"
assert (type(csv.getColumns(['Name', 'IP'])) == dict)
print "[+] Correct return type for getColumns"
assert (type(csv.getRows()) == list)
print "[+] Correct return type for getRows"
for row in csv.getRows():
assert (type(row[1]) == pd.core.series.Series)
print "[+] Successfully found Series object in row"
