Source code for cobamp.utilities.test_utils

import time


def timeit(method):
	'''
	Timer decorator for methods. Courtesy of Fahim Sakri from PythonHive
	Args:
		method:

	Returns:

	'''

	def timed(*args, **kw):
		ts = time.time()
		result = method(*args, **kw)
		te = time.time()

		if 'log_time' in kw:
			name = kw.get('log_name', method.__name__.upper())
			kw['log_time'][name] = int((te - ts) * 1000)
		else:
			print('%r  %2.2f ms' % (method.__name__, (te - ts) * 1000))
		return result

	return timed