def test_add_training_example_with_entity(self): processor = MongoProcessor() results = list(processor.add_training_example( ["Log a [critical issue](priority)"], "get_priority", "tests", "testUser" )) assert results[0]['_id'] assert results[0]['text'] == "Log a [critical issue](priority)" assert results[0]['message'] == "Training Example added successfully!" intents = processor.get_intents("tests") assert any("get_priority" == intent['name'] for intent in intents) entities = processor.get_entities("tests") assert any("priority" == entity['name'] for entity in entities) new_training_example = TrainingExamples.objects(bot="tests").get( text="Log a critical issue" ) slots = Slots.objects(bot="tests") new_slot = slots.get(name="priority") assert slots.__len__() == 1 assert new_slot.name == "priority" assert new_slot.type == "text" assert new_training_example.text == "Log a critical issue"
def test_get_entities(self): processor = MongoProcessor() expected = ["priority", "file_text", "ticketID"] actual = processor.get_entities("tests") assert actual.__len__() == expected.__len__() assert all(item["name"] in expected for item in actual)