示例#1
0
 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)