def tearDown(self): Customer.delete().execute() DB.close()
def setup_db(): """ Sets up database for test cases """ DB.drop_tables([Customer]) DB.close() DB.create_tables([Customer]) DB.close()