tests/dbutils.py

   1
   2 import os
   3 from m_librarian.db import open_db, init_db
   4 from m_librarian.inp import import_inpx
   5
   6 __all__ = ['setup_module', 'teardown_module', 'load_inpx']
   7
   8
   9 def setup_module():
  10     try:
  11         os.remove('/tmp/m_librarian-test.sqlite')
  12     except OSError:
  13         pass
  14     open_db('sqlite:///tmp/m_librarian-test.sqlite')
  15     init_db()
  16
  17
  18 def teardown_module():
  19     try:
  20         os.remove('/tmp/m_librarian-test.sqlite')
  21     except OSError:
  22         pass
  23
  24
  25 def load_inpx(inpx):
  26     import_inpx(os.path.join(os.path.dirname(__file__), inpx))