a kh3(@sddlmZddlmZedZer.srrz1.3g@)r)rrrpowrarrayr )rrrZ sympy_objsZ numpy_objsrrrtest_systematic_basic$s&    r"cCstd}td}tdt|ks$Jt|gt|gksc Cs*tgdtttgdtdttgg}tt2tgdtddgtttdgg}Wdn1sf0Y||tgddttdtdgdtttdtttdggksJ||tgddttdtdgdtttdtttdggksJ||||ks&JdS) Nrrrri)rrr4r)rr2)r rrrrr9r:add)Mr7rrrtest_Matrix_sums & @NPrEc Cstgdtttgg}tt.tddgtdgttdgg}Wdn1sR0Y||tddtddtdgdttttddtdtttdggksJ||tddtddtddtgdtdtdtd tgtttddtttddtttdggksBJtdg}|d |d|ksfJ||d d|ksJdS) Nr?rr4r@rrrA r)r rrrr9r:rr!)rDr7r(rrrtest_Matrix_muls <2  4 rHcCs>Gddd}|}t|tgdgdgdgks:JdS)Nc@seZdZedfddZdS)z#test_Matrix_array..matarrayNcSs:|dur|stdddlm}|gdgdgdgS)Nz=Cannot implement copy=False when converting Matrix to ndarrayr)r!r?r4rr@rArrG) TypeErrorrr!)selfdtypecopyr!rrr __array__s  z-test_Matrix_array..matarray.__array__)__name__ __module__ __qualname__objectrOrrrrmatarraysrTr?rIrJ)r )rTZmatarrrrrtest_Matrix_arraysrUcCsttdtdgdttdgg}t|ts2J|jdks@J|ddksPJ|dtdksdJ|ddttks|J|d dksJdS) Nrrrr)rrrr)rr)rr)rr)r r rr r+r,shaper.rrrtest_matrix2numpys$rXcCstddttgtdttddgg}tddttgtdttddgg}t||ks`Jt|jtdksxJttddgddggdd}ttddgddggdd}|jtdksJ|jtdksJdS) NrrrS Zint8rMfloat64) r r rrr!r r#rMr)r(r0cdrrrtest_matrix2numpy_conversions&&r_cCstddtdtdgttdgks.JtddtdtdgtdttddtddgkspJtdtdtdgttdtdgksJtdtdtdgtdttdtdgksJdS)Nrrrz0.5z1.0)rr!rr#rrrrrtest_issue_3728s.   4 r`cCsdtj_td}ttttd}d}| |d|krD|ksJnJttj tdkrt t |tWdq1s0Yn0t t |tWdn1s0YdS)NrFz0.198669330795061215459412627rgV瞯ddlm}|D]$\}}|tjvs*J|tjvsJqdS)Nr)NUMPY_TRANSLATIONS)sympy.utilities.lambdifyrsitemsrk__dict__r)rssymmatrrrtest_lambdify_transl$s rycCsDddl}ddlm}td}tdd}tdd}|||j|td|d|dksZJtdd}tdd}|d|dkrJtd }|||j|tdtdd }|jd ksJtd \} } |d | ksJ|d | ksJtdd} | jdksJtd\} } }| d| ksJ| d| ks.J| d|ks@JdS)z/Test creation of numpy arrays of SymPy symbols.rNz_0,_1,_2rr[r(r0z a_0,a_1,a_2)rrz a_0_0,a_1_2rV)rr)rrrza_0_0_0,a_1_2_0,a_1_2_1)rrr)rrr)rrr) rZ numpy.testingZtestingr rZassert_array_equalr!rSrW)npZnptZsymss1s2r(r0ZasymsZa2dZa00Za12Za3dZa000Za120Za121rrr test_symarray+s.        r~cCs:ttgdttdtdtdgks6JdS)Nr?rrr)rZ vectorizer r!r#rrrrtest_vectorizeMsr)rErHrUrXr_r`rgrirqrrryr~rrrrrsP       %          "