sort_time_complexity.py

import os
import time
import numpy as np
#=================================================================================================
#                                       TQDM Installation
#=================================================================================================
try:
    from tqdm import tqdm
except:
    print("'tqdm' module not found.");
    print("installing tqdm modle...")
    os.system('pip install tqdm')

#=================================================================================================
#                                  FUNCTION TO RUN EXPERIMENTS
#=================================================================================================

def run_experiment(algorithm,desc='',STEP=1000, REPS_PER_ARR=10, MAX_ARR_LEN=10000):
    len_of_arr = []
    exec_times = []
    for i in tqdm(range(1,MAX_ARR_LEN+STEP, STEP),desc='Experiment Progress '+'('+desc+')'):
        array = np.random.randint(1000, size=i).tolist()
        cumulative_execution_time=0
        for j in range(REPS_PER_ARR):
            start = time.perf_counter()
            algorithm(array)
            end = time.perf_counter()
            cumulative_execution_time = cumulative_execution_time + (end-start)
        avg_execution_time = cumulative_execution_time/REPS_PER_ARR
        len_of_arr.append(i)
        exec_times.append(avg_execution_time)
    return  len_of_arr,exec_times