diff --git a/test/test_kernels.py b/test/test_kernels.py index 44861576043ccb14b7475ea361bbadaaea572cc5..ad53759e12126f7eca109952075f6dcb18f656bd 100644 --- a/test/test_kernels.py +++ b/test/test_kernels.py @@ -339,8 +339,10 @@ def test_translations(ctx_getter, knl, local_expn_class, mpole_expn_class): del eval_offset + from sumpy.expansion import VolumeTaylorExpansionBase + if isinstance(knl, HelmholtzKernel) and \ - isinstance(local_expn_class, VolumeTaylorLocalExpansion): + issubclass(local_expn_class, VolumeTaylorExpansionBase): # FIXME: Embarrassing--but we run out of memory for higher orders. orders = [2, 3] else: