PEP8

tests/test_references.py

@@ -1,46 +1,48 b''
 import os
 import io
 import nose.tools as nt
-from nbconvert import *
+from nbconvert import (
+    ConverterLaTeX, ConverterMarkdown, ConverterPy, ConverterHTML
+)
 from nose.tools import nottest
 
 
 def test_evens():
     ###### 
     # for now, we don't need to really run inkscape to extract svg 
     # from file, on unix, for test, we monkeypathc it to 'true'
     # which does not fail as doing anything.
     ####
     ConverterLaTeX.inkscape = 'true'
 
     # commenting rst for now as travis build 
     # fail because of pandoc version.
     converters = [
                  #(ConverterRST,'rst'),
                   (ConverterMarkdown,'md'),
                   (ConverterLaTeX,'tex'),
                   (ConverterPy,'py'),
                   (ConverterHTML,'html')
                 ]
     reflist = [
             'tests/ipynbref/IntroNumPy.orig'
             ]
     for root in reflist :
         for conv,ext in converters:
             yield test_conversion, conv,root+'.ipynb',root+'.'+ext
 
+
 @nottest
 def compfiles(stra, strb):
-    nt.assert_equal(map(unicode.strip,stra.split('\n')),map(unicode.strip,strb.split('\n')))
+    nt.assert_equal(map(unicode.strip, stra.split('\n')),
+                    map(unicode.strip, strb.split('\n')))
+
 
 @nottest
 def test_conversion(ConverterClass, ipynb, ref_file):
-
     converter = ConverterClass(ipynb)
     converter.read()
     cv =converter.convert()
     with io.open(ref_file) as ref:
         value = ref.read()
         compfiles(cv,value)
-    
-

tests/test_simple.py

@@ -1,70 +1,71 b''
 from nbconvert import ConverterRST, main
 import nose.tools as nt
 
 import os
 import glob
 from IPython.nbformat import current as nbformat
 
 fname = 'tests/test.ipynb'
 out_fname = 'tests/test.rst'
 
 
 def clean_dir():
     "Remove .rst files created during conversion"
     map(os.remove, glob.glob("./tests/*.rst"))
     map(os.remove, glob.glob("./tests/*.png"))
     map(os.remove, glob.glob("./tests/*.html"))
     map(os.remove, glob.glob("./tests/test_files/*"))
 
 
 @nt.with_setup(clean_dir, clean_dir)
 def test_simple():
     c = ConverterRST(fname)
     f = c.render()
     nt.assert_true('rst' in f, 'changed file extension to rst')
 
 
 @nt.with_setup(clean_dir, clean_dir)
 def test_main():
     """
     Test main entry point
     """
     main(fname)
     nt.assert_true(os.path.exists(out_fname))
 
 
 def test_render_heading():
     """ 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)
-        nt.assert_true(isinstance(rst_list, list))  # render should return a list
+        # 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)
 
 
 @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'))