diff --git a/IPython/nbconvert/writers/tests/test_stdout.py b/IPython/nbconvert/writers/tests/test_stdout.py new file mode 100644 index 0000000..648cf02 --- /dev/null +++ b/IPython/nbconvert/writers/tests/test_stdout.py @@ -0,0 +1,46 @@ +""" +Module with tests for stdout +""" + +#----------------------------------------------------------------------------- +# 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 +#----------------------------------------------------------------------------- + +import sys +from StringIO import StringIO + +from ...tests.base import TestsBase +from ..stdout import StdoutWriter + + +#----------------------------------------------------------------------------- +# Class +#----------------------------------------------------------------------------- + +class TestStdout(TestsBase): + """Contains test functions for stdout.py""" + + def test_output(self): + """Test stdout writer output.""" + + # Capture the stdout. Remember original. + stdout = sys.stdout + stream = StringIO() + sys.stdout = stream + + # Create stdout writer, test output + writer = StdoutWriter() + writer.write('a', {'b': 'c'}) + output = stream.getvalue() + self.fuzzy_compare(output, 'a') + + # Revert stdout + sys.stdout = stdout \ No newline at end of file