Create files in a tmp dir, clean up afterwards

theano/misc/tests/test_pkl_utils.py

 import numpy
 from numpy.testing import assert_allclose
 from nose.plugins.skip import SkipTest
+import os
+from tempfile import mkdtemp
+import shutil
+import unittest
 
 import theano
 import theano.sandbox.cuda as cuda_ndarray
@@ -11,7 +15,21 @@ from theano.sandbox.rng_mrg import MRG_RandomStreams
 from theano.misc.pkl_utils import dump, load
 
 
-def test_dump_load():
+class T_dump_load(unittest.TestCase):
+    def setUp(self):
+        # Work in a temporary directory to avoid cluttering the repository
+        self.origdir = os.getcwd()
+        self.tmpdir = None
+        self.tmpdir = mkdtemp()
+        os.chdir(self.tmpdir)
+
+    def tearDown(self):
+        # Get back to the original dir, and delete the temporary one
+        os.chdir(self.origdir)
+        if self.tmpdir is not None:
+            shutil.rmtree(self.tmpdir)
+
+    def test_dump_load(self):
         if not cuda_ndarray.cuda_enabled:
             raise SkipTest('Optional package cuda disabled')
 
@@ -27,8 +45,7 @@ def test_dump_load():
         assert x.name == 'x'
         assert_allclose(x.get_value(), [[1]])
 
-
-def test_dump_load_mrg():
+    def test_dump_load_mrg(self):
         rng = MRG_RandomStreams(use_cuda=cuda_ndarray.cuda_enabled)
 
         with open('test', 'wb') as f:
@@ -39,8 +56,7 @@ def test_dump_load_mrg():
 
         assert type(rng) == MRG_RandomStreams
 
-
-def test_dump_zip_names():
+    def test_dump_zip_names(self):
         foo_1 = theano.shared(0, name='foo')
         foo_2 = theano.shared(1, name='foo')
         foo_3 = theano.shared(2, name='foo')

theano/scan_module/tests/test_scan.py

@@ -256,7 +256,7 @@ class T_Scan(unittest.TestCase):
             finally:
                 f_in.close()
         finally:
-            # Get back to the orinal dir, and delete temporary one.
+            # Get back to the original dir, and delete the temporary one.
             os.chdir(origdir)
             if tmpdir is not None:
                 shutil.rmtree(tmpdir)

theano/tests/test_tutorial.py

@@ -878,7 +878,7 @@ class T_loading_and_saving(unittest.TestCase):
                     loaded_objects.append(pickle.load(f))
                 f.close()
             finally:
-                # Get back to the orinal dir, and temporary one.
+                # Get back to the original dir, and delete the temporary one.
                 os.chdir(origdir)
                 if tmpdir is not None:
                     shutil.rmtree(tmpdir)