Lib/test/test_dbm.py

   1 #! /usr/bin/env python
   2 """Test script for the dbm module
   3    Roger E. Masse
   4 """
   5 import dbm
   6 from dbm import error
   7 from test_support import verbose
   8
   9 filename = '/tmp/delete_me'
  10
  11 d = dbm.open(filename, 'c')
  12 d['a'] = 'b'
  13 d['12345678910'] = '019237410982340912840198242'
  14 d.keys()
  15 if d.has_key('a'):
  16     if verbose:
  17         print 'Test dbm keys: ', d.keys()
  18
  19 d.close()
  20 d = dbm.open(filename, 'r')
  21 d.close()
  22 d = dbm.open(filename, 'rw')
  23 d.close()
  24 d = dbm.open(filename, 'w')
  25 d.close()
  26 d = dbm.open(filename, 'n')
  27 d.close()
  28
  29 try:
  30     import os
  31     if dbm.library == "ndbm":
  32         # classic dbm
  33         os.unlink(filename + '.dir')
  34         os.unlink(filename + '.pag')
  35     elif dbm.library == "BSD db":
  36         # BSD DB's compatibility layer
  37         os.unlink(filename + '.db')
  38     else:
  39         # GNU gdbm compatibility layer
  40         os.unlink(filename)
  41 except:
  42     pass