def drop_all():
"""Drops all database tables"""
input_var = input('Are you sure you wish to drop all tables? y/n: ')
if input_var == 'y' or input_var == 'Y':
db.reflect()
db.drop_all()
print('Database tables dropped')
def db(app):
_db.drop_all()
_db.create_all()
populate_db()
yield _db
_db.session.remove()
_db.drop_all()
def db(app):
print("Creating database")
_db.drop_all()
_db.create_all()
populate_db()
yield _db
print("Dropping database")
_db.session.remove()
_db.drop_all()
def tearDown(self):
db.session.remove()
db.drop_all()
