python: measure time code

import time

# Calculate the power of two for a defined range of number
def power_two(my_range):
    return [x**2 for x in range(my_range)]
 
# Measure time
def measure_time(func):
    start = time.time()
    func()               # any specific function to measure
    end = time.time()
    print(end - start)  
    
measure_time(lambda: power_two(10000000)) # lambda permits to pass the argument of our function

import time
start_time = time.time()
main()
print("--- %s seconds ---" % (time.time() - start_time))