From fa78a7a45c6b86577e99b00f5be06918b65af1b7 Mon Sep 17 00:00:00 2001
From: Dominic Kempf <dominic.kempf@iwr.uni-heidelberg.de>
Date: Fri, 8 Sep 2017 09:26:44 +0200
Subject: [PATCH] Fix glitch in if/else assertion
---
pymbolic/interop/sympy.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pymbolic/interop/sympy.py b/pymbolic/interop/sympy.py
index a9626cf..b480771 100644
--- a/pymbolic/interop/sympy.py
+++ b/pymbolic/interop/sympy.py
@@ -67,7 +67,7 @@ class SympyToPymbolicMapper(SympyLikeToPymbolicMapper):
# We only handle piecewises with 2 arguments!
assert len(expr.args) == 2
# We only handle if/else cases
- assert expr.args[0][1] is True
+ assert expr.args[1][1].is_Boolean and bool(expr.args[1][1]) is True
then = self.rec(expr.args[0][0])
else_ = self.rec(expr.args[1][0])
cond = self.rec(expr.args[0][1])
--
GitLab