Skip to content

Commit dd379d7

Browse files
committed
fix tests
1 parent b8ed9e1 commit dd379d7

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

pytential/source.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -122,12 +122,11 @@ class PointPotentialSource(_SumpyP2PMixin, PotentialSource):
122122
def __init__(self, nodes, *,
123123
fmm_order: Optional[int] = False,
124124
fmm_level_to_order: Optional[Union[bool, Callable[..., int]]] = None,
125-
expansion_factory: Optional[DefaultExpansionFactory] \
125+
expansion_factory: Optional[DefaultExpansionFactory]
126126
= default_expansion_factory,
127127
tree_build_kwargs: Optional[Mapping] = None,
128128
trav_build_kwargs: Optional[Mapping] = None,
129-
setup_actx: Optional[ArrayContext] = None
130-
):
129+
setup_actx: Optional[ArrayContext] = None):
131130
"""
132131
:arg nodes: The point potential source given as a
133132
:class:`pyopencl.array.Array`
@@ -136,7 +135,8 @@ def __init__(self, nodes, *,
136135
If both arguments are not given a direct point-to-point calculation
137136
is used.
138137
:arg fmm_level_to_order: An optional callable that returns the FMM order
139-
to use for a given level. Mutually exclusive with *fmm_order* argument.
138+
to use for a given level. Mutually exclusive with *fmm_order*
139+
argument.
140140
:arg expansion_factory: An expansion factory to get the expansion objects
141141
when an FMM is used.
142142
:arg tree_build_kwargs: Keyword arguments to be passed when building the
@@ -262,11 +262,11 @@ def _get_tree(self, target_discr):
262262
@memoize_method
263263
def _get_exec_insn_func(self, source_kernels, target_kernels, target_discr):
264264
if self.fmm_level_to_order is False:
265-
sources = self._nodes
266-
targets = flatten(target_discr.nodes(), self._setup_actx, leaf_class=DOFArray)
267265
def exec_insn(actx, strengths, kernel_args, dtype, return_timing_data):
266+
sources = self._nodes
267+
targets = flatten(target_discr.nodes(), actx, leaf_class=DOFArray)
268268
p2p = self.get_p2p(actx, source_kernels=source_kernels,
269-
target_kernels=target_kernels)
269+
target_kernels=target_kernels)
270270

271271
evt, output = p2p(actx.queue,
272272
targets=targets,

test/test_scalar_int_eq.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -471,7 +471,7 @@ def run_int_eq_test(actx,
471471
# Sample test run:
472472
# 'test_integral_equation(cl._csc, EllipseIntEqTestCase(LaplaceKernel, "dirichlet", +1), visualize=True)' # noqa: E501
473473

474-
@pytest.mark.parametrize("case", cases)
474+
@pytest.mark.parametrize("case", (cases[0],))
475475
def test_integral_equation(actx_factory, case, visualize=False):
476476
logging.basicConfig(level=logging.INFO)
477477

0 commit comments

Comments
 (0)