如何使用可调用对象作为 timeit.Timer 的设置？

Question

我想对一些依赖于某些设置的代码进行计时。设置代码看起来有点像这样：

>>> b = range(1, 1001)

我想要计时的代码看起来模糊地像这样：

>>> sorted(b)

除了我的代码使用与排序不同的函数。但现在这并不重要。

无论如何，只要我将字符串传递给 timeit，我就知道如何对这段代码进行计时：

>>> import timeit
>>> t = timeit.Timer("sorted(b)", "b = range(1, 1001)")
>>> min(t.repeat(3, 100))

如何使用 setup callable 并将其放入 stmt callable 可以访问的命名空间中？

换句话说，我如何做与上面的代码相同的事情，但是使用可调用对象，而不是包含可调用对象的字符串？

顺便说一句，这里更大的目标是重用单元测试中的代码来衡量性能：

import unittest
class TestSorting(unittest.TestCase):

    def setUp(self):
        self.b = range(1, 1001)

    def test_sorted(self):
        sorted(self.b)

我希望做一些工作。 timeit 设置需要创建 TestSorting 的实例，并且 stmt 代码必须以某种方式使用该特定实例。

一旦我了解了如何让 timeit 设置将内容放入与 timeit stmt 相同的命名空间中，我将研究如何将 unittest.TestCase 实例转换为可以直接输入 timeit.Timer 的内容。

马特

Answer 1

在 2.6 中，您可以根据 文档“直接执行” ：

2.6 版中的更改：stmt 和
setup 参数现在也可以采用
无需调用即可调用的对象
论据。这将嵌入调用
它们在计时器函数中
然后由timeit()执行。笔记
时间开销有点
在这种情况下更大，因为
额外的函数调用。

您是否还在使用旧版本的 Python？在这种情况下，我建议采用 Python 2.6 的 timer.py 源代码并将它们“反向移植”到您所使用的版本 - 如果该版本是 2.5，那么应该不难（变得更难）当然，你需要回到过去的距离越远）。我通常不建议将其用于“生产”代码，但对于支持测试、调试、性能测量等的代码来说，它非常好。

Answer 2

我在这里问了这个问题的一个变体，也得到了一个没有解决我的问题的答案。我相信我们都未能用 Pythonic 的术语来阐明这个问题。

我不知道这是否是一个黑客，解决方法，或者它应该如何完成，但它确实有效：

>>> import timeit
>>> b = range(1, 1001)
>>> t = timeit.Timer('sorted(b)', setup='from __main__ import b')
>>> t.repeat(3, 100)
[0.0024309158325195312, 0.0024671554565429688, 0.0024020671844482422]
# was it really running 'sorted(b)'? let's compare'
>>> t = timeit.Timer('pass', setup='from __main__ import b')
>>> t.repeat(3, 100) 
[3.0994415283203125e-06, 2.1457672119140625e-06, 1.9073486328125e-06] 
# teeny times for 'pass', 'sorted(b)' took more time

我已阅读 timeit.py，它通过构造一个语句然后调用 eval() 使用新的命名空间（似乎）与主命名空间没有连接。请注意，由于 sorted 是内置函数，因此 timeit.repeat() 内的 eval'ed 表达式可以访问该名称；如果它是你的 def，你必须 from __main__ import b, myfunc。

我希望有比这更合适的方法来达到目的。