test_simple.py
71 lines
| 2.1 KiB
| text/x-python
|
PythonLexer
/ tests / test_simple.py
|
r6252 | from nbconvert import ConverterRST, main | ||
|
r6239 | import nose.tools as nt | ||
import os | ||||
import glob | ||||
|
r6259 | from IPython.nbformat import current as nbformat | ||
|
r6239 | |||
|
r6252 | fname = 'tests/test.ipynb' | ||
out_fname = 'tests/test.rst' | ||||
|
r6239 | def clean_dir(): | ||
"Remove .rst files created during conversion" | ||||
|
r6246 | map(os.remove, glob.glob("./tests/*.rst")) | ||
map(os.remove, glob.glob("./tests/*.png")) | ||||
|
r6261 | map(os.remove, glob.glob("./tests/*.html")) | ||
|
r8647 | map(os.remove, glob.glob("./tests/test_files/*")) | ||
|
r6261 | |||
|
r6239 | |||
@nt.with_setup(clean_dir, clean_dir) | ||||
def test_simple(): | ||||
c = ConverterRST(fname) | ||||
f = c.render() | ||||
|
r8729 | nt.assert_true(f.endswith('.rst'), 'changed file extension to rst') | ||
|
r6239 | |||
|
r6261 | |||
|
r6252 | @nt.with_setup(clean_dir, clean_dir) | ||
def test_main(): | ||||
""" | ||||
Test main entry point | ||||
""" | ||||
main(fname) | ||||
nt.assert_true(os.path.exists(out_fname)) | ||||
|
r6259 | |||
|
r6261 | |||
|
r6259 | def test_render_heading(): | ||
""" Unit test for cell type "heading" """ | ||||
# Generate and test heading cells level 1-6 | ||||
|
r6261 | for level in xrange(1, 7): | ||
|
r6259 | cell = { | ||
'cell_type': 'heading', | ||||
|
r8718 | 'level': level, | ||
|
r8749 | 'source': ['Test for heading type H{0}'.format(level)] | ||
|
r6259 | } | ||
# Convert cell dictionaries to NotebookNode | ||||
cell_nb = nbformat.NotebookNode(cell) | ||||
# Make sure "source" attribute is uniconde not list. | ||||
# For some reason, creating a NotebookNode manually like | ||||
# this isn't converting source to a string like using | ||||
# the create-from-file routine. | ||||
if type(cell_nb.source) is list: | ||||
cell_nb.source = '\n'.join(cell_nb.source) | ||||
# Render to rst | ||||
c = ConverterRST('') | ||||
rst_list = c.render_heading(cell_nb) | ||||
|
r8718 | # render should return a list | ||
nt.assert_true(isinstance(rst_list, list)) | ||||
|
r6259 | rst_str = "".join(rst_list) | ||
# Confirm rst content | ||||
chk_str = "Test for heading type H{0}\n{1}\n".format( | ||||
|
r6264 | level, c.heading_level[level] * 24) | ||
|
r6261 | nt.assert_equal(rst_str, chk_str) | ||
@nt.with_setup(clean_dir, clean_dir) | ||||
def test_main_html(): | ||||
""" | ||||
Test main entry point | ||||
""" | ||||
main(fname, format='html') | ||||
nt.assert_true(os.path.exists('tests/test.html')) | ||||