o *ήc&@s<ddlmZddlmZddlmZmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZmZmZmZmZmZmZdd lmZdd lmZdd lm Z m!Z!ddl"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)ddl*m+Z+ddl,m-Z-m.Z.m/Z/m0Z0ddl1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7ddl8m9Z9ddl:m;Z;mZ>e>dZ?e?re?@e?AgjBZCeCjDZEddZFddZGddZHddZIddZJd d!ZKd"d#ZLd$d%ZMd&d'ZNd(d)ZOd*d+ZPd,d-ZQd.d/ZRd0d1ZSd2d3ZTd4d5ZUd6d7ZVd8d9ZWd:d;ZXdd?ZZd@dAZ[dBdCZ\dDdEZ]dFdGZ^dHS)I)Sum)Mod)Equality Unequality)sqrt) Piecewise) BlockMatrix) MatrixSymbol)Identity)lambdify)xijabcd)Pow) MatrixSolve) logaddexp logaddexp2)log1pexpm1hypotlog10exp2log2Sqrt)Array)ArrayTensorProductArrayAdd PermuteDims ArrayDiagonal) NumPyPrinter SciPyPrinter_numpy_known_constants_numpy_known_functions_scipy_known_constants_scipy_known_functions)convert_matrix_to_array)skipraises) import_modulenumpycCsDt}tdtdkfd}||dksJ|jdhdiks JdS)z NumPyPrinter needs to print Piecewise()'s choicelist as a list to avoid breaking compatibility with numpy 1.8. This is not necessary in numpy 1.9+. See gh-9747 and gh-9749 for details. r)rTz?numpy.select([numpy.less(x, 0),True], [1,0], default=numpy.nan)r->nanlessselectN)r#rr doprintmodule_imports)printerpr6F/tmp/pip-target-vg8gfxp4/lib/python/sympy/printing/tests/test_numpy.pytest_numpy_piecewise_regression s r8cCs@ttt}t|dksJttt}t|dksJdS)Nznumpy.logaddexp(a, b)znumpy.logaddexp2(a, b))rrrr#r2r)laelae2r6r6r7test_numpy_logaddexp,s  r;c ststdttttttf}ttttf|d}d\}}tdddt |||t fddt ||dDs>Jttttttf}ttttf|d}d\}}tdddt |||t fd dt ||dDsvJdS) NNumPy not installedr-r r.r>c3s|]}|VqdSNr6.0i_x_r6r7 <ztest_sum..c3s|]}|VqdSr@r6rArDr6r7rFCrG) npr*rr r rrr linspaceallclosesumrangesfa_b_r6rDr7test_sum3s48rRc ststdtttttttfttt f}t tttt tf|d}d\}}d\t dddt |||t fdd t||dDsNJdS) Nr<r-r=) r?r.r>c3s0|]}tdD] }||Vq qdS)r.N)rL)rBrCj_c_d_rEr6r7rFQs.z%test_multiple_sums..)rHr*rr rr rrrrr rIrJrKrLrMr6rVr7test_multiple_sumsFs " rYcCststdtddd}tddd}t||}t||f|d}tddgddgg}tdd gd dgg}|||t||ksEJdS) Nr<MNr-r.r?)rHr*r r)r arraymatmulall)rZr\cgrOmambr6r6r7test_codegen_einsumTs   "rfc Csxtstdtddd}tddd}tddd}tddd}tddgdd gg}tdd gd dgg}tdd gddgg}tdd gd d gg}t||}t||f|d} | ||t|d dg|ddgkslJt||}t||f|d} | ||||ksJt|||}t|||f|d} | ||||||ksJt||||}t||||f|d} | ||||||||ksJt |dd g}t|f|d} | ||j ksJt t||gd}t||f|d} | ||t t|d dg|ddgdksJt t||d}t||f|d} | ||tj t|d dg|ddgdddks:JdS)Nr<rZr[r\PQr.r]r^r_r?rr-)r.r[r]rr.r[)axis1axis2)rHr*r r`rr einsumrbr r!T transposer"diagonal) rZr\rgrhrdremcmdrcrOr6r6r7test_codegen_extracs>     *   &4<rscCsptstdttd}ttf|}tgd}t||gds$Jttd}ttf|}tgd}t||gdsBJtdk}ttf|}tgd}t||gds_Jtdk}ttf|}tgd}t||gds|Jtdk}ttf|}tgd}t||gdsJtdk}ttf|}tgd}t||gd sJdS) Nr<r.)rr.r[)FTF)TFT)TFF)TTF)FFT)FTT)rHr*rr r r` array_equalr)erOrEr6r6r7test_relationals4        rvcCststdttt}tttf|}tgd}d}t|||gds(Jtgd}tgd}t|||gdsCJtgd}tgd}t|||gds^JdS)Nr<)rr.r[r]r[)rr.rr.)r[r[r[r[)r[r]r^)rrrr)rHr*rrrr r`rt)rurOrPrQr6r6r7test_mods rxcCs8tstdtdddd}tg|d}|dksJdS)Nr<r[r?F)evaluater-g?)rHr*rr )exprrOr6r6r7test_pows  r{cCsBtstdttfttd}t|ddddtksJdS)Nr<r-g|=g#B ;)rHr*r rrabsNUMPY_DEFAULT_EPSILONrOr6r6r7 test_expm1s$rcCs>tstdttfttd}t|dddtksJdS)Nr<r-g>N}a+)rHr*r rrr|r}r~r6r6r7 test_log1ps rcCs<tstdttttftttddddtksJdS)Nr<r-r]r^rw)rHr*r|r rrrr}r6r6r6r7 test_hypots0rcC6tstdtttfttdddtksJdS)Nr<r-dr[)rHr*r|r rrr}r6r6r6r7 test_log10*rcCr)Nr<r-rw )rHr*r|r rrr}r6r6r6r7 test_exp2rrcCr)Nr<r-)rHr*r|r rrr}r6r6r6r7 test_log2rrcCrNr<r-r^r[)rHr*r|r rrr}r6r6r6r7 test_SqrtrrcCrr)rHr*r|r rrr}r6r6r6r7 test_sqrtrrcCststdtddd}tddd}|d||}t|||}t||f|}t||f|}tgdgdgd g}tj|dksGJtgd }t||||||s\JdS) Nr<rZr]r r.r?)r.r[r])r]r[rw)rwri)r]r^rw) rHr*r rr r`linalg matrix_rankrJ)rZr rz matsolve_exprrO f_matsolvem0x0r6r6r7 test_matsolves   rcCsztstdtddd}tddd}tddd}tddd}t||g||gg}|jdks/Jt}||d ks;JdS) Nr<a_1r>r]a_2a_3a_4)rz%numpy.block([[a_1, a_2], [a_3, a_4]]))rHr*r rshaper#r2)rrrrAr4r6r6r7 test_16857s    rcststdtddd}t||td}tddgddgg}tddgddgg}|||ks5Jdd lm}|d d d tdt t fd ddS)Nr<rZr[r.r]r^rwr)symbolsnT)integercsttSr@)r r r6r\rr6r78sz"test_issue_17006..) rHr*r r r r`rbsympy.core.symbolrr+NotImplementedError)rZrOrdmrrr6rr7test_issue_17006)s    rcCs4ttddks JttddksJdS)N)rj)r]rwznumpy.array([[1, 2], [3, 5]])rjznumpy.array((1, 2)))r#r2rr6r6r6r7test_numpy_array:srcCDtddksJtddksJtddksJtddks JdS) NNaNz numpy.nan EulerGammaznumpy.euler_gammaacosz numpy.arccoslogz numpy.log)r%r&r6r6r6r7test_numpy_known_funcs_consts>rcCr) N GoldenRatiozscipy.constants.golden_ratioPizscipy.constants.pierfzscipy.special.erf factorialzscipy.special.factorial)r'r(r6r6r6r7test_scipy_known_funcs_constsErrcCs&t}t|ds Jt|dsJdS)N _print_acos _print_log)r#hasattrprntrr6r6r7test_numpy_print_methodsLsrcCsPt}t|ds Jt|dsJt|dsJt|dsJt|ds&JdS)Nrr _print_erf_print_factorial_print_chebyshevt)r$rrr6r6r7test_scipy_print_methodsQs rN)_sympy.concrete.summationsrsympy.core.modrsympy.core.relationalrr(sympy.functions.elementary.miscellaneousr$sympy.functions.elementary.piecewiser&sympy.matrices.expressions.blockmatrixr"sympy.matrices.expressions.matexprr "sympy.matrices.expressions.specialr sympy.utilities.lambdifyr sympy.abcr r rrrrr sympy.corersympy.codegen.matrix_nodesrsympy.codegen.numpy_nodesrrsympy.codegen.cfunctionsrrrrrrrsympy.tensor.arrayr0sympy.tensor.array.expressions.array_expressionsrr r!r"sympy.printing.numpyr#r$r%r&r'r(3sympy.tensor.array.expressions.conv_matrix_to_arrayr)sympy.testing.pytestr*r+sympy.externalr,rHfinfor`dtypedeafult_float_infoepsr}r8r;rRrYrfrsrvrxr{rrrrrrrrrrrrrrrrr6r6r6r7sb       $  $     *)