fix mermaid mime tests

jupyter · Aug 23, 2023 · 12c63d1 · 12c63d1
1 parent 3032063
commit 12c63d1
Showing 1 changed file with 22 additions and 1 deletion.
diff --git a/nbconvert/exporters/tests/test_html.py b/nbconvert/exporters/tests/test_html.py
@@ -120,7 +120,28 @@ def test_mermaid_output(self):
             ]
         )
         (output, resources) = HTMLExporter(template_name="lab").from_notebook_node(nb)
-        self.assertIn("""<div class="jp-Mermaid"><pre class="mermaid">""", output)
+        self.assertIn("""<div class="jp-Mermaid">""", output)
+        self.assertIn("""<pre class="mermaid">""", output)
+
+
+    def test_mermaid_prerendered_output(self):
+        nb = v4.new_notebook(
+            cells=[
+                v4.new_code_cell(
+                    outputs=[
+                        v4.new_output(
+                            output_type="display_data",
+                            data={
+                                "image/svg+xml": "<svg></svg>",
+                                "text/vnd.mermaid": "flowchart LR\na --> b",
+                            },
+                        )
+                    ]
+                )
+            ]
+        )
+        (output, resources) = HTMLExporter(template_name="lab").from_notebook_node(nb)
+        self.assertNotIn("""<div class="jp-Mermaid">""", output)
 
     def test_attachments(self):
         (output, resources) = HTMLExporter(template_name="classic").from_file(