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