a Z>m?Z?m@Z@ddlAmBZBed\ZCZDZEZFed \ ZGZHZIZJZKZLZMZNZOZPZQZRZSd!d"ZTd#d$ZUd%d&ZVd'd(ZWd)d*ZXd+d,ZYd-d.ZZd/d0Z[d1d2Z\d3d4Z]d5d6Z^e:d7d8Z_d9d:Z`e:d;d<Zae:d=d>Zbe:d?d@ZcdAdBZddCdDZedEdFZfdGdHZgdIdJZhdKdLZidMdNZjdOdPZkdQdRZldSdTZmdUdVZndWdXZodYdZZpd[d\Zqd]d^Zrd_d`ZsdadbZtdcddZudedfZvdgdhZwe:didjZxdkdlZydmdnZzdodpZ{dqdrZ|dsdtZ}dudvZ~dwdxZdydzZd{d|Zd}d~ZddZddZdS))reduceN)add)Add)Tuple)Function)Mul)Pow)Eq)S)Symbolsymbols)sympify)exp)sqrt) Piecewise)cossin)Matrix)CRootOf)Ocse)signsimp)Idx IndexedBase count_ops)sub_presub_post)meijerg)cse_maincse_opts)subsets)XFAILraises)MutableDenseMatrixMutableSparseMatrixImmutableDenseMatrixImmutableSparseMatrix) MatrixSymbolzw,x,y,zzx:13cCstjdd}tt|ddddtddDks6Jtjdd}tt|ddddtddDkslJt}tt|ddd dtddDksJdS) Ny)prefixr cSsg|]}td|qSzy%sr .0ir2m/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/sympy/simplify/tests/test_cse.py +z)test_numbered_symbols..cSsg|]}td|qSr-r.r/r2r2r3r4.r5cSsg|]}td|qS)zx%sr.r/r2r2r3r41r5)r Znumbered_symbolslist itertoolsislicerange)nsr2r2r3test_numbered_symbols(s$  r<cCs|tSN)r*exprr2r2r3opt16sr@cCs|tSr=)zr>r2r2r3opt2:srBcCstttdfgttksJttdtfgtks8JttdgtksNJttttfgttkslJtttdftdfgtttksJdSN)NN)r Zpreprocess_for_csexr@r*rBrAr2r2r2r3test_preprocess_for_cse>s rEcCstttdfgtksJttdtfgttks8JttdgtksNJttttfgttkslJttdtfdtfgtttksJdSrC)r Zpostprocess_for_cserDr@r*rBrAr2r2r2r3test_postprocess_for_cseGs rFcCsttttdttt}t|g\}}|tttfgks@J|tttdgksZJtdg\}\}t|dkr~|dksJtdg\}\}t|dkr|dksJdS)N*r?)rrrDr*rrx0lenesubstsreducedZsubst42Zred42Z subst_halfZred_halfr2r2r3test_cse_singleQsrPcCsttttdttt}t|\}}|tttfgks>J|tttdgksXJttdgg\}}t|dts~Jtd\}\}t |dkr|dksJtd\}\}t |dkr|dksJdS)NrGrrHrI) rrrDr*rrrJr isinstancerKrLr2r2r3test_cse_single2^s rScCsxttt}t|g\}}|gks$J|ttgks6Jtdtdfdgttdtdfdgt}t|g|gfkstJdS)N)rQrG))rQ)rrDr*rr)rMrNrOeqr2r2r3test_cse_not_possiblems  rXcCsjtttttdtttt}t|g\}}|ttttfgksLJ|tttdgksfJdSNrG)rrwrDr*rrrJ)rMrNrOr2r2r3test_nested_substitutionys$r[cCsttttttttt}t|gtjtjfgd\}}|tttttfgks^J|t tt gksxJtt ttttt tt}t|gtjtjfgd\}}|tttttfgksJ|tttgksJddt}|t| dd|t}t|tjtjfgdgdgfks@Jtt tttt ttt tdtt tft ttfgt t tt tdgfksJdS)NZ optimizationsrQrGrrV) rDr*rArrr!rrrJrZx1)rMrNrOnr2r2r3test_subtraction_opts($ (   *0r`cCsttt}ttt}t||g\}}|tttfgks>J|ttttgksVJttttttg}t|\}}tt|\}}||ksJ|ttttgksJtttttttttg}t|\}}tt|\}}||ksJ|ttttgksJttttttttg}t|\}}tt|\}}|tt ftttftttfgkstJ|tt ftttftttfgksJ|ttttgksJtttttttttg}t|tttfgtttttttgfksJttttttgtttfgtttgfksJJtttttggttttgfksxJtttttttttdgtttfgtttdttgfksJdS)NrV) rDr*rArZrrJreversedr^x2)e1e2rNrOlZrsubsts_r2r2r3test_multiple_expressionss8    "   **$88.$"rgcCsxtddd\}}}|||||g}t|g|fks:J|||||g}t|t||fgt|tgfkstJdSNA B CFZ commutative)r rrJABCrer2r2r3test_non_commutative_cses rocCstddd\}}}|||||g}t|g|fks:J|||||g}t|g|fksbJ|||||g}t|g|fksJdSrhr rrkr2r2r3test_bypass_non_commutativessrqcCs\tddd\}}}tddd}|||||g}t||||fg|||gfksXJdS)NriFrjrJrp)rlrmrnrJrer2r2r3test_non_commutative_orders rrc Cstttdttgtttfgtttgfks:JtttdttgtttfgtttgfkspJttdtttttdftttfgtttttdgfksJttttttttttttfgttttttgfksJtd\}}}}}}}}||d||d||||||||df}t|t||ft||fg|||tdtt|||tgfksJdS)NrGrQza, b, c, d, f, g, j, mrT)rrDr*rJrZrAr r^) abcdfgjmexprsr2r2r3test_issue_10228s:6"" "$ :>r|cCs<tttdtttttfgtttgfks8JdSrY)rrDr*rJr2r2r2r3 test_powerssr}cCs>tttttttddgttttgfks:JdS)Nbasicr\)rrZrDr*rAr2r2r2r3test_issue_4498srcCsXttdtdtdtdddttdfgttdttdgfksTJdS)NrUrTrVrGr~r\rQ)rrDrJr2r2r2r3test_issue_4020s&&rcCs<tttttttttfgtttgfks8JdSr=)rrrDrJr2r2r2r3test_issue_4203srcCs>ttt dttdd}t|ddgdgfks:JdS)NrQrr~r\T)r rDr)rMr2r2r3test_issue_6263s rcCs0ddlm}td}td}t||ttttfd||ttttfd\}\}|gks^J|||ttttfd||ttttfdksJt||ttttfdttf||ttttfdttf\}\}|tttfgksJ|||ttttfdtf||ttttfdtfks,JdS)Nr)SubsrwrxrrQ)sympy.core.functionrrrrDr*rJ)rrwrxZname_valr?r2r2r3test_dont_cse_tupless*     rcCstdtdtdttdfgtdtgfks8JttdddtdtdttdftdtfgtttdgfksJtdtdddtdtdttdftdtfgttdtgfksJttdtdtdtdttdfgttttgfks*JtttdttdttdfgttttgfkslJttdtdttdtttdfgttdtttgfksJtttdtdtdtdttdfgttdtttgfksJtddtdtdttdfgttdgfksRJttdttdtttdtfgtdtgfksJdS)NrQrG) rrDrJr^rrr*rArr2r2r2r3test_pow_invpow s6"(&("  "& &&  rcCstdttdtdttd}t|tttdtdtdtdgtjdt tdft tdftt ft tdfgt tt t tt tdt t ggksJdS)NrQrG postprocess) rDrr*rrr rAr Z cse_separaterJrbr^rWr2r2r3test_postprocess!s(&&$rcCsddlm}m}td}td}t||tjd||d||dttdd|d|d||tt||dtt|||d||dttttdd|d||tt|d||tt|||d||dttttdd|d||dtt|d||dtt|||d||dttdd|d||tt|d||dtt|||d||dddtjtd| dd||d|f}t |}t d|ft | t ft t dft |dftttft|t tfttddt |||t ftt|t tft||tftt|t tfg ||tjt |t ttttttttttddtjtdt t t fgf}||ksJdS)Nr)rsrtrmGrGrQr)Z sympy.abcrsrtrrr ZHalfrrArrJr^rbx3x4x5x6x7x8x9)rsrtrmrtruansr2r2r3test_issue_4499)sh>8&4  4>0rcCspttddtddd}t|g|gfks4Jttt ttttt ttttkslJdS)NrTrUrGrQ)rrDrrrr*rArr2r2r3test_issue_6169AsrcCsd}td|fd}td|fd}td|d}||d||||d||}d||d||}t||g\}}t|dksJdS)NrUr*)shaperDr1rQr)rrrrK)Zlen_yr*rDr1expr1expr2 replacements reduced_exprsr2r2r3test_cse_IndexedIs(rcCstddd}t|g|gfks"Jtddd}td||}t|g|gfksPJt|d|dg|d|dgfks~Jt|d|d |d|d |d t|d|d fgt|d tgfksJdS) NrlrVr_T)integerrmr)rrr)rrG)r)rr rJ)rlr_rmr2r2r3test_cse_MatrixSymbolUs   .rcCstddd}tddd}|j|j||}|j|||}t||g\}}t|dks^Jt|||g\}}|szJt|d||dg\}}|sJdS)NrlrVr*rQrrG)r)TIrrK)rlr*rrrrr2r2r3test_cse_MatrixExprbs  rcCsvtt ttttdft ttdf}t|}tttfgtttttdft tdfgf}||ksrJdS)NrT)rrArDr*r rrJ)rwrZ actual_ansr2r2r3test_Piecewisers , "rcCshtttddtttdd}t|gttdtttddttdgfksdJdS)NrrVrQrG)rrDZseriesrr*rrrr2r2r3test_ignore_order_termszs&rcsXtt}tt}t||t||ttg}t|\}fdd|D|ksTJdS)Ncsg|]}|tqSr2subsrar0rMrNr2r3r4r5z&test_name_conflict..)rJr*rbrrrZz1Zz2rerOr2rr3test_name_conflicts  rcs^tt}tt}t||t||ttg}t|td\}fdd|D|ksZJdS)Nzx:10csg|]}|tqSr2rrrr2r3r4r5z3test_name_conflict_cust_symbols..)rJr*rbrrrr rr2rr3test_name_conflict_cust_symbolss rcCsnttttttttttt}tttg}ttt||dWdn1s`0YdS)N)r ) rrDr*rZrrAr$ ValueErrorr)resymr2r2r3test_symbols_exhausted_errors,  rc Cstd}td}|d|}i}d|d<d|d<||}t|\}}|D]}|d|||d j<qJ|d |}||ksJtd }t|\}}|d |ksJt|dksJdS) NziPiecewise((C391 - 1.65, C390 < 0.5), (Piecewise((C391 - 1.65, C391 > 2.35), (C392, True)), True))zGPiecewise((2.05*C390**(-1.03), C390 < 0.5), (2.5*C390**(-0.625), True))C391ggv?ZC390gɥ:?ZC392rQraPiecewise((Symbol('ON'), Equality(Symbol('mode'), Symbol('ON'))), (Piecewise((Piecewise((Symbol('OFF'), StrictLessThan(Symbol('x'), Symbol('threshold'))), (Symbol('ON'), true)), Equality(Symbol('mode'), Symbol('AUTO'))), (Symbol('OFF'), true)), true)))r rrnamerK) ZC393rsubZ ss_answerZ substitutionsZnew_eqnpairZ cse_answerr?r2r2r3test_issue_7840s,     rcCsttttfD]t}|ddttdddg}ttt|g}tttfgt|tdgddgggf}||ksjJt|dd|s Jq dS)NrGrrQr]) r%r&r'r(rDr*rrJrR)clsrzresrr2r2r3test_issue_8891s& rc std\}}}}}}||||||||d||||d|g}t|\}}tdd|DrpJddlmddlmtdfd d td D} t| d D]l}t |}t|\}}td d|DrJt |D](} tt |D]}||j | ||<qq||ksJqfdd td D} t| d D]}t |}t|\}}tdd|DrrJt |D],} tt |D]}||j | ||<qqz|fdd |DksBJqBdS)Nz a b f k l irGcss |]}|jD] }|jVq qdSr=argsZis_Mulr0rsr1r2r2r3 r5z#test_issue_11230..r)choice expand_mulza:mcs(g|] }tfddtdDqS)csg|] }qSr2r2r/rsr2r3r4r5/test_issue_11230...rU)rr:r/rr2r3r4r5z$test_issue_11230..rVcss |]}|jD] }|jVq qdSr=rrr2r2r3rr5cs(g|] }tfddtdDqS)csg|]}ddqS)Nrr2r/rr2r3r4r5rrU)rr:r/rr2r3r4r5css |]}|jD] }|jVq qdSr=)rZis_Addrr2r2r3rr5csg|] }|qSr2r2r/rr2r3r4r5) r ranyZsympy.core.randomrrrr:r"r7rarKr) rsrtrwkrer1pRrnexrir2)rrrr3test_issue_11230s2:       rcCsdd}tdtdtdttd}t|ttdftttfgtdtttttgfksnJ||tdtddttd}t|ttdfgtdtdttgfksJ||dS)NcSs>t|\}}|t|tdd|Dt|ks:JdS)NcSsg|]}|dqS)rQrr/r2r2r3r4r5z3test_issue_11577..check..)rrrKsum)rWrrur2r2r3checks  ztest_issue_11577..checkrUrGrTrQ)rDr*rrJr^)rrWr2r2r3test_issue_11577s$2  $ rcCs(tdtdg}t|g|fks$JdS)NrVrT)rDrrr2r2r3test_hollow_rejectionsrcCsttdtdttdttdtdttdg}t|\}}tdd|DsfJdt|tfd\}}tdd|DrJd td d|DsJd dS) NrVrQrUcss|]\}}t|jvVqdSr=r*Z free_symbolsr0rfrr2r2r3rr5z"test_cse_ignore..z&cse failed to identify any term with yignorecss|]\}}t|jvVqdSr=rrr2r2r3r r5z,Sub-expressions containing y must be ignoredcss&|]\}}|ttddkVqdS)rQrN)rrDrr2r2r3r r5z4cse failed to identify sqrt(x + 1) as sub-expression)rr*rrDrr)r{Zsubst1Zred1Zsubst2Zred2r2r2r3test_cse_ignores @ rcsdttttt tttttt g}t|tfd\}fdd|D}||ks`JdS)Nrcsg|]}|tqSr2rrrr2r3r4r5z/test_cse_ignore_issue_15002..)rZrrDrAr*r)rerOZrlr2rr3test_cse_ignore_issue_15002s rcsd\}tdfddt|D}|d|ddksHJt|\}}t|dkshJdt|D]*\}}|t|||dkspJqpdS)N)rVr6zx:%dcs*g|]"ttfddtDqS)cs g|]}|d|qS)r]r2)r0ry)r1rDr2r3r4r5z4test_cse__performance...)rrr:)r0ZntermsrD)r1r3r4sz)test_cse__performance..rrQz!exprs[0] == -exprs[2], i.e. a CSE)r r:ZsimplifyrrK enumeraterra)Znexprsr{substredr1rMr2rr3test_cse__performances   rcCsfttdtttttdtttg}t|\}}dt|tdd|Dt|ksbJdS)NrGrVrcSsg|]\}}|qSr2r)r0rvr2r2r3r4(r5z$test_issue_12070..)rDr*rArrKrr)r{rrr2r2r3test_issue_12070%s , rcCs8tdtdtd}t|dd}||ks4JdS)NrGrQr)rDr*r)rWZcse_eqr2r2r3test_issue_13000,srcCsNttddtddttddtdd}t|g|gfksJJdS)NrU rGrrQ)rrDrrr2r2r3test_issue_182032s4rcCs>tttttdd}t|tttfgtdgfks:JdS)NFevaluaterG)rrDr*rrJrr2r2r3test_unevaluated_mul7srcs$ddlm}td\}}}}}ttddtttttdtdttdddtdttg}t||d\}sJ|tttfttddftdtdf|tttf|ttftdf|tftdf|tftdf|tfg |||||ffk |fdd|Dks JdS) Nr)cse_release_variablesz_:5rQrGrcsg|]}|qSr2)rr/rr2r3r4Hr5z.test_cse_release_variables..) sympy.simplify.cse_mainrr rDr*rrJr^rbreverse)rZ_0Z_1Z_2Z_3Z_4ZeqsrMr2rr3test_cse_release_variables;s  (&$ rcCsdd}|tgtfksJ|dgdfks0Jtg}tttfD] }|||g||fks@Jq@|t|gt|fks~Jtdi}||g|fksJdS)NcSs t|ddS)NF)r7r)rDr2r2r3Kr5ztest_cse_list..rDrQ)rDr7tuplesetr)Z_cseitrurvr2r2r3 test_cse_listJsrcCs4tddd}t| ||| ||ks0JdS)NrlrG)r)r)rlr2r2r3test_issue_18991Vs rcCs(tddddg}t|g|fks$JdS)NrQrGFr)rr)rzr2r2r3test_unevaluated_Mul[sr) functoolsrr8operatorrZsympy.core.addrZsympy.core.containersrrrZsympy.core.mulrZsympy.core.powerrZsympy.core.relationalr Zsympy.core.singletonr Zsympy.core.symbolr r Zsympy.core.sympifyr Z&sympy.functions.elementary.exponentialrZ(sympy.functions.elementary.miscellaneousrZ$sympy.functions.elementary.piecewiserZ(sympy.functions.elementary.trigonometricrrZsympy.matrices.denserZsympy.polys.rootoftoolsrZsympy.series.orderrrrZsympy.simplify.simplifyrZsympy.tensor.indexedrrrZsympy.simplify.cse_optsrrZsympy.functions.special.hyperrZsympy.simplifyr r!Zsympy.utilities.iterablesr"Zsympy.testing.pytestr#r$Zsympy.matricesr%r&r'r(Zsympy.matrices.expressionsr)rZrDr*rArJr^rbrrrrrrrZx10Zx11Zx12r<r@rBrErFrPrSrXr[r`rgrorqrrr|r}rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr2r2r2r3s                      "           % %