utils.py

import time
import matplotlib.pyplot as plt
import random


def createData(sizes=[]):
    """
    生成指定数量的测试数据
    :param sizes: 数据规模大小列表
    :return: 生成的点列表
    """
    datasets = []
    for size in sizes:
        data = [random.randint(1, 100000) for _ in range(size)]
        datasets.append(data)
    return datasets

def run_algorithm(algorithm, data): #测试算法运行时间
    start_time = time.time()
    algorithm(data)
    end_time = time.time()
    return (end_time - start_time) * 1000

def plot_performance_curve(algorithm, data_sets):
    sizes = [len(data) for data in data_sets]
    times = [run_algorithm(algorithm, data) for data in data_sets]
    plt.plot(sizes, times, '-o')
    plt.xlabel("Data Size")
    plt.ylabel("Running Time (ms)")
    plt.title("Performance Curve for {}".format(algorithm.__name__))
    plt.savefig("Performance_Curve_for_{}".format(algorithm.__name__))
    plt.show()