def test_chunk_important_date_registration(self): utterance = "When is registration?" expected_result = ("REGISTRATION", "registration") doc = chunk_important_date(utterance, important_date_dir="../important_date/") for ent in doc.ents: result = (ent.label_, ent.text) self.assertEqual(expected_result, result)
def test_chunk_important_date_break(self): utterance = "When is the break?" expected_result = ("BREAK", "break") doc = chunk_important_date(utterance, important_date_dir="../important_date/") for ent in doc.ents: result = (ent.label_, ent.text) self.assertEqual(expected_result, result)
def test_chunk_important_date_finals(self): utterance = "When are finals?" expected_result = ("FINALS", "finals") doc = chunk_important_date(utterance, important_date_dir="../important_date/") for ent in doc.ents: result = (ent.label_, ent.text) self.assertEqual(expected_result, result)
def test_chunk_important_date_semester_end(self): utterance = "When does the semester end?" expected_result = ("SEMESTER_END", "semester end") doc = chunk_important_date(utterance, important_date_dir="../important_date/") for ent in doc.ents: result = (ent.label_, ent.text) self.assertEqual(expected_result, result)
def test_chunk_important_date_faculty_report(self): utterance = "When does faculty report?" expected_result = ("FACULTY_REPORT", "faculty report") doc = chunk_important_date(utterance, important_date_dir="../important_date/") for ent in doc.ents: result = (ent.label_, ent.text) self.assertEqual(expected_result, result)
def test_chunk_important_date_drop_date(self): utterance = "When is the last day to drop a class?" expected_result = ("DROP_DATE", "drop a class") doc = chunk_important_date(utterance, important_date_dir="../important_date/") for ent in doc.ents: result = (ent.label_, ent.text) self.assertEqual(expected_result, result)
def test_chunk_important_date_withdraw_date(self): utterance = "When is the last day to withdraw from a class?" expected_result = ("WITHDRAW_DATE", "day to withdraw from a class") doc = chunk_important_date(utterance, important_date_dir="../important_date/") for ent in doc.ents: result = (ent.label_, ent.text) self.assertEqual(expected_result, result)
def test_chunk_important_date_open_residence_halls(self): utterance = "When do the residence halls open?" expected_result = ("OPEN_RESIDENCE_HALLS", "residence halls") doc = chunk_important_date(utterance, important_date_dir="../important_date/") for ent in doc.ents: result = (ent.label_, ent.text) self.assertEqual(expected_result, result)
def test_chunk_important_date_curriculum_study(self): utterance = "When is curriculum study?" expected_result = ("CURRICULUM_STUDY", "curriculum study") doc = chunk_important_date(utterance, important_date_dir="../important_date/") for ent in doc.ents: result = (ent.label_, ent.text) self.assertEqual(expected_result, result)
def chunk_utterance(self, intent, utterance): """ chunk_utterance finds variable slots """ print("In chunk utterance") print("Utterance: ", utterance, "\tIntent: ", intent) if intent == "important_date": print("in important date chunk") doc = chunk_important_date(utterance) # print("Entities in '%s'" % utterance) # for ent in doc.ents: # print(ent.label_, ent.text) elif intent == "course": print("in course chunk") doc = chunk_course(utterance) elif intent == "professor": print("professor chunk") doc = chunk_professor_name(utterance) elif intent == "location": doc = None print("location chunk") else: doc = None print("intent didn't match") return doc