26 lines
772 B
Python
26 lines
772 B
Python
|
import gc
|
||
|
|
import timeit
|
||
|
|
import time
|
||
|
|
|
||
|
|
class Timer:
|
||
|
|
def __init__(self, timer=None, disable_gc=False, verbose=True):
|
||
|
|
if timer is None:
|
||
|
|
timer = timeit.default_timer
|
||
|
|
self.timer = timer
|
||
|
|
self.disable_gc = disable_gc
|
||
|
|
self.verbose = verbose
|
||
|
|
self.start = self.end = self.interval = None
|
||
|
|
def __enter__(self):
|
||
|
|
if self.disable_gc:
|
||
|
|
self.gc_state = gc.isenabled()
|
||
|
|
gc.disable()
|
||
|
|
self.start = self.timer()
|
||
|
|
return self
|
||
|
|
def __exit__(self, *args):
|
||
|
|
self.end = self.timer()
|
||
|
|
if self.disable_gc and self.gc_state:
|
||
|
|
gc.enable()
|
||
|
|
self.interval = self.end - self.start
|
||
|
|
if self.verbose:
|
||
|
|
print('time taken: %f seconds' % self.interval)
|