simulacion-permeabilidad/fftma_module/gen/mem_profile.py

from example import generate

from memory_profiler import memory_usage
import sys

if __name__ == '__main__':
    N=int(sys.argv[1])
     
    def run_gen():
        generate(N)

    mem_usage = memory_usage(run_gen)
    print('Maximum memory usage: %s' % max(mem_usage))