Python YAMLStoryWriter.stories_to_yaml Examples

Example #1

File: test_yaml_story_writer.py Project: zoovu/rasa

def test_yaml_writer_stories_to_yaml(domain: Domain):
    reader = YAMLStoryReader(domain, None)
    writer = YAMLStoryWriter()
    steps = reader.read_from_file(
        "data/test_yaml_stories/simple_story_with_only_end.yml")

    result = writer.stories_to_yaml(steps)
    assert isinstance(result, OrderedDict)
    assert "stories" in result
    assert len(result["stories"]) == 1

Example #2

File: test_yaml_story_writer.py Project: zoovu/rasa

def test_yaml_writer_stories_to_yaml_with_null_entities(domain: Domain):
    writer = YAMLStoryWriter()
    stories = textwrap.dedent("""
    version: "3.0"
    stories:
    - story: happy path
      steps:
      - intent: test_intent
        entities:
        - test_entity: null
        - test_entity2: false
    """)

    stories_yaml = YAMLStoryReader().read_from_string(stories)
    result = writer.stories_to_yaml(stories_yaml)
    assert isinstance(result, OrderedDict)
    assert "stories" in result
    assert len(result["stories"]) == 1
    entities = result["stories"][0]["steps"][0]["entities"]
    assert entities[0] == "test_entity"
    assert entities[1] == OrderedDict({"test_entity2": False})