U 9%e'@sddlmZddlmZedZer.srrz1.3g@)rrrpowrarrayr )rrrZ sympy_objsZ numpy_objsrrrtest_systematic_basic$s&    r"cCstd}td}tdt|ks$tt|gt|gksc Cstttddgddgg}W5QRXt|tddgddggksHtt|tddgddggkshtttttddgddgg}W5QRXt|ttddgddggkstt|ttddgddggkstdSr=)rr:r;r r#r r/rrr test_Matrix4s    "$r?c Cstdddgtttgdtdttgg}tt*tdddgtddgtttdgg}W5QRX||tdddgdttdtdgdtttd tttdggkst||tdddgdttdtdgdtttd tttdggkst||||kstdS) Nrrrir5r2) r rrrrr:r;r#add)Mr8rrrtest_Matrix_sums ( .PRrEc Csrtdddgtttgg}tt$tddgtdgttdgg}W5QRX||tddtddtdgdttttddtdtttdggkst||tddtddtddtgdtdtdtd tgtttddtttddtttdggks0ttdg}|d |d|ksTt||d d|ksntdS) Nrrrr5r@rrA r) r rrrr:r;rr#r!)rDr8r)rrrtest_Matrix_muls (2  4 rHcCsDGddd}|}t|tdddgdddgd d d ggks@tdS) Nc@seZdZddZdS)z#test_Matrix_array..matarraycSs,ddlm}|dddgdddgd d d ggS) Nr)r!rrrr5rr@rArrG)rr!)selfr!rrr __array__s z-test_Matrix_array..matarray.__array__N)__name__ __module__ __qualname__rJrrrrmatarraysrNrrrr5rr@rArrG)r r#)rNZmatarrrrrtest_Matrix_arraysrOcCsttdtdgdttdgg}t|ts2t|jdks@t|ddksPt|dtdksdt|ddttks|t|d dkstdS) Nrrrr)rrrr)rr)rr)rr)r r rr r,r-r#shaper/rrrtest_matrix2numpys$rRcCstddttgtdttddgg}tddttgtdttddgg}t||ks`tt|jt dksxtttddgddggdd}ttddgddggdd}|jt dkst|jt dkstdS) Nrrobject Zint8dtypeZfloat64) r r rrr!r r$r#rWr)r)r1cdrrrtest_matrix2numpy_conversions&&rZcCstddtdtdgttdgks.ttddtdtdgtdttddtddgkspttdtdtdgttdtdgksttdtdtdgtdttdtdgkstdS)Nrrrz0.5z1.0)rr!rr$r#rrrrrtest_issue_3728s.   4 r[c Csdtj_td}ttttd}d}| |d|krD|ksJnttt j tdkrzt t |tW5QRXnt t |tW5QRXdS)NrFz0.198669330795061215459412627rgV瞯ddlm}|D]$\}}|tjks*t|tjkstqdS)Nr)NUMPY_TRANSLATIONS)sympy.utilities.lambdifyrqitemsri__dict__r#r)rqsymmatrrrtest_lambdify_transl"s rwcCsDddl}ddlm}td}tdd}tdd}|||j|td|d|dksZttdd}tdd}|d|dkrttd }|||j|tdtdd }|j d ksttd \} } |d | kst|d | ksttdd} | j dksttd\} } }| d| kst| d| ks.t| d|ks@tdS)z/Test creation of numpy arrays of SymPy symbols.rNz_0,_1,_2rrVr)r1z a_0,a_1,a_2)rrz a_0_0,a_1_2rP)rr)rrrza_0_0_0,a_1_2_0,a_1_2_1)rrr)rrr)rrr) rZ numpy.testingtestingr rZassert_array_equalr!rSr#rQ)npZnptZsymss1s2r)r1ZasymsZa2dZa00Za12Za3dZa000Za120Za121rrr test_symarray)s.        r}cCs<ttdddgttdtdtdgks8tdS)Nrrr)rZ vectorizer r!r$r#rrrrtest_vectorizeKsr~)r?rErHrOrRrZr[rcrerorprwr}r~rrrrsP       %          "