More Machete Mode Debugging

machete_mode_redux.py

import os
from unittest import TestCase
import psutil
import humanize

pid = os.getpid()
proc = psutil.Process(pid)
real_doCleanups = TestCase.doCleanups

OUTPUT_FILE = open('/home/ssanderson/pandas16_memory_usage.txt', 'w')

def get_mem_usage():
    return humanize.naturalsize(proc.memory_full_info().uss)

def patched_doCleanups(self, *args, **kwargs):
    OUTPUT_FILE.write("{} {}\n".format(get_mem_usage(), self))
    OUTPUT_FILE.flush()
    return real_doCleanups(self, *args, **kwargs)

TestCase.doCleanups = patched_doCleanups