test_markdown.py 1.1 KB

1234567891011121314151617181920212223242526272829
  1. def get_rendered_contents(nb):
  2. cl = ["text_cell", "render"]
  3. rendered_cells = [cell.find_element_by_class_name("text_cell_render")
  4. for cell in nb.cells
  5. if all([c in cell.get_attribute("class") for c in cl])]
  6. return [x.get_attribute('innerHTML').strip()
  7. for x in rendered_cells
  8. if x is not None]
  9. def test_markdown_cell(notebook):
  10. nb = notebook
  11. cell_text = ["# Foo",
  12. '**Bar**',
  13. '*Baz*',
  14. '```\nx = 1\n```',
  15. '```aaaa\nx = 1\n```',
  16. ]
  17. expected_contents = ['<h1 id="Foo">Foo<a class="anchor-link" href="#Foo">¶</a></h1>',
  18. '<p><strong>Bar</strong></p>',
  19. '<p><em>Baz</em></p>',
  20. '<pre><code>x = 1</code></pre>',
  21. '<pre><code class="cm-s-ipython language-aaaa">x = 1</code></pre>'
  22. ]
  23. nb.append(*cell_text, cell_type="markdown")
  24. nb.run_all()
  25. rendered_contents = get_rendered_contents(nb)
  26. assert rendered_contents == expected_contents