add asymline expansion for the AxisTargetDerivative

Author: ShawnL00

sumpy/qbx.py

@@ -96,21 +96,30 @@ def _expand(self, sac, avec, bvec, rscale, isrc):
         conv = PymbolicToSympyMapper()
         strengths = [conv(self.get_strength_or_not(isrc, idx))
                      for idx in range(len(self.source_kernels))]
+
         coefficients = self.expansion.coefficients_from_source_vec(
             self.source_kernels, avec, bvec, rscale=rscale, weights=strengths,
             sac=sac)
 
         return coefficients
 
     def _evaluate(self, sac, avec, bvec, rscale, expansion_nr, coefficients):
-        from sumpy.expansion.local import LineTaylorLocalExpansion
+        from sumpy.expansion.local import LineTaylorLocalExpansion, AsymLineTaylorLocalExpansion
         tgt_knl = self.target_kernels[expansion_nr]
-        if isinstance(tgt_knl, LineTaylorLocalExpansion):
+        if isinstance(self.expansion, LineTaylorLocalExpansion):
             # In LineTaylorLocalExpansion.evaluate, we can't run
             # postprocess_at_target because the coefficients are assigned
             # symbols and postprocess with a derivative will make them zero.
             # Instead run postprocess here before the coefficients are assigned.
-            coefficients = [tgt_knl.postprocess_at_target(coeff, bvec) for
+            coefficients = [tgt_knl.postprocess_at_target(coeff, avec) for
+                    coeff in coefficients]
+            
+        if isinstance(self.expansion, AsymLineTaylorLocalExpansion):
+            # In LineTaylorLocalExpansion.evaluate, we can't run
+            # postprocess_at_target because the coefficients are assigned
+            # symbols and postprocess with a derivative will make them zero.
+            # Instead run postprocess here before the coefficients are assigned.
+            coefficients = [tgt_knl.postprocess_at_target(coeff, avec) for
                     coeff in coefficients]
 
         assigned_coeffs = [
@@ -512,7 +521,6 @@ def find_jump_term(kernel, arg_provider):
         AxisTargetDerivative,
         DerivativeBase,
         DirectionalSourceDerivative,
-        DirectionalTargetDerivative,
     )
 
     tgt_derivatives = []
@@ -522,9 +530,6 @@ def find_jump_term(kernel, arg_provider):
         if isinstance(kernel, AxisTargetDerivative):
             tgt_derivatives.append(kernel.axis)
             kernel = kernel.kernel
-        elif isinstance(kernel, DirectionalTargetDerivative):
-            tgt_derivatives.append(kernel.dir_vec_name)
-            kernel = kernel.kernel
         elif isinstance(kernel, AxisSourceDerivative):
             src_derivatives.append(kernel.axis)
             kernel = kernel.kernel