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)
