1import numpy as np 2import ffht 3import timeit 4import sys 5 6reps = 1000 7n = 2**20 8chunk_size = 1024 9 10a = np.random.randn(n).astype(np.float32) 11 12t1 = timeit.default_timer() 13for i in range(reps): 14 ffht.fht(a) 15t2 = timeit.default_timer() 16 17if sys.version_info[0] == 2: 18 print (t2 - t1 + 0.0) / (reps + 0.0) 19if sys.version_info[0] == 3: 20 print('{}'.format((t2 - t1 + 0.0) / (reps + 0.0))) 21