Skip to content
Snippets Groups Projects
Commit 81a6c748 authored by Timothy A. Smith's avatar Timothy A. Smith
Browse files

add FluxDerivativeArrays object and setup fixture for multiple arrays

parent df8afa77
No related branches found
No related tags found
1 merge request!8Test refactoring
......@@ -8,6 +8,7 @@ class FluxDerivativeParams:
self.nx = nx
self.ny = ny
self.nz = nz
self.nhalo = 6
class FluxDerivativeArrays:
......@@ -20,3 +21,18 @@ class FluxDerivativeArrays:
def random_array(*shape):
return np.random.random_sample(shape).astype(np.float32).copy(order="F")
def random_flux_derivative_arrays(params):
nvars = params.nvars
ndim = params.ndim
nx_halo = params.nx + params.nhalo
ny_halo = params.ny + params.nhalo
nz_halo = params.nz + params.nhalo
states = random_array(nvars, nx_halo, ny_halo, nz_halo)
fluxes = random_array(nvars, ndim, nx_halo, ny_halo, nz_halo)
metrics = random_array(ndim, ndim, nx_halo, ny_halo, nz_halo)
metric_jacobians = random_array(nx_halo, ny_halo, nz_halo)
return FluxDerivativeArrays(states, fluxes, metrics, metric_jacobians)
......@@ -31,6 +31,8 @@ def test_matvec(ctx_factory):
def test_compute_flux_derivatives(ctx_factory):
params = setup.FluxDerivativeParams(ndim=3, nvars=5, nx=10, ny=10, nz=10)
arrays = setup.random_flux_derivative_arrays(params)
ndim = 3
nvars = 5
nx = 10
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment