Merge 308e2d6 into 7e1fe61

data/test_multifile_stories/stories_part_2.md

@@ -14,11 +14,14 @@
 ## show_it_all
 > check_greet
 > check_hello                   <!-- allows multiple entry points -->
-
 * next_intent
     - utter_greet              <!-- actions taken by the bot -->
-
 > check_intermediate            <!-- allows intermediate checkpoints -->
-
 * change_bank_details
     - utter_default            <!-- allows to end without checkpoints -->
+
+## entities and slots
+* greet{"cuisine": "italian"}
+    - slot{"cuisine": "italian"}
+    - utter_default
+    - utter_default <!-- just so length is different from the checkpoint story -->

data/test_stories/stories.md

@@ -35,3 +35,9 @@
 > check_intermediate            <!-- allows intermediate checkpoints -->
 * change_bank_details
     - utter_default            <!-- allows to end without checkpoints -->
+
+## entities and slots
+* greet{"cuisine": "italian"}
+    - slot{"cuisine": "italian"}
+    - utter_default
+    - utter_default <!-- just so length is different from the checkpoint story -->

tests/core/test_dsl.py

@@ -23,11 +23,13 @@ async def test_can_read_test_story(default_domain):
         tracker_limit=1000,
         remove_duplicates=False,
     )
-    assert len(trackers) == 7
+    assert len(trackers) == 8
     # this should be the story simple_story_with_only_end -> show_it_all
     # the generated stories are in a non stable order - therefore we need to
     # do some trickery to find the one we want to test
-    tracker = [t for t in trackers if len(t.events) == 5][0]
+    trackers = [t for t in trackers if len(t.events) == 5]
+    assert len(trackers) == 1
+    tracker = trackers[0]
     assert tracker.events[0] == ActionExecuted("action_listen")
     assert tracker.events[1] == UserUttered(
         "simple",

tests/core/test_training.py

@@ -23,9 +23,9 @@ async def test_story_visualization(default_domain, tmpdir):
         should_merge_nodes=False,
     )
 
-    assert len(generated_graph.nodes()) == 51
+    assert len(generated_graph.nodes()) == 54
 
-    assert len(generated_graph.edges()) == 56
+    assert len(generated_graph.edges()) == 60
 
 
 async def test_story_visualization_with_merging(default_domain):

tests/core/test_visualization.py

@@ -90,12 +90,13 @@ async def test_graph_persistence(default_domain, tmpdir):
         should_merge_nodes=False,
     )
 
-    generated_graph = nx_pydot.to_pydot(generated_graph)
+    generated_graph = str(nx_pydot.to_pydot(generated_graph))
 
     assert isfile(out_file)
 
     with open(out_file, "r") as graph_file:
         content = graph_file.read()
 
+    assert r'label="/greet\{\"cuisine\":\ \"italian\"\}"' in generated_graph
     assert "isClient = true" in content
-    assert "graph = `{}`".format(generated_graph.to_string()) in content
+    assert "graph = `{}`".format(generated_graph) in content