Helmholtz not checked

Author: hirish99

test/test_recurrence.py

@@ -61,8 +61,8 @@ def test_helmholtz_3D():
 
 def test_helmholtz_2D():
     w = make_identity_diff_op(2)
-    laplace2d = laplacian(w) + w
-    _,_, r = get_processed_and_shifted_recurrence(laplace2d)
+    helmholtz2d = laplacian(w) + w
+    _,_, r = get_processed_and_shifted_recurrence(helmholtz2d)
 
     n = sp.symbols("n")
     s = sp.Function("s")

test/test_recurrenceqbx.py

@@ -1,5 +1,6 @@
 import numpy as np
 import sympy as sp
+from sympy import hankel1
 
 from sumpy.expansion.diff_op import (
     laplacian,
@@ -8,14 +9,39 @@
 from sumpy.recurrenceqbx import recurrence_qbx_lp, _make_sympy_vec
 from sumpy.array_context import PytestPyOpenCLArrayContextFactory, _acf  # noqa: F401
 from sumpy.expansion.local import LineTaylorLocalExpansion, VolumeTaylorLocalExpansion
-from sumpy.kernel import LaplaceKernel
+from sumpy.kernel import LaplaceKernel, HelmholtzKernel
 from sumpy.qbx import LayerPotential
 
 actx_factory = _acf
 expn_class = LineTaylorLocalExpansion
 
 actx = actx_factory()
 lknl = LaplaceKernel(2)
+hlknl = HelmholtzKernel(2, "k")
+
+def _qbx_lp_helmholtz_general(sources,targets,centers,radius,strengths,order):
+        lpot = LayerPotential(actx.context,
+        expansion=expn_class(hlknl, order),
+        target_kernels=(hlknl,),
+        source_kernels=(hlknl,))
+
+        #print(lpot.get_kernel())
+        expansion_radii = actx.from_numpy(radius * np.ones(sources.shape[1]))
+        sources = actx.from_numpy(sources)
+        targets = actx.from_numpy(targets)
+        centers = actx.from_numpy(centers)
+
+        strengths = (strengths,)
+        extra_kernel_kwargs={"k": 1}
+        _evt, (result_qbx,) = lpot(
+                actx.queue,
+                targets, sources, centers, strengths,
+                expansion_radii=expansion_radii,
+                kwargs=extra_kernel_kwargs)
+        result_qbx = actx.to_numpy(result_qbx)
+
+        return result_qbx
+
 
 def _qbx_lp_laplace_general(sources,targets,centers,radius,strengths,order):
         lpot = LayerPotential(actx.context,
@@ -78,4 +104,28 @@ def test_recurrence_laplace_2d_ellipse():
         err.append(np.max(np.abs(exp_res - qbx_res)))
     assert np.max(err) <= 1e-13
 
-test_recurrence_laplace_2d_ellipse()
+
+def test_recurrence_helmholtz_2d_ellipse():
+
+    #------------- 1. Define PDE, Green's Function
+    w = make_identity_diff_op(2)
+    helmholtz2d = laplacian(w) + w
+
+    var = _make_sympy_vec("x", 2)
+    var_t = _make_sympy_vec("t", 2)
+    k = 1
+    abs_dist = sp.sqrt((var[0]-var_t[0])**2 + (var[1]-var_t[1])**2)
+    g_x_y = (1j/4) * hankel1(0, k * abs_dist)
+
+    p = 4
+    err = []
+    for n_p in range(200, 1001, 200):
+        sources, centers, normals, density, h, radius = _create_ellipse(n_p)
+        strengths = h * density
+        exp_res = recurrence_qbx_lp(sources, centers, normals, strengths, radius, helmholtz2d, g_x_y, 2, p)
+        #qbx_res = _qbx_lp_helmholtz_general(sources, sources, centers, radius, strengths, p)
+        #qbx_res,_ = lpot_eval_circle(sources.shape[1], p)
+        #err.append(np.max(np.abs(exp_res - qbx_res)))
+    #assert np.max(err) <= 1e-13
+
+test_recurrence_helmholtz_2d_ellipse()