Skip to content

Instantly share code, notes, and snippets.

@dmmeteo
Forked from britonad/mem.py
Created July 15, 2020 13:28
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save dmmeteo/b2b92cae708805efe21d61916d4ec11c to your computer and use it in GitHub Desktop.
Save dmmeteo/b2b92cae708805efe21d61916d4ec11c to your computer and use it in GitHub Desktop.
A dummy decorator function to measure memory usage of a Python function, method, etc.
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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment