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):
        pre = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / 1024
        print(f'Pre-memory usage: {pre} (mb)')
        f = func(*args, **kwargs)
        post = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / 1024
        print(f'Post-memory usage: {post} (mb)')
        print(f'Used by {func.__name__}: {post - pre} (mb)')

        return f

    return wrapper