From b947da3225619f38d8b3a3c58ac980c571df6601 Mon Sep 17 00:00:00 2001 From: Matt Wala Date: Thu, 11 Apr 2019 18:50:55 -0500 Subject: [PATCH] Fix use of Indexed for sympy 1.4 (closes #11) This test passes for me on sympy 1.3 as well, so I think the change is backwards compatible. --- pymbolic/interop/sympy.py | 2 +- test/test_sympy.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pymbolic/interop/sympy.py b/pymbolic/interop/sympy.py index 76ef76c..86f50dc 100644 --- a/pymbolic/interop/sympy.py +++ b/pymbolic/interop/sympy.py @@ -109,7 +109,7 @@ class PymbolicToSympyMapper(PymbolicToSympyLikeMapper): *[self.sym.Symbol(v) for v in expr.variables]) def map_subscript(self, expr): - return self.sym.tensor.indexed.Indexed( + return self.sym.Indexed( self.rec(expr.aggregate), *tuple(self.rec(i) for i in expr.index_tuple) ) diff --git a/test/test_sympy.py b/test/test_sympy.py index d8a5c94..3cd55ac 100644 --- a/test/test_sympy.py +++ b/test/test_sympy.py @@ -54,7 +54,7 @@ def _test_to_pymbolic(mapper, sym, use_symengine): # indexed accesses if not use_symengine: i, j = sym.symbols("i,j") - assert mapper(sym.tensor.indexed.Indexed(x, i, j)) == x_[i_, j_] + assert mapper(sym.Indexed(x, i, j)) == x_[i_, j_] # constants import math @@ -100,7 +100,7 @@ def _test_from_pymbolic(mapper, sym, use_symengine): assert mapper(x_[0]) == sym.Symbol("x_0") else: i, j = sym.symbols("i,j") - assert mapper(x_[i_, j_]) == sym.tensor.indexed.Indexed(x, i, j) + assert mapper(x_[i_, j_]) == sym.Indexed(x, i, j) assert mapper(prim.Variable("f")(x_)) == sym.Function("f")(x) -- GitLab