def test_update_drinks(): order = Order(1, [['pepperoni', 'medium', ['olives', 'chicken']]], [['pepsi', 'small']]) order.update_drinks([['coke', 'small'], ['pepsi', 'medium']]) assert order.get_drinks()[0].get_type() == 'coke' assert order.get_drinks()[0].get_size() == 'small' assert order.get_drinks()[1].get_type() == 'pepsi' assert order.get_drinks()[1].get_size() == 'medium'
def test_get_drinks(): order = Order(1, [['pepperoni', 'medium', ['olives', 'chicken']]], [['pepsi', 'small']]) assert order.get_drinks()[0].get_type() == 'pepsi' assert order.get_drinks()[0].get_size() == 'small' order = Order(2, [['pepperoni', 'medium', ['olives', 'chicken']]], [['coke', 'medium'], ['pepsi', 'medium']]) assert order.get_drinks()[0].get_type() == 'coke' assert order.get_drinks()[0].get_size() == 'medium' assert order.get_drinks()[1].get_type() == 'pepsi' assert order.get_drinks()[1].get_size() == 'medium' order.add_drink(['coke', 'small']) assert order.get_drinks()[2].get_type() == 'coke' assert order.get_drinks()[2].get_size() == 'small'
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