def test_concertbot_training():
from examples.concertbot.train import train_dialogue
assert train_dialogue(domain_file='examples/concertbot/domain.yml',
stories_file='examples/concertbot/data/stories.md',
model_path='examples/concertbot/models/dialogue',
policy_config='examples/concertbot/'
'policy_config.yml')
def test_restaurantbot_example():
sys.path.append("examples/restaurantbot/")
from bot import train_dialogue
p = "examples/restaurantbot/"
agent = train_dialogue(os.path.join(p, "restaurant_domain.yml"),
os.path.join(p, "models", "dialogue"),
os.path.join(p, "data", "babi_stories.md"))
responses = agent.handle_message("_greet")
assert responses[0] == 'how can I help you?'
def test_restaurantbot_example():
sys.path.append("examples/restaurantbot/")
from bot import train_dialogue
p = "examples/restaurantbot/"
stories = os.path.join("data", "test_stories", "stories_babi_small.md")
agent = train_dialogue(os.path.join(p, "restaurant_domain.yml"),
os.path.join(p, "models", "dialogue"), stories)
responses = agent.handle_text("/greet")
assert responses[0]['text'] == 'how can I help you?'
def test_restaurantbot_example():
sys.path.append("examples/restaurantbot/")
from bot import train_dialogue
p = "examples/restaurantbot/"
stories = os.path.join("data", "test_stories", "stories_babi_small.md")
agent = train_dialogue(os.path.join(p, "restaurant_domain.yml"),
os.path.join(p, "models", "dialogue"),
stories)
responses = agent.handle_text("/greet")
assert responses[0]['text'] == 'how can I help you?'