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'
def test_get_type(): drink = Drink(['coke', 'medium']) assert drink.get_type() == 'coke' drink = Drink(['pepsi', 'medium']) assert drink.get_type() == 'pepsi'