Python Drink.get_typeの例

コード例 #1

def test_update():
    drink = Drink(['coke', 'medium'])
    drink.update(['pepsi', -1])
    assert drink.get_type() == 'pepsi'

    drink.update([-1, 'small'])
    assert drink.get_size() == 'small'

    drink.update(['invalid', 'invalid'])
    assert drink.get_type() == 'pepsi'
    assert drink.get_size() == 'small'

コード例 #2

def test_get_type():
    drink = Drink(['coke', 'medium'])
    assert drink.get_type() == 'coke'

    drink = Drink(['pepsi', 'medium'])
    assert drink.get_type() == 'pepsi'