2

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

当我做

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

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

4

1 回答 1

4

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

"from setup import name_1, name_2, name_3"

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

如果您不想使用单独的模块,请将代码放在主模块中并使用

"from __main__ import name_1, name_2, name_3"
于 2012-07-25T16:28:13.627 回答