Uploaded Test files

venv/Lib/site-packages/nbconvert/exporters/tests/test_html.py

@@ -0,0 +1,148 @@
+"""Tests for HTMLExporter"""
+
+# Copyright (c) IPython Development Team.
+# Distributed under the terms of the Modified BSD License.
+
+import re
+
+from .base import ExportersTestsBase
+from ..html import HTMLExporter
+
+from traitlets.config import Config
+from nbformat import v4
+
+
+class TestHTMLExporter(ExportersTestsBase):
+    """Tests for HTMLExporter"""
+    
+    exporter_class = HTMLExporter
+    should_include_raw = ['html']
+
+    def test_constructor(self):
+        """
+        Can a HTMLExporter be constructed?
+        """
+        HTMLExporter()
+
+
+    def test_export(self):
+        """
+        Can a HTMLExporter export something?
+        """
+        (output, resources) = HTMLExporter().from_filename(self._get_notebook())
+        assert len(output) > 0
+
+
+    def test_export_classic(self):
+        """
+        Can a HTMLExporter export using the 'classic' template?
+        """
+        (output, resources) = HTMLExporter(template_name='classic').from_filename(self._get_notebook())
+        assert len(output) > 0
+
+
+    def test_export_notebook(self):
+        """
+        Can a HTMLExporter export using the 'lab' template?
+        """
+        (output, resources) = HTMLExporter(template_name='lab').from_filename(self._get_notebook())
+        assert len(output) > 0
+
+    def test_prompt_number(self):
+        """
+        Does HTMLExporter properly format input and output prompts?
+        """
+        (output, resources) = HTMLExporter(template_name='lab').from_filename(
+            self._get_notebook(nb_name="prompt_numbers.ipynb"))
+        in_regex = r"In \[(.*)\]:"
+        out_regex = r"Out\[(.*)\]:"
+
+        ins = ["2", "10", " ", " ", "0"]
+        outs = ["10"]
+
+        assert re.findall(in_regex, output) == ins
+        assert re.findall(out_regex, output) == outs
+        
+    def test_prompt_number(self):
+        """
+        Does HTMLExporter properly format input and output prompts?
+        """
+        no_prompt_conf = Config(
+            {"TemplateExporter":{
+                "exclude_input_prompt": True,
+                "exclude_output_prompt": True,
+                }
+            }
+        )
+        exporter = HTMLExporter(config=no_prompt_conf, template_name='lab')
+        (output, resources) = exporter.from_filename(
+            self._get_notebook(nb_name="prompt_numbers.ipynb"))
+        in_regex = r"In \[(.*)\]:"
+        out_regex = r"Out\[(.*)\]:"
+
+        assert not re.findall(in_regex, output)
+        assert not re.findall(out_regex, output)
+
+    def test_png_metadata(self):
+        """
+        Does HTMLExporter with the 'classic' template treat pngs with width/height metadata correctly?
+        """
+        (output, resources) = HTMLExporter(template_name='classic').from_filename(
+            self._get_notebook(nb_name="pngmetadata.ipynb"))
+        check_for_png = re.compile(r'<img src="[^"]*?"([^>]*?)>')
+        result = check_for_png.search(output)
+        attr_string = result.group(1)
+        assert 'width' in attr_string
+        assert 'height' in attr_string
+
+    def test_javascript_output(self):
+        nb = v4.new_notebook(
+            cells=[
+                v4.new_code_cell(
+                    outputs=[v4.new_output(
+                        output_type='display_data',
+                        data={
+                            'application/javascript': "javascript_output();"
+                        }
+                    )]
+                )
+            ]
+        )
+        (output, resources) = HTMLExporter(template_name='classic').from_notebook_node(nb)
+        self.assertIn('javascript_output', output)
+
+    def test_attachments(self):
+        (output, resources) = HTMLExporter(template_name='classic').from_file(
+            self._get_notebook(nb_name='attachment.ipynb')
+        )
+        check_for_png = re.compile(r'<img src="[^"]*?"([^>]*?)>')
+        result = check_for_png.search(output)
+        self.assertTrue(result.group(0).strip().startswith('<img src="data:image/png;base64,iVBOR'))
+        self.assertTrue(result.group(1).strip().startswith('alt="image.png"'))
+
+        check_for_data = re.compile(r'<img src="(?P<url>[^"]*?)"')
+        results = check_for_data.findall(output)
+        assert results[0] != results[1], 'attachments only need to be unique within a cell'
+        assert 'image/svg' in results[1], 'second image should use svg'
+
+
+    def test_custom_filter_highlight_code(self):
+        # Overwriting filters takes place at: Exporter.from_notebook_node
+        nb = v4.new_notebook()
+        nb.cells.append(v4.new_code_cell("some_text"))
+
+        def custom_highlight_code(source, language="python", metadata=None):
+            return source + " ADDED_TEXT"
+
+        filters = {
+            "highlight_code": custom_highlight_code
+        }
+        (output, resources) = HTMLExporter(template_name='classic', filters=filters).from_notebook_node(nb)
+        self.assertTrue("ADDED_TEXT" in output)
+
+    def test_basic_name(self):
+        """
+        Can a HTMLExporter export using the 'basic' template?
+        """
+        (output, resources) = HTMLExporter(template_name='basic').from_filename(self._get_notebook())
+        assert len(output) > 0