add mem profile from outside

fftma_module/gen/example.py

@@ -2,9 +2,8 @@ from FFTMA import gen
 import numpy as np
 import sys
 
-N=int(sys.argv[1])
-
-nx, ny, nz = N,N,N
+def generate(n):
+    nx, ny, nz = n,n,n
     dx, dy, dz = 1.0, 1.0, 1.0
     seed= 1548762 #rdi(10000,99999)
     var=1
@@ -28,4 +27,8 @@ variance=3.5682389
     typ=3
 
     k=gen(nx, ny, nz, dx, dy, dz, seed, variograms, mean, variance, typ)
-np.save(f"out_{N}.npy",k)
+    np.save(f"out_{n}.npy",k)
+
+if __name__ == '__main__':
+    N=int(sys.argv[1])
+    generate(N)

fftma_module/gen/mem_profile.py

@@ -0,0 +1,16 @@
+from sympy import N
+from example import generate
+
+from memory_profiler import memory_usage
+
+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))