A dummy decorator function to measure memory usage of a Python function, method, etc.

mem.py

import resource
from functools import wraps


def mem_it(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        print(
            'Pre-memory usage: {} (mb)'.format(
                resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / 1024
            )
        )
        f = func(*args, **kwargs)
        print(
            'Post-memory usage: {} (mb)'.format(
                resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / 1024
            )
        )

        return f

    return wrapper