def test_responses_can_be_added_to_choices():
"""A choice can have responses added to it."""
decision = Decision()
decision.add_choice('Truck')
decision.add_choice('Van')
decision.add_response('Truck', 'It goes offroad real nice')
decision.add_response('Van', 'You can sleep in it')
expected = {
'Truck': ['It goes offroad real nice'],
'Van': ['You can sleep in it'],
}
assert decision.choices == expected
def test_it_can_return_the_choice_with_the_most_responses():
"""The winner is the choices with the most responses."""
decision = Decision()
decision.add_choice('Truck')
decision.add_choice('Van')
decision.add_response('Truck', 'It goes offroad real nice')
decision.add_response('Van', 'You can sleep in it')
decision.add_response('Van', 'Its got a swivel chair')
decision.add_response('Van', 'It can hold more dogs')
assert decision.winner() == 'Van'