Check runtime with timeit

Question

I need to check the running time of a function using the module timeit of Python 3, but it does not accept that I pass the value inside the function by a variable.

If I replace the line:

tempo = timeit.timeit("comp_string(string, 'Fred')", "from __main__ import 

for

tempo = timeit.timeit("comp_string('Fred', 'Fred')", "from __main__ import 

works, but passing the value Fred in the variable does not work.

How can I fix this?
Follows the code:

'''
Comparação de Strings por Força Bruta
'''
import timeit

def comp_string(string_new, string_origin):
    for i, c in enumerate(string_origin):
        if string_new[i] != c:
            return False
    return True

def calcula_exec(string):       
    #Calcula tempo de execução do código
    tempo = timeit.timeit("comp_string(string, 'Fred')", "from __main__ import comp_string")
    return tempo

Answer 1

You can use include the literal value of the parameter string in the function of the defendant timeit using the method format(), thus:

"comp_string('{0}', 'Fred')".format(string)

The full function would be:

def calcula_exec(string):       
    #Calcula tempo de execução do código
    tempo = timeit.timeit("comp_string('{0}', 'Fred')".format(string), "from __main__ import comp_string")
    return tempo