Python timeit Problem

Question

Ich versuche, das Timeit-Modul zu verwenden, aber ich weiß nicht wie. Ich habe einen Haupt:

from Foo import Foo 
if __name__ == '__main__': 
... 
    foo = Foo(arg1, arg2) 
    t = Timer("foo.runAlgorithm()") 
    print t.timeit(2) 

und meine Klasse Foo ein Verfahren, wie runAlgorithm genannt hat()

der Fehler ist dies:

NameError: global name 'foo' is not defined

Was mache ich falsch? Kann ich mir die Zeit von einer Klassenmethode nehmen?

Answer 1

Statt die notwendige setup parameter der Verwendung für die timeit Umgebung einrichten, können Sie einfach die Methode übergeben (oder alles, was aufrufbar ist):

t = Timer(foo.runAlgorithm) 

Aus der Dokumentation:

Changed in version 2.6: The stmt and setup parameters can now also take objects that are callable without arguments.

Wenn Sie einige Argumente zu übergeben müssen, können Sie Funktion mit functools.partial, zum Beispiel currying verwenden:

class C: 
    def printargs(self, a, b): 
     print a, b 

from functools import partial 
foo = C() 
t = Timer(partial(foo.printargs, 1, 2)) 

Answer 2

als docs example zeigen Sie setup Aussagen übergeben müssen:

t = Timer("foo.runAlgorithm()", 'from __main__ import foo')