def test_create4(self): """Test create logic - no options""" TempfileManager.sequential_files(2) fname = TempfileManager.create_tempfile() OUTPUT = open(fname, 'w') OUTPUT.write('tempfile\n') OUTPUT.close() self.assertEqual(len(list(glob.glob(tempdir + '*'))), 1) fname = os.path.basename(fname) self.assertEqual(fname, 'tmp2') # TempfileManager.unique_files() fname = TempfileManager.create_tempfile() OUTPUT = open(fname, 'w') OUTPUT.write('tempfile\n') OUTPUT.close() self.assertEqual(len(list(glob.glob(tempdir + '*'))), 2) fname = os.path.basename(fname) self.assertNotEqual(fname, 'tmp3') self.assertTrue(fname.startswith('tmp'))
def tearDown(self): global tmpdir TempfileManager.clear_tempfiles() TempfileManager.unique_files() os.chdir(tmpdir)
def tearDownModule(): TempfileManager.clear_tempfiles() TempfileManager.tempdir = None TempfileManager.unique_files()