&]\c@`s ddlmZmZmZddlZddlZddlZddlmZddl m Z m Z m Z m Z mZmZddlZddlmZddlZddlmZmZmZmZmZmZmZmZddlmZmZddl m!Z"dd l m#Z#m$Z$m%Z%m&Z&dd l'm(Z(ydd l m)Z*Wne+k redZ*nXdd l'm-Z-dd l.m/Z/ej0ej1gZ2ej3ej4gZ5e2e5Z6de7fdYZ8de7fdYZ9de7fdYZ:de7fdYZ;de7fdYZ<de7fdYZ=dZ>dZ?dZ@ejAjBdZCde7fdYZDd e7fd!YZEd"ZFd#ZGd$ZHd%ZId&ZJd'ZKd(ZLd)ZMd*ZNd+ZOd,ZPd-ZQd.ZRdS(/i(tdivisiontprint_functiontabsolute_importN(treduce(t assert_equaltassert_array_almost_equaltassert_tassert_allclosetassert_almost_equaltassert_array_equal(traises(teyetonestzerost zeros_likettriuttrilt tril_indicest triu_indices(trandtseed(t_flapack(tinvtsvdtcholeskytsolve(t_compute_lwork(t_clapack(tget_lapack_funcs(tget_blas_funcstTestFlapackSimplecB`s,eZdZdZdZdZRS(c C`scdddgdddgddd gg}dd d d gdd d d gddd d gd dd d gg}xd D]}tt|dd}|dkrqsn||\}}}}} t| t| t||t||fd t|d dft|tj t|||dddd\}}}}} t| t| qsWdS(Niiiiiiiii iga2U0*3?gMb`?tsdzctgebaltpermutetscale( tgetattrtflapacktNoneRtreprRRtlentnpR ( tselftata1tptftbatlothitpivscaletinfo((s=lib/python2.7/site-packages/scipy/linalg/tests/test_lapack.pyt test_gebal*s *   ''cC`sdddgdddgddd gg}x_d D]W}tt|d d}|dkr_q1n||\}}}t| t|q1WdS( Nikiifiii"iiitdtgehrd(R#R$R%RR&(R)R*R,R-thtttauR2((s=lib/python2.7/site-packages/scipy/linalg/tests/test_lapack.pyt test_gehrd?s    c C`stjddgddgg}tjddgddgg}tjdd gd d gg}d }xd D]w}|j||j||j|}}}td|f\} |jr|dcd7(ii(ii(ii(ii(ii( R(RRR9RDtconjRtcopyRqRRER R( ta0ta0jRIRRR*talphaRLR7texpectedR((s=lib/python2.7/site-packages/scipy/linalg/tests/test_lapack.pyttest_larfg_larfs*&   :   WW#cC`stjtjddgdtjdtj}xPtdD]2}tjd|j dk r:|j }Pq:q:Wd}|j t |dd|jjdS( Ns-csfimport numpy as np; from scipy.linalg import svd; a = np.zeros([9537, 9537], dtype=np.float32); svd(a)tstdouttstderri2g?isCode apparently failed: (t subprocesstPopentsyst executabletPIPEtSTDOUTRttimetsleeptpollR%t returncodet terminateRRtread(R,RR((s=lib/python2.7/site-packages/scipy/linalg/tests/test_lapack.pyt test_sgesdd_lwork_bug_workaround9s        t TestSytrdcB`seZdZRS(cC`sxtD]}tjdd|}td|f\}}tt||d}tj||fd|}tjd||dddd||tj|<||\}}t|d||ddd |\}} } } }t|dt ||d d tj |j d d t | tj |t | dt | d||d |\}} } } }t|dtj |d|} tj|jd} | | | | fiNitordertF(R;R<(RRnRoRRBRRR RRR9RRtreshape(RuRIRwtA_fullRSR;R<tA_tf_UR2tA_tf_LtA_tf_U_TtA_tf_L_TtA_tf_U_mtA_tf_L_mtA_tr_UtA_tr_LtA_tr_U_TtA_tr_L_T((s=lib/python2.7/site-packages/scipy/linalg/tests/test_lapack.pyttest_tfttr_trttfsR  )  <W <S!!cC`stdxttD]\}}d}|dkrat||t||dj|}nt||j|}td d|\}}||\}}t|dk||d d \}}t|dkt|} t||dd d|} t |j | | (t |} t||dd d|} t |j | | (t || t || |||\} }t|dk|||d d \} }t|dkt | t |t | t |qWd S(sz Test conversion routines between the Rectengular Full Packed (RFP) format and Standard Triangular Array (TR) iiiy?ttrttpttpttrRIiRQRRiN(RcRd(RRnRoRRBRRRR RR9RRR(RuRIRwRWRcRdtA_tp_UR2tA_tp_LtindstA_tp_U_mtA_tp_L_mR^R_((s=lib/python2.7/site-packages/scipy/linalg/tests/test_lapack.pyttest_tpttr_trttp<s2  ,    c C`s=tdx,ttD]\}}d}|dkrt||t||dj|}||jj|t|}n3t||j|}||j|t|}td d|\}}}||\}}|||\} }t |d k||| \} } t |} t | | qWd S( sk Test Cholesky factorization of a positive definite Rectengular Full Packed (RFP) format array iiiy?tpftrfR;R<RIiN(RkR;R<( RRnRoRRBRR9R RRRR( RuRIRwRRkR;R<R?R2t Achol_rfptA_chol_rR{tAchol((s=lib/python2.7/site-packages/scipy/linalg/tests/test_lapack.pyt test_pftrfes   )$  cC`swtdxfttD]X\}}d}|dkrt||t||dj|}||jj|t|}n3t||j|}||j|t|}tdd |\}}}}||\}} |||\} } ||| \} } t | d k||| \} } t |}t | t |d |d d krhd ndqWdS(s Test Cholesky factorization of a positive definite Rectengular Full Packed (RFP) format array to find its inverse iiiy?tpftriRkR;R<RIiR?iiiN(RpRkR;R<( RRnRoRRBRR9R RRRRR(RuRIRwRRpRkR;R<R?R2t A_chol_rfpt A_inv_rfptA_inv_rR{tAinv((s=lib/python2.7/site-packages/scipy/linalg/tests/test_lapack.pyt test_pftris*  )$ cC`stdxttD]\}}d}|dkrt||t||dj|}||jj|t|}n3t||j|}||j|t|}t|dfd|}t|ddfd|}t|ddfd|}t dd|\}}} } | |\} } ||| \} } ||| |\}} t | d kt t ||| |||| |\}} t | d kt t|||d |dd krdndqWdS(s Test Cholesky factorization of a positive definite Rectengular Full Packed (RFP) format array and solve a linear system iiiy?iRIitpftrsRkR;R<iR?iiN(RvRkR;R<(RRnRoRRBRR9R R RRRRRR(RuRIRwRR*tBf1tBf2RvRkR;R<R?R2RqR@((s=lib/python2.7/site-packages/scipy/linalg/tests/test_lapack.pyt test_pftrss2  )$c C`stdxttD]\}}d}|dkrt||t||dj|}||jj|t|}n3t||j|}||j|t|}|dkrdnd}tdd d j |fd |\}}}||\}} t j j|dj|} ||dd | d|} ||| \} } t | t | j| jj d|d |ddkrdndqWdS(sT Test for performing a symmetric rank-k operation for matrix in RFP format. iiiy?iRthR;R<s{}frkRIiR?iiiN(RRnRoRRBRR9R RtformatR(RRRRD( RuRIRwRtprefixR;R<tshfrkR?R{R<tAfp_outtA_out((s=lib/python2.7/site-packages/scipy/linalg/tests/test_lapack.pyttest_sfrk_hfrks$  )$ -(St __future__RRRRRRt functoolsRt numpy.testingRRRRRR tpytestR RtnumpyR(R R R RRRRRt numpy.randomRRt scipy.linalgRR$RRRRtscipy.linalg.lapackRRRet ImportErrorR%Rtscipy.linalg.blasRtfloat32RRpt complex64t complex128RtRotobjectRRaRgRRRRRRtmarktxslowRRR RR&R-R1R:RBRPRbRjRoRuRyR(((s=lib/python2.7/site-packages/scipy/linalg/tests/test_lapack.pyts^   .  :"   `   * *#HO 1    " ) : : )  %