def setup_db():
""" Sets up database for test cases """
DB.drop_tables([Customer])
# DB.close()
DB.create_tables([Customer])
add_customer(1, "Leo", "Messi", "100 Las Ramblas", 1234567890,
"*****@*****.**", True, 5000000.00)
add_customer(2, "Ivan", "Rakitic", "111 Las Ramblas", 2345678901,
"*****@*****.**", True, 2000000.00)
add_customer(3, "Gerard", "Pique", "123 Las Ramblas", 3333333333,
"*****@*****.**", True, 300000.00)
def setup_db():
""" Sets up database for test cases """
DB.drop_tables([Customer])
DB.close()
DB.create_tables([Customer])
DB.close()