fix: change file location

jina-ai · Sep 30, 2020 · 797750c · 797750c
1 parent fefa597
commit 797750c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/tests/unit/drivers/test_flow_visualization.py b/tests/unit/drivers/test_flow_visualization.py
diff --git a/tests/unit/flow/test_flow_visualization.py b/tests/unit/flow/test_flow_visualization.py
@@ -0,0 +1,9 @@
+from jina.flow import Flow
+
+def test_visualization():
+    flow = (Flow().add(name='pod_a')
+                    .add(name='pod_b', needs='gateway')
+                    .join(needs=['pod_a', 'pod_b']))
+
+    flow.mermaidstr_to_jpg()
+