add perf test

improvement-remove_generate_array
Oli 3 years ago
parent 9c57554345
commit 90bd8b0995

@ -0,0 +1,56 @@
import numpy as np
#from tests.utils.gen_sample import generate
import sys
from FFTMA import gen
import time
def generate(N):
start_time = time.time()
nx, ny, nz = N,N,N
dx, dy, dz = 1.0, 1.0, 1.0
seed= 1548762 #rdi(10000,99999)
var=1
vario=2
alpha=1
lcx=2
lcy=4
lcz=16
ap1x=1
ap1y=0
ap1z=0
ap2x=0
ap2y=1
ap2z=0
v1 = (var, vario, alpha, lcx, lcy, lcz, ap1x, ap1y, ap1z, ap2x, ap2y, ap2z)
variograms = [v1]
mean=15.3245987
variance=3.5682389
typ=3
k = gen(nx, ny, nz, dx, dy, dz, seed, variograms, mean, variance, typ)
duration = time.time() - start_time
return k, duration
N=int(sys.argv[1])
SAMPLE_SIZE=1000
def time_measurement(n, sample_size):
sample = []
for i in range(sample_size):
_, duration = generate(n)
print(i)
sample.append(duration)
print("sali")
npsample = np.array(sample)
print(np.mean(npsample))
print(np.median(npsample))
time_measurement(N, SAMPLE_SIZE)

@ -1,6 +1,7 @@
from FFTMA import gen
import numpy as np
import unittest
from FFTMA import gen
import time
def generate(N):
@ -30,18 +31,18 @@ def generate(N):
k = gen(nx, ny, nz, dx, dy, dz, seed, variograms, mean, variance, typ)
print(f"Generation with N = {N} time = {time.time() - start_time}s")
return k
duration = time.time() - start_time
return k, duration
def test(N):
k = generate(N)
k, duration = generate(N)
k_correct = np.load(f"out_{N}.npy")
comparison = k == k_correct
print(f"Generation with N = {N} time = {duration}s")
return comparison.all()
class TestGeneration(unittest.TestCase):
def test_8(self):
def test_08(self):
self.assertTrue(test(8))
def test_16(self):

@ -0,0 +1,32 @@
from FFTMA import gen
import time
def generate(N):
start_time = time.time()
nx, ny, nz = N,N,N
dx, dy, dz = 1.0, 1.0, 1.0
seed= 1548762 #rdi(10000,99999)
var=1
vario=2
alpha=1
lcx=2
lcy=4
lcz=16
ap1x=1
ap1y=0
ap1z=0
ap2x=0
ap2y=1
ap2z=0
v1 = (var, vario, alpha, lcx, lcy, lcz, ap1x, ap1y, ap1z, ap2x, ap2y, ap2z)
variograms = [v1]
mean=15.3245987
variance=3.5682389
typ=3
k = gen(nx, ny, nz, dx, dy, dz, seed, variograms, mean, variance, typ)
duration = time.time() - start_time
return k, duration
Loading…
Cancel
Save