ó \ K]c@ sddlmZmZddlZddlZddlmZmZddlZddl Z ddl m Z m Z ddl Z ddlZddlmZddlmZmZddlmZdd lmZmZdd lmZmZmZe jƒd kZyddl Z!e"Z#Wne$k r3e%Z#nXej&e#d ƒZ'd „Z(d„Z)d„Z*defd„ƒYZ+d„Z,d„Z-d„Z.d„Z/d„Z0d„Z1dd„Z2d„Z3dd„Z4d„Z5dd„Z6d„Z7d„Z8e9d „Z:e9d!„Z;e9d"„Z<d#„Z=d$d%„Z>d&„Z?edd'fkrXe9d(„Z@n d)„Z@d*„ZAd+efd,„ƒYZBd-efd.„ƒYZCd/eBfd0„ƒYZDd1eBfd2„ƒYZEd3eBfd4„ƒYZFd5eBfd6„ƒYZGd7eBfd8„ƒYZHd9eBfd:„ƒYZId;eIfd<„ƒYZJd=eIfd>„ƒYZKd?eBfd@„ƒYZLdAeBfdB„ƒYZMdCeIfdD„ƒYZNdEeBfdF„ƒYZOdGeIfdH„ƒYZPdIeBfdJ„ƒYZQdKeBfdL„ƒYZRdMeIfdN„ƒYZSeTdOkrejUƒndS(Piÿÿÿÿ(tdivisiontprint_functionN(tproducttcycle(tNumbertIntegral(tunittest_support(tjitterrors(tversioni(tTestCasettag(tmatmul_usecaset needs_matmult needs_blastarmv7lsLAPACK needs Scipy 0.16+cC stj||ƒS(N(tnptdot(tatb((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytdot2scC stj||d|ƒS(Ntout(RR(RRR((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytdot3#scC stj||ƒS(N(Rtvdot(RR((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyR'st TestProductcB s%eZdZejejejejfZd„Z d„Z d„Z e j d„ƒZd„Zd„Zd„Zed„Zd d „Zed „ƒZd „Zd „Zed„ƒZd„Zed„ƒZedƒd„ƒZed„ƒZed„ƒZ ed„ƒZ!ed„ƒZ"RS(s! Tests for dot products. cC stjƒdS(N(tgctcollect(tself((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytsetUp2scC sOtj|ƒ}t|tjƒr6|ddj|ƒS|ddj|ƒSdS(Niyà?y@gà?yð?à¿(Rtaranget issubclasstcomplexfloatingtastype(Rtntdtypetbase((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt sample_vector6scC s#|j|||ƒj||fƒS(N(R$treshape(RtmR!R"((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt sample_matrix?scc sÂtjdtƒ}tjdtjƒdVWdQX|jt|ƒdƒ|j|dj tjƒ|j dt |dj ƒƒ|j |dj|jjƒ|j |dj|jjdƒdS(sB Check performance warning(s) for non-contiguity. trecordtalwaysNiisfaster on contiguous arrays(twarningstcatch_warningstTruet simplefilterRtNumbaPerformanceWarningtassertGreaterEqualtlentassertIstcategorytassertIntstrtmessaget assertEqualtfilenamet__code__t co_filenametlinenotco_firstlineno(Rtpyfunctw((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytcheck_contiguity_warningBs c C sK|jƒ9||Œ}||Œ}|j||dtƒ~~WdQXdS(Ntignore_sign_on_zero(tassertNoNRTLeaktassertPreciseEqualR,(RR<tcfunctargstexpectedtgot((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt check_funcQs    cC sý|jd|jd}|j|j}tj|dtjƒ}xWt|jdƒD]6}||||!jd|jƒ}|jj rTPqTqTWt dƒ‚|jj rÄtj ||j ddƒ}ntj ||j ddƒ}||(|jj sùt‚|S(NiR"sCould not obtain aligned arraytordertCtF(tsizetitemsizeRtemptytuint8trangetviewR"tflagstalignedt Exceptiont c_contiguousR%tshapetAssertionError(RtarrRJtdatasizettmptitnew((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt _aligned_copyYs    c C s›tr|jntj}|jƒq||ƒ}||ƒ}|j|d||Œ|ƒ|j|d||Œ|ƒ|j||dtƒ~~WdQXdS(NRR?(t _is_armv7lR[RtcopyR@R1RAR,(RR<RBRCRtcopierRDRE((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytcheck_func_outns   cC sQ|jtƒ}||ŒWdQX|r.dnd}|j|t|jƒƒdS(Nsincompatible output array sizesincompatible array sizes(t assertRaisest ValueErrorR3R4t exception(RRBRCtis_outtraisestmsg((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytassert_mismatching_sizesxs  snp.dot()cC sI|jtjƒ}||ŒWdQX|jd|ft|jƒƒdS(Ns)%s arguments must all have the same dtype(R`Rt TypingErrorR3R4Rb(RRBRCt func_nameRd((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytassert_mismatching_dtypess  c C s-d}tdtƒ|ƒ}x|jD]v}|j||ƒ}|j||ƒ}|j||||fƒ|j|||ddd…|ddd…fƒq%W|j|dtjƒ}|j|tjƒ}|j|||fƒ|j|tjƒ}|j|tjƒ}|j |||fd|ƒdS(NitnopythoniÿÿÿÿiRh( RR,tdtypesR$RFRtfloat64Rftfloat32Ri(RR<RhR!RBR"RR((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt check_dot_vv†s7cC s|jtdƒdS(s/ Test vector * vector np.dot() snp.dot()N(RnR(R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt test_dot_vvšscC s|jtdƒdS(s Test np.vdot() s np.vdot()N(RnR(R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt test_vdot sc  s"d\}}‡fd†}tdtƒ|ƒ}|dk rTtdtƒ|ƒ}nxR|||ƒD]A\} } ˆj||| | fƒˆj||| | jfƒqdW|dk r+xs|||ƒD]_\} } tj|d| jƒ} ˆj||| | f| ƒˆj||| | jf| ƒqÅWnˆj ||dtj ƒ} ˆj |tj ƒ} ˆj || | fƒˆj || | jfƒ|dk rktj|tj ƒ} ˆj || | | fƒˆj || | j| fƒˆj ||tj ƒ} ˆj |tj ƒ} tj|dtj ƒ} ˆj || | | fdtƒˆj || | j| fdtƒnˆj ||tj ƒ} ˆj |tj ƒ} ˆj|| | f|ƒ|dk rˆj ||tj ƒ} ˆj |tj ƒ} tj|tj ƒ} ˆj|| | | f|ƒndS( Niic3 sÄxRdD]J}ˆj||tjƒjd|ƒ}ˆj|tjƒ}||fVqWxCˆjD]8}ˆj|||ƒ}ˆj||ƒ}||fVq_W|ddd…|ddd…fVdS(NtCFRGiÿÿÿÿ(R'RRlR]R$Rk(R&R!RGRRR"(R(s6lib/python2.7/site-packages/numba/tests/test_linalg.pytsamplesªs $RjR"iRc(ii(RR,tNoneRFtTRRLR"R_R'RlR$RfRmRi( Rtpyfunc2tpyfunc3RhR&R!Rrtcfunc2tcfunc3RRR((Rs6lib/python2.7/site-packages/numba/tests/test_linalg.pyt check_dot_vm¦sD     & % cC s|jttdƒdS(sC Test vector * matrix and matrix * vector np.dot() snp.dot()N(RyRR(R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt test_dot_vmÚsc  s?‡fd†}tdtƒ|ƒ}|dk rHtdtƒ|ƒ}nx'd d d gD]\}}} xX|||| ƒD]D\} } ˆj||| | fƒˆj||| j| jfƒqzW|dk rXx|||| ƒD]†\} } tj||fd| jƒ} ˆj||| | f| ƒtj||fd| jƒ} ˆj||| j| jf| ƒqáWqXqXWd \}}} ˆj || dtj ƒ} ˆj | |tj ƒ} ˆj || | fƒ|dk r|tj||ftj ƒ} ˆj || | | fƒˆj || tj ƒ} ˆj | |tj ƒ} tj||dftj ƒ} ˆj || | | fdtƒnˆj || tj ƒ} ˆj | |tj ƒ} ˆj || | f|ƒ|dk r;ˆj || tj ƒ} ˆj | |tj ƒ} tj||ftj ƒ} ˆj || | | f|ƒndS( Nc3 såxptddƒD]_\}}ˆj||tjƒjd|ƒ}ˆj||tjƒjd|ƒ}||fVqWxFˆjD];}ˆj|||ƒ}ˆj|||ƒ}||fVq}W|ddd…|ddd…fVdS(NRqRGiÿÿÿÿ(RR'RRlR]Rk(R&R!tktorder_atorder_bRRR"(R(s6lib/python2.7/site-packages/numba/tests/test_linalg.pyRrãs$$RjiiiiR"Rc(iii(iii(iii(iii(RR,RsRFRtRRLR"R_R'RlRfRmRi( RRuRvRhRrRwRxR&R!R{RRR((Rs6lib/python2.7/site-packages/numba/tests/test_linalg.pyt check_dot_mmàsF  # - " t importantcC s|jttdƒdS(s/ Test matrix * matrix np.dot() snp.dot()N(R~RR(R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt test_dot_mmscC s|jtdƒdS(s& Test vector @ vector s'@'N(RnR (R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_matmul_vv!scC s|jtddƒdS(s: Test vector @ matrix and matrix @ vector s'@'N(RyR Rs(R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_matmul_vm(scC s|jtddƒdS(s& Test matrix @ matrix s'@'N(R~R Rs(R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_matmul_mm/sc C sjd\}}}tj}|j|||ƒddd…}|j|||ƒddd…}tj||f|ƒ}tdtƒtƒ}|j|jƒ|||ƒWdQXtdtƒt ƒ}|j|jƒ||||ƒWdQX|j ||ƒddd…}|j ||ƒddd…}tdtƒt ƒ}|j|jƒ|||ƒWdQXdS(NiiiiÿÿÿÿRj(iii( RRlR'RLRR,RR>tpy_funcRR$R( RR&R{R!R"RRRRB((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_contiguity_warnings6s  ""(#t__name__t __module__t__doc__RRlRmt complex128t complex64RkRR$R't contextlibtcontextmanagerR>RFR[R_tFalseRfRiRRnRoRpRyRzR~R R€R RR‚RƒR…(((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyR+s,        4 :cC stjj|ƒS(N(Rtlinalgtinv(R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt invert_matrixOscC stjj|ƒS(N(RRŽtcholesky(R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytcholesky_matrixSscC stjj|ƒS(N(RRŽteig(R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt eig_matrixWscC stjj|ƒS(N(RRŽteigvals(R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyteigvals_matrix[scC stjj|ƒS(N(RRŽteigh(R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt eigh_matrix_scC stjj|ƒS(N(RRŽteigvalsh(R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyteigvalsh_matrixcscC stjj||ƒS(N(RRŽtsvd(Rt full_matrices((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt svd_matrixgscC stjj|ƒS(N(RRŽtqr(R((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt qr_matrixkscC stjj|||ƒS(N(RRŽtlstsq(tAtBtrcond((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt lstsq_systemoscC stjj||ƒS(N(RRŽtsolve(R¡R¢((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt solve_systemssgV瞯Ò=1.is+Negative dimensions given for matrix shape.sRank must be greater than zero.sRank must an integer.s"Rank given greater than full rank.s8Condition number was specified for a vector (always 1.).NiÿÿÿÿR"RG(RIRH(R0RaRRmRlRŠR‰Rstrandomtseedtmint isinstanceRtmaxR$R%RŽRžtlinspacetzerostnonzeroteyeRRttarray(RRJR"RGtrankt conditiontd_condR&R!tminmntrvtmaxmntQRXtUt_tVtsvtStidx((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytspecific_sample_matrix¼sX $          !"/ !cG sLtdkrHx9|D].}t|tjƒrd|jkrAtSqqWntS(s returns True if an input argument has a dimension that is zero and Numpy version is < 1.13, else False. This is due to behaviour changes in handling dimension zero arrays: https://github.com/numpy/numpy/issues/10573 ii i(ii (t numpy_versionRÄRtndarrayRTR,R(RRCtx((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytshape_with_0_inputs   cC s?|j|ƒ}||ŒWdQX|j|t|jƒƒdS(N(R`R3R4Rb(RRBRCReterrRd((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt assert_errorscC s&d}|j|||tjjƒdS(Ns.Last 2 dimensions of the array must be square.(RÞRRŽt LinAlgError(RRBRCRe((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytassert_non_square#scC s'd|}|j|||tjƒdS(Ns9np.linalg.%s() only supported on float and complex arrays(RÞRRg(RtnameRBRCRe((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytassert_wrong_dtype's cC s?|r dnd}d||f}|j|||tjƒdS(Ns np.linalgRs$%s.%s() only supported on 2-D arrays(RÞRRg(RRáRBRCt la_prefixtprefixRe((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytassert_wrong_dimensions+scC s&d}|j|||tjjƒdS(Ns$Array must not contain infs or NaNs.(RÞRRŽRß(RRBRCRe((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytassert_no_nan_or_inf0scC sËt|tƒr3xµ|D]}|j||ƒqWn”t|tƒsÇ|jj}|jj}d}|j||B|ƒd}|dkr™|j||ƒqÇ|dkr¸|j||ƒqÇtdƒ‚ndS(s6 This checks that in a computed result from numba (array, possibly tuple of arrays) all the arrays are contiguous in memory and that they are all at least one of "C_CONTIGUOUS" or "F_CONTIGUOUS". The computed result of the contiguousness is then compared against a hardcoded expected result. got: is the computed results from numba expected_contig: is "C" or "F" and is the expected type of contiguousness across all input values (and therefore tests). s6Results are not at least one of all C or F contiguous.s0Computed contiguousness does not match expected.RHRIsUnknown contigN( RÄttupletassert_contig_sanityRRPRSt f_contiguoust assertTrueRa(RREtexpected_contigRtc_contigtf_contigRe((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyRè4s     cC s)d}|j|||dtjjƒdS(Ns(Matrix is singular to machine precision.RÝ(RÞRRŽRß(RRBRCRe((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytassert_raise_on_singularYscC s¥|j|jd|jdƒtj|jdd|jƒ}dtj|jƒj}|dkrod|}n|dkrˆd|}ntjj ||||ƒdS(sE Checks if a matrix is equal to the identity matrix. iÿÿÿÿiþÿÿÿR"ii idN( R6RTRRÉR"tfinfot resolutionRsttestingtassert_allclose(RREtrtoltatolRÉRð((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytassert_is_identity_matrix]s    cC s d}|j|||tƒdS(s5 For use in norm() and cond() tests. s Invalid norm order for matrices.N(RÞRa(RRBRCRe((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytassert_invalid_norm_kindmscC s&d}|j|||tjjƒdS(NsArrays cannot be empty(RÞRRŽRß(RRBRCRe((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytassert_raise_on_emptytsN(R†R‡RˆRRlRmR‰RŠRkRR$RsRØRÜRaRÞRàRâR,RåRæRèRîRõRöR÷(((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyRÀ§s   S       %  tTestTestLinalgBasecB seZdZd„ZRS(s† The sample matrix code TestLinalgBase.specific_sample_matrix() is a bit involved, this class tests it works as intended. c  sÆtdƒ‰d#d$d%d&d'g}x¿t|ˆjdƒD]¨\}}}|\}}t||ƒ}ˆj|||ƒ}ˆj|j|ƒˆjtjj |ƒ|ƒ|dkr|d} ˆj|||d| ƒ}ˆj|j|ƒˆjtjj |ƒ| ƒndtj |ƒj } ˆj|||ƒ}ˆj|j|ƒtj j tjj|ƒd d | d | ƒ|dkr7d } ˆj|||d | ƒ}ˆj|j|ƒtj j tjj|ƒd d | d | ƒq7q7Wt‡‡fd†} ˆjtƒ,} d}| d(tjdf|dtƒWdQXd}| d)tjdf|dtƒd}| d*tjdf|dtƒd}| d+tjdf|dtƒd}| d,tjdddf|dtƒd}| d-tjdddf|dtƒd}| d.tjdf|dtƒd}| d/tjddf|dtƒd}| d0tjddf|dtƒd}| d1tjddd f|dtƒd!}| d2tjdd"f|dtƒdS(3NRØiii iitFCRËgð?RóRôg$@RÌc sBˆj|ƒ}ˆj|ŒWdQXˆj|t|jƒƒdS(N(R`RØR3R4Rb(RCReRÝRd(tinstR(s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt check_error®stblankiRIRÝssize must be a length 2 tuple.s order must be one of 'F' or 'C'.tzs*dtype must be a numpy floating point type.s/Only one of rank or condition can be specified.sCondition number must be >=1.iÿÿÿÿs+Negative dimensions given for matrix shape.iýÿÿÿsRank must be greater than zero.s"Rank given greater than full rank.is8Condition number was specified for a vector (always 1.).i sRank must an integer.gø?(ii(i i(ii (ii(ii(ii(i(ii(ii(ii(ii(iiýÿÿÿ(ii(ii(ii(ii(RÀRRkRÃRØR6RTRRŽR³RïRðRñRòR°RaR`RURltint32Rs(RtsizesRJR"RGR&R!RÎR¡RËRðRÌRûRdRe((RúRs6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_specific_sample_matrixsj %    %%%""%(R†R‡RˆR(((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyRøyst TestLinalgInvcB s)eZdZedƒed„ƒƒZRS(s" Tests for np.linalg.inv. Rc s d}tdtƒtƒ‰‡‡fd†}xEtˆjdƒD]1\}}ˆj||f||ƒ}||ƒq@W|tjd ƒƒˆjˆtj d ƒfƒˆj dˆtj d d tj ƒfƒˆj dˆtj dƒfƒˆj ˆtjdƒfƒd S(s$ Test np.linalg.inv i Rjc s®t|ƒ}ˆ|ƒ}ˆj|dƒt}ytjj||ddƒWntk rdt}nX|rtj||ƒ}ˆj |ƒnˆj ƒˆ|ƒWdQXdS(NRItnulpi ( RRèRRRñtassert_array_almost_equal_nulpRUR,RRõR@(RtkwargsRDREtuse_reconstructiontrec(RBR(s6lib/python2.7/site-packages/numba/tests/test_linalg.pytcheckñs      RqiiiRR"N(ii(ii(ii(ii(RR,RRRkRØRRLRàtonesRâRþRåRîRÇ(RR!RR"RGR((RBRs6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_linalg_invès (R†R‡RˆR t needs_lapackR (((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyRãs tTestLinalgCholeskycB s/eZdZd„Zd„Zed„ƒZRS(s' Tests for np.linalg.cholesky. c C sœtjjdƒtjj||ƒ}tjj|ƒ\}}tjd|dƒ}tjtj|jtj |ƒƒ|ƒ}tj |d|d|ƒ}|S(NiiR"RG( RRÁRÂtrandRŽRžRRRttdiagRÊ( RR&R"RGR¡tqRÓtLRÑ((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyR'"s*cC s&d}|j|||tjjƒdS(Ns Matrix is not positive definite.(RÞRRŽRß(RRBRCRe((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt assert_not_pd.sc s2d}tdtƒtƒ‰‡‡fd†}x?tˆjdƒD]+\}}ˆj|||ƒ}||ƒq@W|tjd ƒƒd}ˆjˆtj d d tj ƒfƒˆj |ˆtj dd tj ƒfƒˆj |ˆtj dd tj ƒfƒˆjˆtj d d tj ƒjddƒfƒd S(s) Test np.linalg.cholesky i Rjc sˆj|ƒrˆ|ƒdSt|ƒ}ˆ|ƒ}t}ˆj|dƒytjj||ddƒWntk rt}nX|rátj |tj |j ƒƒ}dtj |j ƒj}tjj||d|d|ƒnˆjƒˆ|ƒWdQXdS(NRHRi iRóRô(RÜR’RRèRRñRRUR,RtconjRtRïR"RðRòR@(RRDRERRRð(RBR(s6lib/python2.7/site-packages/numba/tests/test_linalg.pyR:s,         RùiR‘iiR"iN(ii(ii(ii(RR,R’RRkR'RRLRàRRlRâRþRåRR%(RR!RR"RGRtrn((RBRs6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_linalg_cholesky2s#%   (R†R‡RˆR'RR R(((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyR s tTestLinalgEigenSystemscB sheZdZd„Zd„Zdd„Zed„ƒZed„ƒZ ed„ƒZ ed„ƒZ RS( s* Tests for np.linalg.eig/eigvals. cC s´|j||ƒ}tj|ƒ}tjtj|jd|jddƒƒ}|d||R?(((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyR<ùs  tTestLinalgLstsqcB seZdZed„ƒZRS(s$ Tests for np.linalg.lstsq. c sktdtƒtƒ‰‡‡fd†‰dddddg}tˆjƒ}dd g}t|ƒ‰d }‡‡‡fd †}xò|D]ê}t|ƒ}tˆƒ}ˆj|||ƒ} || |ƒ|\} } t| | ƒ} | dkr…| dkr…| d} ˆj|||d | ƒ} || |ƒˆj|||d |ƒ} d|}| |}|| |d|ƒq…q…Wd d!gd"d#gd$d%gd&d'gg}xB|D]:\} }tj | ƒtj |ƒf}ˆj ˆ|ƒq¤Wtj ddgddggdtj ƒ}ˆ||ffd}tj ddgddggdtj ƒ}ˆj|ˆ||fƒˆj|ˆ||fƒtj ddgddggdtjƒ}ˆj|ˆ||fƒˆj|ˆ||fƒtj ddgdtj ƒ}ˆj|ˆ||fƒtj ddgtjtjggdtj ƒ}ˆjˆ||fƒˆjˆ||fƒtj ddgdtj ƒ}ˆ||fftj ddgddggddgddgggdtj ƒ}ˆj|ˆ||fƒtj dgdtj ƒ}tj dgdgdggdtj ƒ}ˆjˆ||fƒˆjˆ||fƒdS((s& Test np.linalg.lstsq Rjc  s-t|||}ˆ|||}ˆjt|ƒt|ƒƒˆjt|ƒdƒˆj|dƒt}yxˆj|d|dƒxYtt|ƒƒD]E}y%tjj||||ddƒWqšt k rÞt }qšXqšWWnt k rýt }nX|r|\}}} } dddg} y`ˆj| |dƒxE| D]=}t ||tj ƒrCˆj||j ||j ƒqCqCWWn2t k r¹ˆjt| |dƒdkƒnXtj|jƒj} y=tj||ƒ} tjj|| d d| d d| ƒWqt k rxä| D]Ø}y3tjj||||d d | d d | ƒWn;t k r™tjj|ƒ}ˆjd|d | ƒnXtjj|tj||dƒƒ}tjj|tj||ƒƒ}tjj||d d ƒq#WqXnˆjƒˆ|||WdQXdS(NiRHiRi iiiRóRôidgð?g$@(R¤R6R0RèRRNRRñRRUR,RÄRÚRTRêtabsRïR"RðRRòRŽR°t assertGreaterR­R@(R¡R¢RRDRERR{RÛtresRËR4t out_array_idxRðRtct res_expectedtres_got(RBR(s6lib/python2.7/site-packages/numba/tests/test_linalg.pyR%sh     ) %         "  iii iiRIRHg$@c s—d}xŠ|D]‚}tˆƒ}ˆj|jd|f||ƒ}ˆ|||tˆƒ}|dd…dfjd|ƒ}ˆ|||q WdS(Nii iRG(ii (tnextRØRTR](R¡tdtRtb_sizestb_sizetb_orderR¢RX(Rt cycle_orderR(s6lib/python2.7/site-packages/numba/tests/test_linalg.pytinner_test_loop_fn”s   "RËRÌgð?R£ig@g@g@R"R iiiiN(ii(i i(ii (ii(ii(ii(i(ii(i(ii(i(ii(ii(RR,R¤RRkRHRØRÃRRLR÷RÊRlRþRâRmR?RåR%R&RæR=R>(RRÿtcycle_dttorderst specific_condRNta_sizeRIta_orderR¡R&R!RÎR:R£tapprox_half_rank_rcondtemptiesRRCtokRtbadtoneDtbad1Dtbad2D((RBRRMRs6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_linalg_lstsqsn\               ***0B*(R†R‡RˆR R[(((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyR@s tTestLinalgSolvecB seZdZed„ƒZRS(s$ Tests for np.linalg.solve. c sÎtdtƒtƒ‰‡‡fd†}dddg}x¿t|ˆjdƒD]¨\}}}ˆj|||ƒ}d}x{t|dƒD]j\}} ˆj|jd|f|| ƒ} ||| ƒ| d d …dfjd | ƒ} ||| ƒq†WqLWˆtj dƒtj dƒƒtj d d gd d ggd tj ƒ} ˆ| | ƒd} tj ddgddggd tj ƒ}ˆj | ˆ| |fƒˆj | ˆ|| fƒtj ddgddggd tjƒ}ˆj| ˆ| |fƒˆj| ˆ|| fƒtj ddgd tj ƒ}ˆj| ˆ|| fƒtj d d gtjtjggd tj ƒ}ˆjˆ| |fƒˆjˆ|| fƒtj d dgd tj ƒ}ˆ| |ƒtj ddgddggddgddgggd tj ƒ}ˆj| ˆ| |fƒtj d gd tj ƒ}tj d gdgdggd tj ƒ}ˆjˆ| |fƒˆjˆ| |fƒˆjdtj dddƒ}ˆjˆ|| fƒd S(s& Test np.linalg.solve Rjc st|||}ˆ|||}ˆj|dƒt}ytjj||ddƒWntk rpt}nX|rÞˆj|j |j ƒtj ||ƒ}tj |j ƒj }tjj||dd|dd|ƒnˆjƒˆ|||WdQXdS(NRIRi RóRôid(R¦RèRRRñRRUR,R6RTRRïR"RðRòR@(RRRRDRERRRð(RBR(s6lib/python2.7/site-packages/numba/tests/test_linalg.pyRs(      iiiRùi iNRGgð?gR"R¥iig@iiig@RHRË(ii(ii(ii(ii (ii(i(ii(RR,R¦RRkRØRTR]RRLRÊRlRþRâRmR?RåR%R&RæR=R>Rî(RRRÿRJR"RGR¡RJRKRLR¢RXRVRRWtok_oneDRYRZ((RBRs6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_linalg_solvesL+" "* **0 B*(R†R‡RˆR R^(((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyR\stTestLinalgPinvcB seZdZed„ƒZRS(s# Tests for np.linalg.pinv. c sÀtdtƒtƒ‰‡‡fd†}dddddg}d}xºt|ˆjd ƒD]£\}}}ˆj|||ƒ}||ƒ|\}} |dkrX| dkrXt|| ƒ} ˆj|||d |ƒ}d |} | | } ||d | ƒqXqXWx'ddgD]} |tj| ƒƒq Wd}ˆj |ˆtj ddtj ƒfƒˆj |ˆtj ddtj ƒfƒˆjˆtjd dgtjtjggdtj ƒfƒdS(s% Test np.linalg.pinv Rjc  süˆj|ƒr ˆ||dSt||}ˆ||}ˆj|dƒt}ytjj||ddƒWntk rŠt}nX|r؈j |j |j ƒtj ||ƒ}yˆj |ƒWqØtk rÔdtj |jƒj}ˆ|ƒ}tjj||dd|dd|ƒ|j d|j d kr±td tƒtƒ}||tj|j dƒj|jƒ|d}tjj||dd|dd|ƒnˆjtjj||ƒ|ƒqØXnˆjƒˆ||WdQXdS( NRIRi iRóRôidiiRj(RÜR¨RèRRRñRRUR,R6RTRRõRïR"RðRòRR¤RÉR t assertLessRŽR­R@( RRRDRERRRðR t lstsq_pinv(RBR(s6lib/python2.7/site-packages/numba/tests/test_linalg.pyR‰sN            ' iii iig$@RùRÌgð?R£iR§iR"i g@N(ii(i i(ii (ii(ii(ii(ii(ii(RR,R¨RRkRØRÃRRLRâRRþRåRlRæRÊR%R&(RRRÿRQRJR"RGRR&R!RÎR£RTR6R((RBRs6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_linalg_pinv‚s4I"        $(R†R‡RˆR Rb(((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyR_}stTestLinalgDetAndSlogdetcB sGeZdZd„Zd„Zd„Zed„ƒZed„ƒZRS(sÄ Tests for np.linalg.det. and np.linalg.slogdet. Exactly the same inputs are used for both tests as det() is a trivial function of slogdet(), the tests are therefore combined. cK s”|j|ƒr |||dSt||}|||}dtj|jƒj}tjj||d|ƒ|jƒ|||WdQXdS(NiRó( RÜR¬RRïR"RðRñRòR@(RRBRRRDRERð((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt check_dets  c K sH|j|ƒr |||dSt||}|||}|jt|ƒt|ƒƒ|jt|ƒdƒx>tdƒD]0}|jtj||ƒtj||ƒƒq}W|jj|dƒ}tj j ||dddƒdtj |jƒj }tj j |d|dd|d|ƒ|jƒ|||WdQXdS( NiiRi iiRóRô(RÜRªR6R0RNRRR"ttypeRñRRïRðRòR@( RRBRRRDRER{tgot_convRð((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyt check_slogdets&    c C sFd d dg}xHt||jdƒD]1\}}}|j|||ƒ}|||ƒq%WxBt|jdƒD].\}}tjdd|ƒ}|||ƒqmW||tjdƒƒ|j||tjddtjƒfƒ|j ||tjd dtj ƒfƒ|j |tj d d gtj tjggdtj ƒfƒdS(NiiiRùiR"iii gð?g@(ii(ii(ii(ii(ii(ii(RRkRØRRÇRLRâRRþRåRlRæRÊR%R&( RRRRBRÿRJR"RGR((s6lib/python2.7/site-packages/numba/tests/test_linalg.pytdo_test>s "   $cC s/tdtƒtƒ}|jd|j|ƒdS(NRjR«(RR,R¬RhRd(RRB((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_linalg_det_scC s/tdtƒtƒ}|jd|j|ƒdS(NRjR©(RR,RªRhRg(RRB((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_linalg_slogdetds( R†R‡RˆRdRgRhR RiRj(((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyRcüs   ( !tTestLinalgNormcB seZdZed„ƒZRS(s# Tests for np.linalg.norm. c  s¤tdtƒtƒ‰‡‡fd†}dddg}dtjtj ddddd d d d g }xHt|ˆj|ƒD]1\}}}ˆj||ƒ}||d |ƒqzWxOtˆj|ƒD];\}}ˆjd|ƒddd…}||d |ƒqÂWdd d!d"d#g}dtjtj dddd g}xQt|ˆjd|ƒD]7\}}}}ˆj |||ƒ}||d |ƒqQWdg}x˜tˆj|dƒD]\}}}ˆj d$||ƒ}||d d |ƒ||dd…dd…fd |ƒ||dddd…fd |ƒq«WxŽtˆj|dƒD]w\}}}tj d%d|d|ƒ}ˆj ˆ||ƒdƒtj d&d|d|ƒ}ˆj ˆ||ƒdƒqFWd} ˆj | ˆtj d'dtjƒfƒˆj| ˆtj ddtjƒjdddƒfƒˆjˆtjddgtjtjggdtjƒdfƒˆjˆtjddgddggdtjƒdfƒdS((s% Test np.linalg.norm Rjc s‹t||}ˆ||}ˆjtj|ƒ ƒdtj|jƒj}tjj||d|ƒˆj ƒˆ||WdQXdS(NiRó( R¯RêRRRïR"RðRñRòR@(RRRDRERð(RBR(s6lib/python2.7/site-packages/numba/tests/test_linalg.pyRys iiiiiÿÿÿÿiiþÿÿÿigÍÌÌÌÌÌ@g333333ÀR®i Nii Rùii R"RGgR­i gð?g@g@g@i(ii(i i(ii (ii(ii(ii (i(ii(ii(RR,R¯RsRR%RRkR$RØRLR6RâRRþR=RlR%RæRÊR&Rö( RRRÿt nrm_typesRJR"tnrm_typeRRGR((RBRs6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_linalg_normrsV.""( "&'"   $'(R†R‡RˆR Rn(((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyRkmstTestLinalgCondcB seZdZed„ƒZRS(s# Tests for np.linalg.cond. c  s#tdtƒtƒ‰‡‡fd†}dtjtj ddddg}ddg}xQt|ˆjd |ƒD]7\}}}}ˆj|||ƒ}||d |ƒqnWd d!d"d#g}xEt|ˆjd ƒD].\}}}ˆj|||ƒ}||ƒqÑWx3d$d%d&gD]"} ˆj ˆtj | ƒfƒqWt d'krÆtj dd gd d ggdtj ƒ} || ƒ|| d dƒtj d d gd d ggdtj ƒ} || d dƒntjƒ~tj dd gd dggdtj ƒ}t d(kr2tjdtƒˆjtd||ƒntjdtƒ||ƒWdQXd} ˆj| ˆtjd)dtjƒfƒˆj| ˆtjddtj ƒfƒˆjˆtj ddgtjtjggdtj ƒfƒˆjˆtj ddgddggdtj ƒdfƒdS(*s% Test np.linalg.cond Rjc s‹t||}ˆ||}ˆjtj|ƒ ƒdtj|jƒj}tjj||d|ƒˆj ƒˆ||WdQXdS(NiRó( R²RêRRRïR"RðRñRòR@(RRRDRERð(RBR(s6lib/python2.7/site-packages/numba/tests/test_linalg.pyRãs iiÿÿÿÿiiþÿÿÿiiRùR±i iiiR"g Èë…óÌágš™™™™™¹?iterrorsoverflow encountered in.*tignoreNR°i gð?g@g@g@i(ii(ii(ii(i i(ii (ii(ii(ii(ii(ii(ii(ii(RR,R²RsRR%RRkRØR÷RLRÙRÊRlR*R+R-tRuntimeWarningtassertRaisesRegexpRâRRþRåRæR&Rö( RRtpsRÿRJR"RGR±RR6RÛR((RBRs6lib/python2.7/site-packages/numba/tests/test_linalg.pyttest_linalg_condÛsP" ("  * * *      $'(R†R‡RˆR Ru(((s6lib/python2.7/site-packages/numba/tests/test_linalg.pyRoÖstTestLinalgMatrixRankcB seZdZed„ƒZRS(s* Tests for np.linalg.matrix_rank. c sãtdtƒtƒ‰‡‡fd†}dddddg}x;t|ˆjdƒD]$\}}}ˆj|||ƒ}||ƒd }xbtdt|ƒdƒD]G}ˆj|||d |ƒ}ˆjˆ|ƒ|ƒ||ƒ|j \} } d |d d …d d …fsŠ               ÿ%                     Òj:XÁd_ò{qi]kBTd