summaryrefslogtreecommitdiffstats
path: root/Lib
diff options
context:
space:
mode:
Diffstat (limited to 'Lib')
-rwxr-xr-xLib/test/test_dbm.py16
1 files changed, 12 insertions, 4 deletions
diff --git a/Lib/test/test_dbm.py b/Lib/test/test_dbm.py
index b4f7f89..94949cf 100755
--- a/Lib/test/test_dbm.py
+++ b/Lib/test/test_dbm.py
@@ -6,7 +6,7 @@ import dbm
from dbm import error
from test_support import verbose
-filename= '/tmp/delete_me'
+filename = '/tmp/delete_me'
d = dbm.open(filename, 'c')
d['a'] = 'b'
@@ -15,7 +15,7 @@ d.keys()
if d.has_key('a'):
if verbose:
print 'Test dbm keys: ', d.keys()
-
+
d.close()
d = dbm.open(filename, 'r')
d.close()
@@ -28,7 +28,15 @@ d.close()
try:
import os
- os.unlink(filename + '.dir')
- os.unlink(filename + '.pag')
+ if dbm.library == "ndbm":
+ # classic dbm
+ os.unlink(filename + '.dir')
+ os.unlink(filename + '.pag')
+ elif dbm.library == "BSD db":
+ # BSD DB's compatibility layer
+ os.unlink(filename + '.db')
+ else:
+ # GNU gdbm compatibility layer
+ os.unlink(filename)
except:
pass