IPython/nbconvert/exporters/tests/test_exporter.py

"""
Module with tests for exporter.py
"""

#-----------------------------------------------------------------------------
# Copyright (c) 2013, the IPython Development Team.
#
# Distributed under the terms of the Modified BSD License.
#
# The full license is in the file COPYING.txt, distributed with this software.
#-----------------------------------------------------------------------------

#-----------------------------------------------------------------------------
# Imports
#-----------------------------------------------------------------------------

from IPython.config import Config

from .base import ExportersTestsBase
from .cheese import CheeseTransformer
from ..exporter import Exporter


#-----------------------------------------------------------------------------
# Class
#-----------------------------------------------------------------------------

class TestExporter(ExportersTestsBase):
    """Contains test functions for exporter.py"""


    def test_constructor(self):
        """
        Can an Exporter be constructed?
        """
        Exporter()


    def test_export(self):
        """
        Can an Exporter export something?
        """
        exporter = self._make_exporter()
        (output, resources) = exporter.from_filename(self._get_notebook())
        assert len(output) > 0


    def test_extract_figures(self):
        """
        If the ExtractFigureTransformer is enabled, are figures extracted?
        """
        config = Config({'ExtractFigureTransformer': {'enabled': True}})
        exporter = self._make_exporter(config=config)
        (output, resources) = exporter.from_filename(self._get_notebook())
        assert resources is not None
        assert 'figures' in resources
        assert len(resources['figures']) > 0


    def test_transformer_class(self):
        """
        Can a transformer be added to the transformers list by class type?
        """
        config = Config({'Exporter': {'transformers': [CheeseTransformer]}})
        exporter = self._make_exporter(config=config)
        (output, resources) = exporter.from_filename(self._get_notebook())
        assert resources is not None
        assert 'cheese' in resources
        assert resources['cheese'] == 'real'


    def test_transformer_instance(self):
        """
        Can a transformer be added to the transformers list by instance?
        """
        config = Config({'Exporter': {'transformers': [CheeseTransformer()]}})
        exporter = self._make_exporter(config=config)
        (output, resources) = exporter.from_filename(self._get_notebook())
        assert resources is not None
        assert 'cheese' in resources
        assert resources['cheese'] == 'real'


    def test_transformer_dottedobjectname(self):
        """
        Can a transformer be added to the transformers list by dotted object name?
        """
        config = Config({'Exporter': {'transformers': ['IPython.nbconvert.exporters.tests.cheese.CheeseTransformer']}})
        exporter = self._make_exporter(config=config)
        (output, resources) = exporter.from_filename(self._get_notebook())
        assert resources is not None
        assert 'cheese' in resources
        assert resources['cheese'] == 'real'


    def test_transformer_via_method(self):
        """
        Can a transformer be added via the Exporter convinience method?
        """
        exporter = self._make_exporter()
        exporter.register_transformer(CheeseTransformer, enabled=True)
        (output, resources) = exporter.from_filename(self._get_notebook())
        assert resources is not None
        assert 'cheese' in resources
        assert resources['cheese'] == 'real'


    def _make_exporter(self, config=None):
        #Create the exporter instance, make sure to set a template name since
        #the base Exporter doesn't have a template associated with it.
        exporter = Exporter(config=config)
        exporter.template_file = 'python'
        return exporter