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"