Update test for 4 dimensions (3 spatial + time)

Author: isuruf

test/test_kernels.py

@@ -500,8 +500,12 @@ def test_p2e2p(actx_factory, base_knl, expn_class, order, with_source_derivative
 # {{{ test_translations
 
 @pytest.mark.parametrize("knl, local_expn_class, mpole_expn_class", [
+    (HeatKernel(1), LinearPDEConformingVolumeTaylorLocalExpansion,
+      LinearPDEConformingVolumeTaylorMultipoleExpansion),
     (HeatKernel(2), LinearPDEConformingVolumeTaylorLocalExpansion,
       LinearPDEConformingVolumeTaylorMultipoleExpansion),
+    (HeatKernel(3), LinearPDEConformingVolumeTaylorLocalExpansion,
+      LinearPDEConformingVolumeTaylorMultipoleExpansion),
     (LaplaceKernel(2), VolumeTaylorLocalExpansion, VolumeTaylorMultipoleExpansion),
     (LaplaceKernel(2), LinearPDEConformingVolumeTaylorLocalExpansion,
      LinearPDEConformingVolumeTaylorMultipoleExpansion),
@@ -542,7 +546,7 @@ def test_translations(actx_factory, knl, local_expn_class, mpole_expn_class,
 
     # Just to make sure things also work away from the origin
     rng = np.random.default_rng(18)
-    origin = np.array([0, 1, 2][-knl.dim:], np.float64)
+    origin = np.array([0, 0, 1, 2][-knl.dim:], np.float64)
     sources = actx.from_numpy(
         0.7 * (-0.5 + rng.random((knl.dim, nsources), dtype=np.float64))
         + origin[:, np.newaxis])
@@ -556,19 +560,19 @@ def test_translations(actx_factory, knl, local_expn_class, mpole_expn_class,
 
     from sumpy.visualization import FieldPlotter
 
-    eval_offset = np.array([0.0, 0.0, 5.5][-knl.dim:])
+    eval_offset = np.array([0.0, 0.0, 0.0, 5.5][-knl.dim:])
     fp = FieldPlotter(eval_offset + origin, extent=0.3, npoints=res)
 
     centers = actx.from_numpy((np.array(
             [
                 # box 0: particles, first mpole here
-                [0, 0, 0][-knl.dim:],
+                [0, 0, 0, 0][-knl.dim:],
 
                 # box 1: second mpole here
-                np.array([0, 0.1, -0.2][-knl.dim:], np.float64),
+                np.array([0, 0, 0.1, -0.2][-knl.dim:], np.float64),
 
                 # box 2: first local here
-                eval_offset + np.array([0, -0.2, 0.3][-knl.dim:], np.float64),
+                eval_offset + np.array([0, 0, -0.2, 0.3][-knl.dim:], np.float64),
 
                 # box 3: second local and eval here
                 eval_offset
@@ -577,7 +581,7 @@ def test_translations(actx_factory, knl, local_expn_class, mpole_expn_class,
 
     del eval_offset
 
-    if knl.dim == 2:
+    if knl.dim == 2 and not isinstance(knl, HeatKernel):
         orders = [2, 3, 4]
     else:
         orders = [3, 4, 5]