def db_factory(): path = os.path.join(get_data_dir(), DBNAME) db = dvc_db_factory(path) # from pychron.database.adapters.isotope_adapter import IsotopeAdapter # from pychron.database.orms.isotope.util import Base # # db = IsotopeAdapter() # db.verbose_retrieve_query = True # db.trait_set(kind='sqlite', path=os.path.join(get_data_dir(), DBNAME)) # db.connect() # # if os.path.isfile(db.path): # os.remove(db.path) # # metadata = Base.metadata # db.create_all(metadata) # with db.session_ctx() as sess: # metadata.create_all(sess.bind) return db
def setUpClass(cls): cls.db = dvc_db_factory(os.path.join(get_data_dir(), 'similar.db'), remove=False, echo=False)