stepik code revisor stage

#Learn more: https://stepik.org/lesson/9173

stepik (background revisor code)

import random

def generate():  #create test cases
    num_tests = 50
    tests = []
    tests.append("5\n")
    tests.append("11\n")
    tests.append("18\n")
    for test in range(num_tests):
        test_case = "%d\n" % (test)
        tests.append(test_case)
    tests.append("200\n")    
    return tests

def solve(dataset):  #real solution
    n = int(dataset.strip())

    if (n == 0):
        f = 0;
    elif (n == 1):
        f = 1;
    else:
        fm1 = 0
        fm2 = 1
        for i in range(n):
            f = fm1 + fm2
            fm2 = fm1
            fm1 = f

    
    return str(f)


def check(reply, clue):  #validator
    return int(reply) == int(clue)