test_dbm.py 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. from test import test_support
  2. import unittest
  3. dbm = test_support.import_module('dbm')
  4. class DbmTestCase(unittest.TestCase):
  5. def setUp(self):
  6. self.filename = test_support.TESTFN
  7. self.d = dbm.open(self.filename, 'c')
  8. self.d.close()
  9. def tearDown(self):
  10. for suffix in ['', '.pag', '.dir', '.db']:
  11. test_support.unlink(self.filename + suffix)
  12. def test_keys(self):
  13. self.d = dbm.open(self.filename, 'c')
  14. self.assertEqual(self.d.keys(), [])
  15. a = [('a', 'b'), ('12345678910', '019237410982340912840198242')]
  16. for k, v in a:
  17. self.d[k] = v
  18. self.assertEqual(sorted(self.d.keys()), sorted(k for (k, v) in a))
  19. for k, v in a:
  20. self.assertIn(k, self.d)
  21. self.assertEqual(self.d[k], v)
  22. self.assertNotIn('xxx', self.d)
  23. self.assertRaises(KeyError, lambda: self.d['xxx'])
  24. self.d.close()
  25. def test_modes(self):
  26. for mode in ['r', 'rw', 'w', 'n']:
  27. try:
  28. self.d = dbm.open(self.filename, mode)
  29. self.d.close()
  30. except dbm.error:
  31. self.fail()
  32. def test_main():
  33. test_support.run_unittest(DbmTestCase)
  34. if __name__ == '__main__':
  35. test_main()