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))