python timeitの使用について

704 ワード

1、テスト単行コード
import timeit

timeit.timeit('x=1',number=10)

2、関数テスト
import timeit

def myfunc():
    return 100

timeit.timeit('myfunc()','from __main__ import myfunc',number=10)

3、複数回のテスト
import timeit

t = timeit.repeat('myfunc()','from __main__ import myfunc',number=10,repeat=5)

NOTE:
パラメータは渡されないようです.例えば
import timeit
import pandas as pd

df = pd.DataFrame({'a':[1,2,3]})

def myfunc(df):
    return 100
timeit.timeit('myfunc(df)','from __main__ import myfunc',number=10)

この場合、テストが必要な機能をパラメータのない関数を再定義してテストするしかありません.