Benchmark function

timeutil.py

# Function decorator to save system CPU and memory consumption
# as well as execution duration of the function supplied
import time
import psutil


def timeit(report):
  """
  Use as decorator for a function, provide a dictionary object as parameter
  """
    def timef(method):
        def timed(*args, **kwargs):
            ts = time.time()
            result = method(*args, **kwargs)
            te = time.time()

            system_status = {
                "time": (te - ts),
                "cpu": psutil.cpu_percent(),
                "mem": psutil.virtual_memory()[3] / 2.**30
            }

            report[method.__name__] = system_status
            return result
        return timed
    return timef