def test_update_pizzas(): order = Order(1, [['pepperoni', 'medium', ['olives', 'chicken']]], [['pepsi', 'small']]) order.update_pizzas([['vegetarian', 'small', ['beef']]]) assert order.get_pizzas()[0].get_type() == 'vegetarian' assert order.get_pizzas()[0].get_size() == 'small' assert order.get_pizzas()[0].get_toppings() == ['beef'] order.update_pizzas([['margherita', 'medium', ['tomatoes', 'olives']]]) assert order.get_pizzas()[0].get_type() == 'margherita' assert order.get_pizzas()[0].get_size() == 'medium' assert order.get_pizzas()[0].get_toppings() == ['tomatoes', 'olives']
def test_order(): order = Order(1, [['pepperoni', 'medium', ['olives', 'chicken']]], [['pepsi', 'small']]) assert order.get_order_number() == 1 assert order.get_pizzas()[0].get_type() == 'pepperoni' assert order.get_pizzas()[0].get_size() == 'medium' assert order.get_pizzas()[0].get_toppings() == ['olives', 'chicken'] assert order.get_drinks()[0].get_type() == 'pepsi' assert order.get_drinks()[0].get_size() == 'small' order = Order(2, [['vegetarian', 'small', ['beef']]], [['pepsi', 'small'], ['coke', 'medium']]) assert order.get_order_number() == 2 assert order.get_pizzas()[0].get_type() == 'vegetarian' assert order.get_pizzas()[0].get_size() == 'small' assert order.get_pizzas()[0].get_toppings() == ['beef'] assert order.get_drinks()[0].get_type() == 'pepsi' assert order.get_drinks()[0].get_size() == 'small' assert order.get_drinks()[1].get_type() == 'coke' assert order.get_drinks()[1].get_size() == 'medium' try: order = Order(3, [['invalid', 'invalid', ['invalid']]], [['invalid', 'invalid']]) assert False except: assert True try: order = Order(4, [['pepperoni', 'small', 'invalid']], [['coke', 'small']]) assert False except: assert True try: order = Order(5, [['pepperoni', 'small', ['beef']]], ['invalid']) assert False except: assert True try: order = Order(6, [['pepperoni', 'small', ['beef']]], 'invalid') assert False except: assert True
def test_get_pizzas(): order = Order(1, [['pepperoni', 'medium', ['olives', 'chicken']]], [['pepsi', 'small']]) assert order.get_pizzas()[0].get_type() == 'pepperoni' assert order.get_pizzas()[0].get_size() == 'medium' assert order.get_pizzas()[0].get_toppings() == ['olives', 'chicken'] order = Order(2, [['vegetarian', 'small', ['chicken']]], [['pepsi', 'small']]) assert order.get_pizzas()[0].get_type() == 'vegetarian' assert order.get_pizzas()[0].get_size() == 'small' assert order.get_pizzas()[0].get_toppings() == ['chicken'] order.add_pizza(['pepperoni', 'small', ['beef']]) assert order.get_pizzas()[1].get_type() == 'pepperoni' assert order.get_pizzas()[1].get_size() == 'small' assert order.get_pizzas()[1].get_toppings() == ['beef']