Clean up simple_test.py:

  - Use test class for better setup/teardown methods and sharing data.

  - Use temporary directory to do the conversion in to avoid cluttering
    the test/ directory.

tests/test_simple.py

@@ -3,38 +3,46 b' import nose.tools as nt'
 
 import os
 import glob
+import tempfile
+import shutil
 from IPython.nbformat import current as nbformat
 
-fname = 'tests/test.ipynb'
-out_fname = 'tests/test.rst'
 
+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 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/*"))
+    def tearDown(self):
+        shutil.rmtree(self.wrkdir)
 
+    def outfile_exists(self, fmt):
+        return os.path.exists(os.path.join(self.wrkdir,
+                                           self.basename + '.' + fmt))
 
[e-mail]_setup(clean_dir, clean_dir)
-def test_simple():
-    c = ConverterRST(fname)
+    def test_simple(self):
+        c = ConverterRST(self.infile)
     f = c.render()
     nt.assert_true(f.endswith('.rst'), 'changed file extension to rst')
 
-
[e-mail]_setup(clean_dir, clean_dir)
-def test_main():
+    def test_main(self):
     """
     Test main entry point
     """
-    main(fname)
-    nt.assert_true(os.path.exists(out_fname))
+        main(self.infile)
+        nt.assert_true(self.outfile_exists('rst'))
 
-
-def test_render_heading():
-    """ Unit test for cell type "heading" """
+    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 = {
@@ -42,30 +50,33 b' def test_render_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)
 
-
[e-mail]_setup(clean_dir, clean_dir)
-def test_main_html():
+    def test_main_html(self):
     """
     Test main entry point
     """
-    main(fname, format='html')
-    nt.assert_true(os.path.exists('tests/test.html'))
+        main(self.infile, format='html')
+        nt.assert_true(self.outfile_exists('html'))