EdFdZddlmZddlmZddlmZddlmZm Z d dZ dS) z5 Functions to support rewriting of SymPy expressions )Expr)ask)subs)rebuildunifyNc$dfd }|S)a Rewrite rule. Transform expressions that match source into expressions that match target treating all ``variables`` as wilds. Examples ======== >>> from sympy.abc import w, x, y, z >>> from sympy.unify.rewrite import rewriterule >>> from sympy import default_sort_key >>> rl = rewriterule(x + y, x**y, [x, y]) >>> sorted(rl(z + 3), key=default_sort_key) [3**z, z**3] Use ``condition`` to specify additional requirements. Inputs are taken in the same order as is found in variables. >>> rl = rewriterule(x + y, x**y, [x, y], lambda x, y: x.is_integer) >>> list(rl(z + 3)) [3**z] Use ``assume`` to specify additional requirements using new assumptions. >>> from sympy.assumptions import Q >>> rl = rewriterule(x + y, x**y, [x, y], assume=Q.integer(x)) >>> list(rl(z + 3)) [3**z] Assumptions for the local context are provided at rule runtime >>> list(rl(w + z, Q.integer(z))) [z**w] Tc3.Kt|iD]|rfdDsr$t|s<t}t |t rt |}|V}dS)N) variablesc<g|]}||Sr)get).0varmatchs 3lib/python3.11/site-packages/sympy/unify/rewrite.py z3rewriterule..rewrite_rl../s'III #s 3 3III)rrxreplacer isinstancerr) expr assumptionsexpr2rassume conditionsourcetargetr s @r rewrite_rlzrewriterule..rewrite_rl,s64yAAA  E IIIIIyIIIJ  s6??5#9#9;GG DKK''E%&& 'KKKK  r)Tr)rrr rrrs````` r rewriterulersBH           r)rNN) __doc__sympy.core.exprrsympy.assumptionsrsympy.strategies.toolsrsympy.unify.usympyrrrrrrr$sw;; !!!!!!''''''--------//////r