util.py 文件源码

def leakcheck(fn):
    # Compare the memory (after GC) before and after the test to determine
    # if we leaked memory
    def wrapper(fn, *args, **kwargs):
        gc.collect()
        mem_before = MemoryUsage()
        rv = fn(*args, **kwargs)
        gc.collect()
        mem_after = MemoryUsage()
        assert mem_before >= mem_after
        return rv
    return decorator.decorate(fn, wrapper)