From 75dc33f30c6e8058e6372b53cdd920cc6fbe41fe Mon Sep 17 00:00:00 2001
From: Isuru Fernando <isuruf@gmail.com>
Date: Wed, 27 Mar 2019 20:36:01 -0500
Subject: [PATCH] Fix tests

---
 sumpy/expansion/__init__.py | 4 ++--
 test/test_kernels.py        | 4 +++-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/sumpy/expansion/__init__.py b/sumpy/expansion/__init__.py
index f9ffc4f5..85d9a71b 100644
--- a/sumpy/expansion/__init__.py
+++ b/sumpy/expansion/__init__.py
@@ -400,7 +400,7 @@ class LinearRecurrenceBasedExpansionTermsWrangler(ExpansionTermsWrangler):
             n = nullspace(pde_mat)
             n = n[offset*iexpr:offset*(iexpr+1), :]
             idx = self.get_reduced_coeffs(n)
-            n = n[:,:len(idx)]
+            n = n[:, :len(idx)]
             s = solve_symbolic(n.T[:, idx], n.T)
             stored_identifiers = [mis[i] for i in idx]
         else:
@@ -417,7 +417,7 @@ class LinearRecurrenceBasedExpansionTermsWrangler(ExpansionTermsWrangler):
 
         plog.done()
 
-        logger.debug("number of Taylor coefficients was reduced from {orig} to {red}"
+        print("number of Taylor coefficients was reduced from {orig} to {red}"
                      .format(orig=len(self.get_full_coefficient_identifiers()),
                              red=len(stored_identifiers)))
 
diff --git a/test/test_kernels.py b/test/test_kernels.py
index 6a96d1ad..a1fa8230 100644
--- a/test/test_kernels.py
+++ b/test/test_kernels.py
@@ -412,8 +412,10 @@ def test_translations(ctx_getter, knl, local_expn_class, mpole_expn_class):
            issubclass(local_expn_class, VolumeTaylorExpansionBase):
         # FIXME: Embarrassing--but we run out of memory for higher orders.
         orders = [2, 3]
-    else:
+    elif isinstance(knl, StokesletKernel):
         orders = [3, 4, 5]
+    else:
+        orders = [2, 3, 4]
     nboxes = centers.shape[-1]
 
     def eval_at(e2p, source_box_nr, rscale):
-- 
GitLab