def database(action): if action == 'create': db.create_all() elif action == 'drop': db.drop_all() else: print(u'Invalid action: "{}"'.format(action)) sys.exit(1)
def tearDown(self): db.session.remove() db.drop_all()