def test_wit_response(): from rasa.nlu.emulators.wit import WitEmulator em = WitEmulator() data = { "text": "I want italian food", "intent": { "name": "inform", "confidence": 0.4794813722432127 }, "entities": [{ "entity": "cuisine", "value": "italian", "start": 7, "end": 14 }], } norm = em.normalise_response_json(data) assert norm == [{ "entities": { "cuisine": { "confidence": None, "type": "value", "value": "italian", "start": 7, "end": 14, } }, "intent": "inform", "_text": "I want italian food", "confidence": 0.4794813722432127, }]
def test_wit_response(): from rasa.nlu.emulators.wit import WitEmulator em = WitEmulator() data = { "text": "I want italian food", "intent": {"name": "inform", "confidence": 0.4794813722432127}, "entities": [{"entity": "cuisine", "value": "italian", "start": 7, "end": 14}]} norm = em.normalise_response_json(data) assert norm == [{ 'entities': { 'cuisine': { 'confidence': None, 'type': 'value', 'value': 'italian', 'start': 7, 'end': 14 } }, 'intent': 'inform', '_text': 'I want italian food', 'confidence': 0.4794813722432127, }]
def test_wit_response(): from rasa.nlu.emulators.wit import WitEmulator em = WitEmulator() data = { "text": "I want italian food", "intent": { "name": "inform", "confidence": 0.4794813722432127 }, "entities": [ { "entity": "cuisine", "value": "italian", "start": 7, "end": 14, "confidence_entity": 0.1234, }, { "entity": "cuisine", "value": "italian", "role": "desert", "start": 7, "end": 14, "confidence_entity": 0.1234, }, ], } norm = em.normalise_response_json(data) expected = { "text": "I want italian food", "intents": [{ "name": "inform", "confidence": 0.4794813722432127 }], "entities": { "cuisine:cuisine": [{ "name": "cuisine", "role": "cuisine", "start": 7, "end": 14, "body": "italian", "value": "italian", "confidence": 0.1234, "entities": [], }], "cuisine:desert": [{ "name": "cuisine", "role": "desert", "start": 7, "end": 14, "body": "italian", "value": "italian", "confidence": 0.1234, "entities": [], }], }, } assert norm == expected