def setUp(self): self.ctx = app.test_request_context() self.ctx.push() db.drop_all() db.create_all() init_data() self.client = app.test_client()
def tearDown(self): db.session.rollback() db.drop_all() self.ctx.pop()
def drop(env): "Drops database tables" init_for(env) if prompt_bool("Are you sure you want to lose all your data?"): db.drop_all()