3
0
Fork 0
mirror of https://github.com/Z3Prover/z3 synced 2025-04-10 19:27:06 +00:00

Fixing array select for lambda expressions in Python API (#6516)

* fix: making array select work for lambda expressions

* more elegant solution
This commit is contained in:
Walden Yan 2023-01-01 18:27:54 -05:00 committed by GitHub
parent f6d411d54b
commit dbf93c5fbd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -4597,10 +4597,10 @@ class ArrayRef(ExprRef):
def _array_select(ar, arg):
if isinstance(arg, tuple):
args = [ar.domain_n(i).cast(arg[i]) for i in range(len(arg))]
args = [ar.sort().domain_n(i).cast(arg[i]) for i in range(len(arg))]
_args, sz = _to_ast_array(args)
return _to_expr_ref(Z3_mk_select_n(ar.ctx_ref(), ar.as_ast(), sz, _args), ar.ctx)
arg = ar.domain().cast(arg)
arg = ar.sort().domain().cast(arg)
return _to_expr_ref(Z3_mk_select(ar.ctx_ref(), ar.as_ast(), arg.as_ast()), ar.ctx)