def test_get_order_number(): order = Order(1, [['pepperoni', 'medium', ['olives', 'chicken']]], [['pepsi', 'small']]) assert order.get_order_number() == 1 order = Order(2, [['pepperoni', 'medium', ['olives', 'chicken']]], [['pepsi', 'small']]) assert order.get_order_number() == 2
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