# It might be better to invert the conditional
# putting the early exit at the top of the
# function

# BEFORE
# Passes test for f(0), f(1)
def f(term):
    # assert term >= 0
    if term > 2:
        # This part of the implementation
        # does not return a correct value,
        # BUT preserves the behavior that
        # passed the test.
        return term

    if term > 1:
        # This is the block of code that we
        # expect to change when we make the
        # RED -> GREEN transition for f(2)
        return term

    # This part of the implementation
    # returns the correct value
    return term
end

# AFTER
# Passes test for f(0), f(1), f(2)
def f(term):
    # assert term >= 0
    if term > 2:
        # This part of the implementation
        # does not return a correct value,
        # BUT preserves the behavior that
        # passed the test.
        return term

    if term > 1:
        # Simplest thing that could possibly work?
        return 1

    # This part of the implementation
    # returns the correct value
    return term
end