"""

You are driving a little too fast, and a police officer stops you.
Write code to compute the result, encoded as an int value: 0=no ticket, 1=small ticket, 2=big ticket.
If speed is 60 or less, the result is 0. If speed is between 61 and 80 inclusive, the result is 1.
If speed is 81 or more, the result is 2.
Unless it is your birthday -- on that day, your speed can be 5 higher in all cases.


caught_speeding(60, False) = 0
caught_speeding(65, False) = 1
caught_speeding(65, True) = 0
"""

def encoded_values(choice):
	return {0:'no ticket', 1:'small ticket', 2:'big ticket'}[choice]

def caught_speeding(speed, is_birthday):
	if is_birthday:
		if speed <= 65:
			# return encoded_values[0]
			return 0
		elif speed >= 61 and speed <= 85:
			return 1
		else:
			return 2
	else:
		if speed <= 60:
			# return encoded_values[0]
			return 0
		elif speed >= 61 and speed <= 80:
			return 1
		else:
			return 2



def test_caught_speeding():
	assert caught_speeding(60, False) == 0
	assert caught_speeding(65, False) == 1
	assert caught_speeding(65, True) == 0
	assert caught_speeding(81, False) == 2


# test_caught_speeding()