def setUp(self): from pygr.sqlgraph import SQLiteServerInfo if not sqlite_enabled(): raise SkipTest self.sqlite_file = tempdatafile('test_sqlite.db', False) self.tearDown(False) # delete the file if it exists self.serverInfo = SQLiteServerInfo(self.sqlite_file) self.sqlite_load() # load data provided by subclass method
class SQLite_Mixin(object): 'use this as a base for any test' def setUp(self): from pygr.sqlgraph import SQLiteServerInfo if not sqlite_enabled(): raise SkipTest self.sqlite_file = tempdatafile('test_sqlite.db', False) self.tearDown(False) # delete the file if it exists self.serverInfo = SQLiteServerInfo(self.sqlite_file) self.sqlite_load() # load data provided by subclass method def tearDown(self, closeConnection=True): 'delete the sqlite db file after (optionally) closing connection' if closeConnection: self.serverInfo.close() # close the database try: os.remove(self.sqlite_file) except OSError: pass