def test_set_status(self):
database.create_menu_table('speedy')
database.update_status_restaurant('speedy', 'Not taking orders.')
database.cursor.execute("SELECT status \
FROM restaurants WHERE name = 'speedy'")
status = database.cursor.fetchone()
self.assertEqual('Not taking orders.', status[0])
def update_status_restaurant():
restaurant = input("Which restaurant's status you want to change: ")
new_status = input("What should be the new status? ")
database.update_status_restaurant(restaurant, new_status)