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())