def test_yaml_writer_dumps_user_messages():
events = [
UserUttered("Hello", {"name": "greet"}),
ActionExecuted("utter_greet")
]
tracker = DialogueStateTracker.from_events("default", events)
dump = YAMLStoryWriter().dumps(tracker.as_story().story_steps)
assert (dump.strip() == textwrap.dedent("""
version: "2.0"
stories:
- story: default
steps:
- intent: greet
user: |-
Hello
- action: utter_greet
""").strip())
def test_yaml_writer_doesnt_dump_action_unlikely_intent():
events = [
UserUttered("Hello", {"name": "greet"}),
ActionExecuted("utter_hello"),
ActionExecuted(ACTION_UNLIKELY_INTENT_NAME,
metadata={"key1": "value1"}),
ActionExecuted("utter_bye"),
]
tracker = DialogueStateTracker.from_events("default", events)
dump = YAMLStoryWriter().dumps(tracker.as_story().story_steps,
is_test_story=True)
assert (dump.strip() == textwrap.dedent("""
version: "3.0"
stories:
- story: default
steps:
- intent: greet
user: |-
Hello
- action: utter_hello
- action: utter_bye
""").strip())
