Python Drink.get_size 예제들

예제 #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_size():
    drink = Drink(['coke', 'medium'])
    assert drink.get_size() == 'medium'

    drink = Drink(['coke', 'small'])
    assert drink.get_size() == 'small'