def test_story_visualization(default_domain):
story_steps = StoryFileReader.read_from_file(
"data/test_stories/stories.md", default_domain,
interpreter=RegexInterpreter())
generated_graph = visualize_stories(story_steps, default_domain,
output_file=None,
max_history=3,
should_merge_nodes=False)
assert len(generated_graph.nodes()) == 33
def visualize(self,
filename,
output_file,
max_history,
nlu_training_data=None,
fontsize=12):
from rasa_core.training.visualization import visualize_stories
from rasa_core.training import StoryFileReader
story_steps = StoryFileReader.read_from_file(filename, self.domain)
visualize_stories(story_steps, self.domain, output_file, max_history,
self.interpreter, nlu_training_data, fontsize)