Create mem_profile.py

CoreyMSchafer · web-flow · commit aa0b245265ac · 2017-01-06T14:43:37.000-07:00
diff --git a/mem_profile.py b/mem_profile.py
@@ -0,0 +1,19 @@
+from pympler import summary, muppy
+import psutil
+import resource
+import os
+import sys
+
+def memory_usage_psutil():
+    # return the memory usage in MB
+    process = psutil.Process(os.getpid())
+    mem = process.get_memory_info()[0] / float(2 ** 20)
+    return mem
+
+def memory_usage_resource():
+    rusage_denom = 1024.
+    if sys.platform == 'darwin':
+        # ... it seems that in OSX the output is different units ...
+        rusage_denom = rusage_denom * rusage_denom
+    mem = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / rusage_denom
+    return mem
