def testCsvRowToVector(self): eCsvReader = EgoCsvReader() csvRow = ["1", "5", "2", "12", ""] csvRow2 = ["2", "4", "8", "2", "1"] csvTitles = ["A", "B", "C", "D", "E"] csvErrorTitles1 = ["A", "B", "C", "D", "E" ,"F"] csvErrorTitles2 = ["A", "B", "C"] questionIds = [("B", 0), ("A", 0), ("E", 1)] questionIdsError1 = [("B", 0), ("A", 0), ("E", 0)] questionIdsError2 = [("B", 0), ("A", 0), ("Z", 1)] self.assertRaises(ValueError, eCsvReader.csvRowToVector, csvRow, questionIds, csvErrorTitles1) self.assertRaises(ValueError, eCsvReader.csvRowToVector, csvRow, questionIds, csvErrorTitles2) v = eCsvReader.csvRowToVector(csvRow, questionIds, csvTitles) self.assertTrue((v==numpy.array([5, 1, 0])).all()) v = eCsvReader.csvRowToVector(csvRow2, questionIds, csvTitles) self.assertTrue((v==numpy.array([4, 2, 1])).all()) self.assertRaises(ValueError, eCsvReader.csvRowToVector, csvRow, questionIdsError1, csvTitles) self.assertRaises(ValueError, eCsvReader.csvRowToVector, csvRow, questionIdsError2, csvTitles)