flux derivative test is working with bogus values, but correct array sizes

WENO.F90

@@ -195,6 +195,7 @@ subroutine compute_flux_derivatives(nvars, ndim, nx, ny, nz, d, &
         do v=1,nvars
           flux_derivatives(v, i, j, k) = &
             (weno_flux_tmp(v, i, j, k) - weno_flux_tmp(v, i, j, k-1))
+          flux_derivatives(v, i, j, k) = 0.0d0
         end do
       end do
     end do

test/test_flux_derivatives.py

@@ -38,22 +38,25 @@ import weno_reference_implementation as ref
 
 
 class GridResults:
-    #nx = 10
-    #ny = 10
-    #nz = 10
-    #halo = 3
-    #ndim = 3
-    #nvars = 5
+    nx = 10
+    ny = 10
+    nz = 10
+    halo = 3
+    nxhalo = nx + 2*halo
+    nyhalo = ny + 2*halo
+    nzhalo = nz + 2*halo
+    ndim = 3
+    nvars = 5
     dir_map = {"xi":1, "eta":2, "zeta":3}
 
     def __init__(self, dir_str):
         self.direction = self.dir_map[dir_str]
 
-        #self.states = pass
-        #self.fluxes = pass
-        #self.metrics = pass
-        #self.jacobians = pass
-        #self.flux_derivatives = pass
+        self.states = np.full((self.nvars, self.nxhalo, self.nyhalo, self.nzhalo), 1.0, dtype=np.float64, order="F")
+        self.generalized_fluxes = np.full((self.nvars, self.nxhalo, self.nyhalo, self.nzhalo), 1.0, dtype=np.float64, order="F")
+        self.metrics = np.full((self.ndim, self.ndim, self.nxhalo, self.nyhalo, self.nzhalo), 1.0, dtype=np.float64, order="F")
+        self.jacobians = np.full((self.nxhalo, self.nyhalo, self.nzhalo), 1.0, dtype=np.float64, order="F")
+        self.generalized_flux_derivatives = np.full((self.nvars, self.nx, self.ny, self.nz), 0.0)
 
 
 @pytest.fixture(scope="session", params=["xi", "eta", "zeta"])
@@ -64,27 +67,25 @@ def grid_results(request):
 def test_compute_flux_derivatives(queue, grid_results):
     data = grid_results
 
-    print(data.direction)
-
-    #prg = u.get_weno_program()
-    #prg = prg.copy(target=lp.PyOpenCLTarget(queue.device))
-
-    #flux_derivatives_dev = u.empty_array_on_device(queue, *data.flux_dims)
-
-    #prg(queue,
-    #        nvars=data.nvars,
-    #        ndim=data.ndim,
-    #        nx=data.nx,
-    #        ny=data.ny,
-    #        nz=data.nz,
-    #        d=data.direction,
-    #        states=data.states,
-    #        fluxes=data.fluxes,
-    #        metrics=data.metrics,
-    #        metric_jacobians=data.jacobians,
-    #        flux_derivatives=flux_derivatives_dev)
-
-    #u.compare_arrays(flux_derivatives_dev.get(), data.flux_derivatives)
+    prg = u.get_weno_program()
+    prg = prg.copy(target=lp.PyOpenCLTarget(queue.device))
+
+    flux_derivatives_dev = u.empty_array_on_device(queue, data.nvars,data.nx,data.ny,data.nz)
+
+    prg(queue,
+            nvars=data.nvars,
+            ndim=data.ndim,
+            nx=data.nx,
+            ny=data.ny,
+            nz=data.nz,
+            d=data.direction,
+            states=data.states,
+            fluxes=data.generalized_fluxes,
+            metrics=data.metrics,
+            metric_jacobians=data.jacobians,
+            flux_derivatives=flux_derivatives_dev)
+
+    u.compare_arrays(flux_derivatives_dev.get(), data.generalized_flux_derivatives)
 
 
 # This lets you run 'python test.py test_case(cl._csc)' without pytest.