コード例 #1
0
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']
コード例 #2
0
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
コード例 #3
0
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']