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))