From d0d365ca48128f5f48bd97cccd71b04bcc690e99 Mon Sep 17 00:00:00 2001
From: Isuru Fernando <isuruf@gmail.com>
Date: Fri, 28 Jun 2019 17:38:58 -0400
Subject: [PATCH] Fix test error
---
sumpy/expansion/__init__.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/sumpy/expansion/__init__.py b/sumpy/expansion/__init__.py
index 37ada3c0..2e741eec 100644
--- a/sumpy/expansion/__init__.py
+++ b/sumpy/expansion/__init__.py
@@ -422,8 +422,8 @@ class LinearPDEBasedExpansionTermsWrangler(ExpansionTermsWrangler):
pde = self.get_single_pde()
assert len(pde.eqs) == 1
pde_dict = pde.eqs[0]
- max_mi_idx = max(coeff_ident_enumerate_dict[ident.mi] for
- ident in pde_dict.keys())
+ max_mi_idx = max(coeff_ident_enumerate_dict[ident.mi] for ident in
+ pde_dict.keys() if ident.mi in coeff_ident_enumerate_dict)
max_mi = mis[max_mi_idx]
max_mi_coeff = pde_dict[CoeffIdentifier(max_mi, 0)]
max_mi_mult = -1/sym.sympify(max_mi_coeff)
--
GitLab