test_nlu.py

from rasa_nlu.model import Interpreter, RasaNLUConfig
from rasa_core import utils
import logging


def test_loop(user_messages):
    for message in user_messages:
        result = interpreter.parse(message)
        logger.info("test message: `{}`, result: {} with confidence {}".format(message, result['intent']['name'],
                                                                               result['intent']['confidence']))

logger = logging.getLogger(__name__)
utils.configure_colored_logging(loglevel="INFO")

model_path = './models/nlu/default/jokerbot'
nlu_config_path = './nlu_model_config.json'
interpreter = Interpreter.load(model_path, RasaNLUConfig(nlu_config_path))

test_greetings = [
    u"hey doucheface!",
    u"hey asshole!",
    u"hi guy",
    u"hello sir",
    u"what's up dude?",
    u"good morning"
]
test_loop(test_greetings)

test_joke_reqs = [
    "tell me a marriage joke",
    "tell me a joke",
    "give me a joke about marriage",
    "tell me something about marriage",
    "tell me a joke about holy matrimony",
    "tell me something funny about marriage",
    "what's a joke about marriage?",
    "what's something funny?"
]
test_loop(test_joke_reqs)