python代码测试benchmark设计

Python 的 `time` 模块提供了一个 `timeit` 函数，可以用于测试代码的性能。

`timeit` 函数接受三个参数：

- `stmt`：要测试的代码块，可以是一个字符串形式的语句，也可以是一个函数对象。
- `setup`：运行测试代码前需要执行的代码块，可以是一个字符串形式的语句，也可以是一个函数对象。
- `timer`：计时器函数，用于计算代码执行时间，默认为系统计时器。

基本的 benchmark 测试框架可以设计如下：

import timeit

def benchmark(stmt, setup='pass', timer=timeit.default_timer):
    """
    测试代码执行时间的函数
    :param stmt: 要测试的代码块，可以是一个字符串形式的语句，也可以是一个函数对象
    :param setup: 运行测试代码前需要执行的代码块，可以是一个字符串形式的语句，也可以是一个函数对象
    :param timer: 计时器函数，用于计算代码执行时间，默认为系统计时器
    :return: 测试结果，返回一个元组，包括执行时间和执行结果
    """
    t0 = timer()
    result = eval(stmt, globals(), locals())
    t1 = timer()
    return t1 - t0, result

使用这个 `benchmark` 函数可以测试单个代码块的执行时间，例如：

import numpy as np

stmt = 'np.random.rand(1000)'
setup = 'import numpy as np'

result = benchmark(stmt, setup)
print(result)

输出结果类似：

(6.959999999997114e-05, array([0.56775618, 0.82598723, 0.33654756, ..., 0.18083606, 0.56682199,
       0.33647645]))

其中第一个元素为代码执行时间，第二个元素为代码执行结果。您可以根据需要进行扩展，例如批量测试一组代码块的执行时间，或者进行多次测试取平均值等。