python - 使用 timeit 为不同的测试语句只运行一次设置代码

Question

使用 timeit，我有一个 setup 代码块，它设置了一个填充了虚拟数据的数据结构，我有两个语句(比如，test1 和 test2)，它们以不同的方式从此数据结构中检索数据。

当我做的时候

t = timeit.Timer(test1, setup)
print t.timeit(3000)
t = timeit.Timer(test2, setup)
print t.timeit(3000)

我注意到设置运行了两次，每个测试运行一次。是否可以使两个测试共享相同的设置，即只运行一次 setup 代码块并使用为两个测试创建的完全相同的数据结构？

Answer 1

最简单的实现方法是将设置代码放入模块setup.py，然后使用

"from setup import name_1, name_2, name_3"

作为 setup 参数到 timeit.timeit()。 (请注意，在这种情况下，通配符导入将无法正常工作。)

如果不想单独使用模块，把代码放在主模块中使用

"from __main__ import name_1, name_2, name_3"