diff options
Diffstat (limited to 'Lib/bsddb/test/test_join.py')
-rw-r--r-- | Lib/bsddb/test/test_join.py | 15 |
1 files changed, 5 insertions, 10 deletions
diff --git a/Lib/bsddb/test/test_join.py b/Lib/bsddb/test/test_join.py index 0e41152..5657bd7 100644 --- a/Lib/bsddb/test/test_join.py +++ b/Lib/bsddb/test/test_join.py @@ -1,6 +1,7 @@ """TestCases for using the DB.join and DBCursor.join_item methods. """ +import shutil import sys, os import tempfile import time @@ -13,7 +14,7 @@ except ImportError: have_threads = 0 import unittest -from .test_all import verbose +from bsddb.test.test_all import verbose from bsddb import db, dbshelve, StringKeys @@ -47,19 +48,13 @@ class JoinTestCase(unittest.TestCase): def setUp(self): self.filename = self.__class__.__name__ + '.db' - homeDir = os.path.join(tempfile.gettempdir(), 'db_home') - self.homeDir = homeDir - try: os.mkdir(homeDir) - except os.error: pass + self.homeDir = tempfile.mkdtemp() self.env = db.DBEnv() - self.env.open(homeDir, db.DB_CREATE | db.DB_INIT_MPOOL | db.DB_INIT_LOCK ) + self.env.open(self.homeDir, db.DB_CREATE | db.DB_INIT_MPOOL | db.DB_INIT_LOCK ) def tearDown(self): self.env.close() - import glob - files = glob.glob(os.path.join(self.homeDir, '*')) - for file in files: - os.remove(file) + shutil.rmtree(self.homeDir) def test01_join(self): if verbose: |