tests/test_simple.py

from nbconvert import ConverterRST, main, converters
import nose.tools as nt

import os
import glob
import tempfile
import shutil
from IPython.nbformat import current as nbformat


class TestSimple(object):
    def setUp(self):
        """
        Create a new temporary directory and copy the input file
        'test.ipynb' there.
        """
        self.fname = 'test.ipynb'
        self.basename = os.path.splitext(self.fname)[0]
        self.wrkdir = tempfile.mkdtemp()
        self.infile = os.path.join(self.wrkdir, self.fname)
        shutil.copy(os.path.join('tests', self.fname), self.infile)

    def tearDown(self):
        shutil.rmtree(self.wrkdir)

    def outfile_exists(self, fmt):
        extension = converters[fmt].extension
        return os.path.exists(os.path.join(self.wrkdir,
                                           self.basename + '.' + extension))

    def test_simple(self):
        c = ConverterRST(self.infile)
        f = c.render()
        nt.assert_true(f.endswith('.rst'), 'changed file extension to rst')

    def run_main(self, fmt):
        """
        Run the 'main' method to convert the input file to the given
        format and check that the expected output file exists.
        """
        main(self.infile, format=fmt)
        nt.assert_true(self.outfile_exists(fmt))

    def test_main(self):
        """
        Test main entry point with all known formats.
        """
        for fmt in converters:
            yield self.run_main, fmt

    def test_render_heading(self):
        """
        Unit test for cell type "heading"
        """
        # Generate and test heading cells level 1-6
        for level in xrange(1, 7):
            cell = {
                'cell_type': 'heading',
                'level': level,
                'source':  ['Test for heading type H{0}'.format(level)]
                }

            # 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)

            # render should return a list
            nt.assert_true(isinstance(rst_list, list))
            rst_str = "".join(rst_list)

            # Confirm rst content
            chk_str = "Test for heading type H{0}\n{1}\n".format(
                 level, c.heading_level[level] * 24)
            nt.assert_equal(rst_str, chk_str)