diff options
Diffstat (limited to 'Lib/bsddb/test/test_pickle.py')
| -rw-r--r-- | Lib/bsddb/test/test_pickle.py | 11 |
1 files changed, 4 insertions, 7 deletions
diff --git a/Lib/bsddb/test/test_pickle.py b/Lib/bsddb/test/test_pickle.py index d691e37..b1d3805 100644 --- a/Lib/bsddb/test/test_pickle.py +++ b/Lib/bsddb/test/test_pickle.py @@ -6,7 +6,6 @@ try: except ImportError: cPickle = None import unittest -import tempfile try: # For Pythons w/distutils pybsddb @@ -15,6 +14,8 @@ except ImportError, e: # For Python 2.3 from bsddb import db +from test_all import get_new_environment_path, get_new_database_path + try: from bsddb3 import test_support except ImportError: @@ -25,14 +26,10 @@ except ImportError: class pickleTestCase(unittest.TestCase): """Verify that DBError can be pickled and unpickled""" - db_home = 'db_home' db_name = 'test-dbobj.db' def setUp(self): - homeDir = os.path.join(tempfile.gettempdir(), 'db_home%d'%os.getpid()) - self.homeDir = homeDir - try: os.mkdir(homeDir) - except os.error: pass + self.homeDir = get_new_environment_path() def tearDown(self): if hasattr(self, 'db'): @@ -47,7 +44,7 @@ class pickleTestCase(unittest.TestCase): self.db = db.DB(self.env) self.db.open(self.db_name, db.DB_HASH, db.DB_CREATE) self.db.put('spam', 'eggs') - assert self.db['spam'] == 'eggs' + self.assertEqual(self.db['spam'], 'eggs') try: self.db.put('spam', 'ham', flags=db.DB_NOOVERWRITE) except db.DBError, egg: |