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)