From c9370b1a42c188adf982662c4195792c5cd85045 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Mon, 25 Sep 2017 18:08:23 -0500 Subject: [PATCH] Fix tests for passing tree to fmm_level_to_nterms --- test/test_fmm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_fmm.py b/test/test_fmm.py index c1df96a6..179994d5 100644 --- a/test/test_fmm.py +++ b/test/test_fmm.py @@ -210,7 +210,7 @@ def test_sumpy_fmm(ctx_getter, knl, local_expn_class, mpole_expn_class): partial(local_expn_class, knl), out_kernels) wrangler = wcc.get_wrangler(queue, tree, dtype, - fmm_level_to_order=lambda lev: order, + fmm_level_to_order=lambda tree, lev: order, kernel_extra_kwargs=extra_kwargs) from boxtree.fmm import drive_fmm @@ -283,7 +283,7 @@ def test_sumpy_fmm_exclude_self(ctx_getter): exclude_self=True) wrangler = wcc.get_wrangler(queue, tree, dtype, - fmm_level_to_order=lambda lev: order, + fmm_level_to_order=lambda tree, lev: order, self_extra_kwargs=self_extra_kwargs) from boxtree.fmm import drive_fmm -- GitLab