From 21ddcebc8f11df5381e4395b2f2ba5d105c6b08d Mon Sep 17 00:00:00 2001
From: Isuru Fernando <idf2@illinois.edu>
Date: Wed, 22 Sep 2021 02:18:25 -0500
Subject: [PATCH] pass translation_classes_data in tests
---
test/test_fmm.py | 17 +++++++++++------
1 file changed, 11 insertions(+), 6 deletions(-)
diff --git a/test/test_fmm.py b/test/test_fmm.py
index f7bf0540..73b80ef0 100644
--- a/test/test_fmm.py
+++ b/test/test_fmm.py
@@ -36,6 +36,7 @@ from sumpy.expansion.local import (
VolumeTaylorLocalExpansion,
H2DLocalExpansion, Y2DLocalExpansion,
LinearPDEConformingVolumeTaylorLocalExpansion)
+from sumpy.fmm import SumpyTranslationClassesData
import pytest
import warnings
@@ -177,7 +178,6 @@ def test_sumpy_fmm(ctx_factory, knl, local_expn_class, mpole_expn_class,
target_kernels = [knl]
from sumpy.fmm import (SumpyExpansionWranglerCodeContainer,
- SumpyTranslationClassesData,
SumpyTranslationClassesDataNotSuppliedWarning)
if optimized_m2l:
@@ -301,7 +301,8 @@ def test_unified_single_and_double(ctx_factory):
strength_usage=strength_usage)
wrangler = wcc.get_wrangler(queue, tree, dtype,
fmm_level_to_order=lambda kernel, kernel_args, tree, lev: order,
- source_extra_kwargs=source_extra_kwargs)
+ source_extra_kwargs=source_extra_kwargs,
+ translation_classes_data=SumpyTranslationClassesData(queue, trav))
from boxtree.fmm import drive_fmm
@@ -362,7 +363,8 @@ def test_sumpy_fmm_timing_data_collection(ctx_factory):
target_kernels)
wrangler = wcc.get_wrangler(queue, tree, dtype,
- fmm_level_to_order=lambda kernel, kernel_args, tree, lev: order)
+ fmm_level_to_order=lambda kernel, kernel_args, tree, lev: order,
+ translation_classes_data=SumpyTranslationClassesData(queue, trav))
from boxtree.fmm import drive_fmm
timing_data = {}
@@ -421,7 +423,8 @@ def test_sumpy_fmm_exclude_self(ctx_factory):
wrangler = wcc.get_wrangler(queue, tree, dtype,
fmm_level_to_order=lambda kernel, kernel_args, tree, lev: order,
- self_extra_kwargs=self_extra_kwargs)
+ self_extra_kwargs=self_extra_kwargs,
+ translation_classes_data=SumpyTranslationClassesData(queue, trav))
from boxtree.fmm import drive_fmm
@@ -496,7 +499,8 @@ def test_sumpy_axis_source_derivative(ctx_factory):
wrangler = wcc.get_wrangler(queue, tree, dtype,
fmm_level_to_order=lambda kernel, kernel_args, tree, lev: order,
- self_extra_kwargs=self_extra_kwargs)
+ self_extra_kwargs=self_extra_kwargs,
+ translation_classes_data=SumpyTranslationClassesData(queue, trav))
from boxtree.fmm import drive_fmm
@@ -562,7 +566,8 @@ def test_sumpy_target_point_multiplier(ctx_factory):
wrangler = wcc.get_wrangler(queue, tree, dtype,
fmm_level_to_order=lambda kernel, kernel_args, tree, lev: order,
- self_extra_kwargs=self_extra_kwargs)
+ self_extra_kwargs=self_extra_kwargs,
+ translation_classes_data=SumpyTranslationClassesData(queue, trav))
from boxtree.fmm import drive_fmm
--
GitLab