def test_status_unexisting_restaurant(self):
     self.assertEqual("There is no such restaurant",
                      database.status_restaurant('speedy'))
 def test_status_existing_restaurant(self):
     database.create_menu_table('speedy')
     self.assertEqual("Status is Not busy",
                      database.status_restaurant('speedy'))
def status_restaurant():
    restaurant = input("Which restaurant's status you want to check? ")
    print(database.status_restaurant(restaurant))