B <ÓFd,êã@s~ dZddlmZmZmZddlZddlZddlZddlZddl Z ddl Zddl mZm Z mZmZmZmZmZddl mZmZmZmZmZddl mZddlmZmZmZmZmZddlm Z dd l!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+d d„Z,dd „Z-e&Z.d¿dd„Z&dd„Z/dd„Z0dd„Z1ddddddhZ2Gdd„de3ƒZ4d d!„Z5ej6 7d"¡gZ8e8e5de4d#ed$d%gd&d'gge d(ed%d$ge d(ƒe4d)ed$d%gd&d'gged(ed%d$ged(ƒe4d*ed$d%gd&d'gged(ed%d$d'gd&d'd+gged(ƒe4d,ed-d.gd/d0gged(ed1d-ged(ƒe4d2ed-d.gd/d0gged(ed1d-ged(ƒe4d3ed-d.gd/d0gged(ed1d-d4gd5d6d7gged(ƒe4d8ej9d9ed(ej9d:ed(dhd;e4dej6 ;d?d?¡ej6 ;d?¡ƒe4d@ej6 ;dAdA¡ej6 ;dA¡ƒe4dBdAdCgdDdEggdCdAgƒe4dFed$d%gd&d'ggƒed%d$gƒj<ƒe4dGed$d%gd&d'ggƒed%d$gƒj<ƒg ƒ7Z8e8e5de4dHed$d%d&gd&d'd+gge d(ed%d$ge d(ƒe4dIed$d%gd&d'gdJd+gge d(ed%d$d&ge d(ƒe4dKed$d%d&gd&d'd+gged(ed%d$ged(ƒe4dLed$d%gd&d'gdJd+gged(ed%d$d&ged(ƒe4dMedNdOdPgdQdRdSgged(ed1d-ged(ƒe4dTedNdOgdPdUgdVdWgged(ed1d-dPged(ƒe4dXedNdOdPgdQdRdSgged(ed1d-ged(ƒe4dYedNdOgdPdUgdVdWgged(ed1d-dPged(ƒe4dZedNdOdPgdQdRdSgged(ed1d-gd[d\gged(ƒe4d]edNdOgdPdUgdVdWgged(ed1d-gd[d\gd[d\gged(ƒe4d^ej6 ;d?d_¡ej6 ;d?¡ƒe4d`ej6 ;dAda¡ej6 ;dA¡ƒe4dbej6 ;dadA¡ej6 ;da¡ƒe4dcej6 ;ddE¡ej6 ;d¡dhd;e4ddej6 ;dEd¡ej6 ;dE¡dhd;gƒ7Z8e8e5de4deed$d%gd%d$gge d(dƒe4dfed$d%gd%d$gged(dƒe4dged$d.gdhdAgged(dƒe4died$d.gdhdAgged(dƒe4djej9d9ed(ddhd;e4dkdAdCgdCdAggdƒe4dFed$d%gd%d$ggƒdƒe4dled$d%gd%d$ggƒdƒe4dmej6 ;dAdA¡dƒg ƒ7Z8dndo„Z=e8e=ƒ7Z8dpdq„Z>drds„Z?e8e?ƒ7Z8e@ƒe@ƒfdtdu„ZAGdvdw„dwe3ƒZBGdxdy„dye3ƒZCGdzd{„d{e3ƒZDGd|d}„d}e3ƒZEGd~d„de3ƒZFGd€d„de3ƒZGd‚dƒ„ZHd„d…„ZIGd†d‡„d‡eBeEƒZJGdˆd‰„d‰eBeEƒZKGdŠd‹„d‹eBeEƒZLGdŒd„deBeEƒZMGdŽd„deBeEƒZNGdd‘„d‘eBeEƒZOGd’d“„d“eBƒZPGd”d•„d•e3ƒZQGd–d—„d—eBeCƒZRGd˜d™„d™eBeEƒZSGdšd›„d›eBeCƒZTGdœd„de3ƒZUGdždŸ„dŸe3ƒZVGd d¡„d¡eDeGƒZWGd¢d£„d£eDeGƒZXGd¤d¥„d¥e3ƒZYGd¦d§„d§e3ƒZZGd¨d©„d©eYƒZ[Gdªd«„d«eYƒZ\Gd¬d„deYƒZ]Gd®d¯„d¯e3ƒZ^d°d±„Z_Gd²d³„d³e3ƒZ`Gd´dµ„dµe3ƒZad¶d·„Zbd¸d¹„Zcdºd»„ZdGd¼d½„d½e3ƒZeefd¾k rze(ƒdS)Àz# Test functions for linalg module é)ÚdivisionÚabsolute_importÚprint_functionN)ÚarrayÚsingleÚdoubleÚcsingleÚcdoubleÚdotÚidentity)ÚmultiplyÚ atleast_2dÚinfÚasarrayÚmatrix)Úlinalg)Úmatrix_powerÚnormÚmatrix_rankÚ multi_dotÚLinAlgError)Ú_multi_dot_matrix_chain_order) Úassert_Úassert_equalÚ assert_raisesÚassert_array_equalÚassert_almost_equalÚassert_allcloseÚrun_module_suiteÚdecÚSkipTestÚsuppress_warningscCs |p|S)N©)ÚaÚbr"r"úQ/opt/alt/python37/lib64/python3.7/site-packages/numpy/linalg/tests/test_linalg.pyÚifthensr&cCs |p|S)Nr")r#r$r"r"r%Úimplysr'éécKs8t|ƒjjttfkr|}n|}t||fd|i|—ŽdS)NÚdecimal)rÚdtypeÚtyperrÚold_assert_almost_equal)r#r$Úsingle_decimalÚdouble_decimalÚkwr*r"r"r%r$srcCstttttttti|S)N)rrrr )r+r"r"r%Úget_real_dtype,sr1cCstttttttti|S)N)rrrr )r+r"r"r%Úget_complex_dtype1sr2cCs|ttfkrdSdSdS)Ngñhãˆµøä>g•dyáý¥=)rr)r+r"r"r%Úget_rtol6sr3ÚsquareÚ nonsquareÚ hermitianÚgeneralizedzsize-0Ústridedc@s*eZdZeƒfdd„Zdd„Zdd„ZdS)Ú LinalgCasecCs.tt|tƒƒ||_||_||_t|ƒ|_dS)z¢ A bundle of arguments to be passed to a test case, with an identifying name, the operands a and b, and a set of tags to filter the tests N)rÚ isinstanceÚstrÚnamer#r$Ú frozensetÚtags)Úselfr<r#r$r>r"r"r%Ú__init__Es zLinalgCase.__init__cCs||j|j|jddS)zN Run the function `do` on this test case, expanding arguments )r>N)r#r$r>)r?Údor"r"r%ÚcheckPszLinalgCase.checkcCsd|jfS)Nz)r<)r?r"r"r%Ú__repr__VszLinalgCase.__repr__N)Ú__name__Ú __module__Ú__qualname__Úsetr@rBrCr"r"r"r%r9Dsr9cCs0|tkstdƒ‚x|D]}|j|hB|_qW|S)z^ Add the given tag (a string) to each of the cases (a list of LinalgCase objects) zInvalid tag)Úall_tagsÚAssertionErrorr>)ÚtagZcasesÚcaser"r"r%Ú apply_tagYs rLiÒrgð?g@g@g@)r+rZdouble_2g@ryð?@y@@y@@y@@y@ð?r Z cdouble_2yð?@yð?Àyð?Àyð?ÀZ0x0)rr)r)r>Z 0x0_matrix)réZ8x8éZ1x1rMZnonarrayéééZ matrix_b_onlyZmatrix_a_and_bZsingle_nsq_1Zsingle_nsq_2g@Zdouble_nsq_1Zdouble_nsq_2Z csingle_nsq_1yð?ð?y@@y@Ày@Ày@"@y@@Z csingle_nsq_2y@"Ày@Ày@ @Z cdouble_nsq_1Z cdouble_nsq_2Zcdouble_nsq_1_2yð?ð¿y@ÀZcdouble_nsq_2_2Z8x11éZ1x5éZ5x1Z0x4Z4x0ZhsingleZhdoubleZhcsingley@ÀZhcdoubleZhemptyZ hnonarrayZhmatrix_a_and_bZhmatrix_1x1cCsg}xtD]}t|jtjƒs"qt |jd|jd|jg¡}|jdkrPd}nt |jd|jd|jg¡}t|jd|||j dhBd}| |¡t |jgdd¡ d|jj¡}|jdkrÊd}n$t |jgdd¡ d|jj¡}t|jd |||j dhBd}| |¡qW|S) NrOrPér(Z_tile3r7)r>)rPrOZ_tile213) ÚCASESr:r#ÚnpÚndarrayrr$r9r<r>ÚappendÚreshapeÚshape)Ú new_casesrKr#r$Únew_caser"r"r%Ú_make_generalized_casesös& $ $r]ccsît|tjƒs|dfVdSdg|j}d|d<|jdkr@d|d<|jdkrRd |d <x”tjt|ƒŽD]€}dd„t|j|ƒDƒ}td d„|Dƒƒ}tj ||j d}| tj¡ d¡||}||d<| |j¡}tt ||k¡ƒ|dd dd„|Dƒ¡fV|jdkrJ|jddkrJt|jƒ}d|d<tjjj||d}|dfV|jdkr”|jddkr”t|jƒ}d|d<tjjj||d}|dfV|jdkrd|jdd…dkrdt|jƒ}d|d<d|d<tjjj||d}|dfVqdWdS)z< Generate cartesian product of strides for all axes ZnopN)rM)rMrPéüÿÿÿéÿÿÿÿrMéþÿÿÿrO)rMr^éýÿÿÿcSsg|]\}}t||ƒ‘qSr")Úabs)Ú.0r#r$r"r"r%ú (sz%_stride_comb_iter..cSsg|]}tdd|ƒ‘qS)N)Úslice)rcÚrepeatr"r"r%rd)s)r+lï>[=.Zstride_Ú_cSsg|]}d|‘qS)z%+dr")rcÚjr"r"r%rd2sr)ÚstridesZstride_xxx_0Zstride_xxx_0_x)rMrMZstride_xxx_0_0)r:rVrWÚndimÚ itertoolsÚproductÚtupleÚziprZÚemptyr+ÚviewZuint32ZfillÚ __class__rÚallÚjoinÚlistriÚlibZ stride_tricksZ as_strided)ÚxZ stride_setZrepeatsÚ new_shapeZslicesZxiÚsr"r"r%Ú_stride_comb_itersF ryc Cszg}xptD]h}xbt|jƒD]T\}}xJt|jƒD]<\}}t|jd|d||||jdhBd}| |¡q.WqWq W|S)Nrgr8)r>)rUryr#r$r9r<r>rX)r[rKr#Za_labelr$Zb_labelr\r"r"r%Ú_make_strided_casesFs rzc CsnxhtD]`}|j|@|krq|j|@r&qy| |¡Wqtk rdd|}|t ¡7}t|ƒ‚YqXqWdS)zl Run func on each of the cases with all of the tags in require, and none of the tags in exclude zIn test case: %r N)rUr>rBÚ ExceptionÚ tracebackÚ format_excrI)ÚfuncÚrequireÚexcluderKÚmsgr"r"r%Ú_check_casesWs r‚c@seZdZdd„Zdd„ZdS)ÚLinalgSquareTestCasecCst|jdhddhddS)Nr4r7zsize-0)rr€)r‚rA)r?r"r"r%Ú test_sq_casesmsz"LinalgSquareTestCase.test_sq_casescCst|jddhdhddS)Nr4zsize-0r7)rr€)r‚rA)r?r"r"r%Útest_empty_sq_casespsz(LinalgSquareTestCase.test_empty_sq_casesN)rDrErFr„r…r"r"r"r%rƒksrƒc@seZdZdd„Zdd„ZdS)ÚLinalgNonsquareTestCasecCst|jdhddhddS)Nr5r7zsize-0)rr€)r‚rA)r?r"r"r%Útest_nonsq_casesvsz(LinalgNonsquareTestCase.test_nonsq_casescCst|jddhdhddS)Nr5zsize-0r7)rr€)r‚rA)r?r"r"r%Útest_empty_nonsq_casesysz.LinalgNonsquareTestCase.test_empty_nonsq_casesN)rDrErFr‡rˆr"r"r"r%r†tsr†c@seZdZdd„Zdd„ZdS)ÚHermitianTestCasecCst|jdhddhddS)Nr6r7zsize-0)rr€)r‚rA)r?r"r"r%Útest_herm_cases~sz!HermitianTestCase.test_herm_casescCst|jddhdhddS)Nr6zsize-0r7)rr€)r‚rA)r?r"r"r%Útest_empty_herm_casessz'HermitianTestCase.test_empty_herm_casesN)rDrErFrŠr‹r"r"r"r%r‰|sr‰c@s(eZdZejdd„ƒZejdd„ƒZdS)ÚLinalgGeneralizedSquareTestCasecCst|jddhdhddS)Nr7r4zsize-0)rr€)r‚rA)r?r"r"r%Útest_generalized_sq_cases‡sz9LinalgGeneralizedSquareTestCase.test_generalized_sq_casescCst|jdddhddS)Nr7r4zsize-0)r)r‚rA)r?r"r"r%Útest_generalized_empty_sq_cases‹sz?LinalgGeneralizedSquareTestCase.test_generalized_empty_sq_casesN)rDrErFrÚslowrrŽr"r"r"r%rŒ…srŒc@s(eZdZejdd„ƒZejdd„ƒZdS)Ú"LinalgGeneralizedNonsquareTestCasecCst|jddhdhddS)Nr7r5zsize-0)rr€)r‚rA)r?r"r"r%Útest_generalized_nonsq_cases’sz?LinalgGeneralizedNonsquareTestCase.test_generalized_nonsq_casescCst|jdddhddS)Nr7r5zsize-0)r)r‚rA)r?r"r"r%Ú"test_generalized_empty_nonsq_cases–szELinalgGeneralizedNonsquareTestCase.test_generalized_empty_nonsq_casesN)rDrErFrrr‘r’r"r"r"r%rsrc@s(eZdZejdd„ƒZejdd„ƒZdS)ÚHermitianGeneralizedTestCasecCst|jddhdhddS)Nr7r6zsize-0)rr€)r‚rA)r?r"r"r%Útest_generalized_herm_casessz8HermitianGeneralizedTestCase.test_generalized_herm_casescCst|jdddhdhddS)Nr7r6zsize-0Znone)rr€)r‚rA)r?r"r"r%Ú!test_generalized_empty_herm_cases£sz>HermitianGeneralizedTestCase.test_generalized_empty_herm_casesN)rDrErFrrr”r•r"r"r"r%r“›sr“cCsÆt|ƒ}|jdkr¸|j|jkr<|jdd…|jdd…}n(|j|jdkr\|jdd…}ntdƒ‚tj|t ||¡d}x8tjt t |jdd…ƒŽD]}t||||ƒ||<q–W|St||ƒSdS)NrPr_rMzNot implemented...)r+r`)rrjrZÚ ValueErrorrVroZcommon_typerkrlÚmapÚranger )r#r$rwÚrÚcr"r"r%Údot_generalizedªs r›cCspt|ƒ}|jdkr^tj|j|jd}x4tjtt |jdd…ƒŽD]}t |jdƒ||<q@W|St |jdƒSdS)NrP)r+r`r)rrjrVrorZr+rkrlr—r˜r)r#r™ršr"r"r%Úidentity_like_generalized½s rœc@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) Ú TestSolvecCs:t ||¡}t|t||ƒƒttt|tƒt|tƒƒƒdS)N)rÚsolverr›rr'r:r)r?r#r$r>rvr"r"r%rAÊszTestSolve.doccs,dd„}xttttgD]}||fVqWdS)NcSs2tjddgddgg|d}tt ||¡j|ƒdS)NrMgà?)r+)rVrrrržr+)r+rvr"r"r%rBÐsz#TestSolve.test_types..check)rrrr )r?rBr+r"r"r%Ú test_typesÏszTestSolve.test_typesc Cs.Gdd„dtjƒ}t d¡ ddd¡}t d¡ ddd¡ |¡}t ||¡dd…dd…dd…f}t |dd…dd…dd…f|dd…dd…dd…f¡}t||ƒtt ||ƒƒt tjtj|dd…dd…dd…f|ƒt ttj||dd…dd…dd…fƒt d¡ ddd¡}t ttj||ƒt ttj|dd…|dd…ƒt d¡ dd¡ |¡}t ||¡dd…dd…f}t |dd…dd…dd…f|dd…dd…f¡}t||ƒtt ||ƒƒt d¡ dd¡}t ttj||ƒt ttj|dd…|dd…ƒt ttj|dd…dd…dd…f|ƒdS) Nc@seZdZdS)z,TestSolve.test_0_size..ArraySubclassN)rDrErFr"r"r"r%Ú ArraySubclass×sr rNrOr(rMrPr) rVrWÚarangerYrprržrrr:rrr–)r?r r#r$ÚexpectedÚresultr"r"r%Útest_0_sizeÖs*"8 (& 2 zTestSolve.test_0_sizecCsGdd„dtjƒ}t d¡ ddd¡}t d¡ ddd¡ |¡}t ||¡dd…dd…dd…f}t ||dd…dd…dd…f¡}t||ƒtt ||ƒƒt ||¡dd…dd…dd…f}t |dd…dd…dd…f|dd…dd…dd…f¡}t||ƒtt ||ƒƒdS) Nc@seZdZdS)z.TestSolve.test_0_size_k..ArraySubclassN)rDrErFr"r"r"r%r ùsr rQrMrOr(rPr) rVrWr¡rYrprržrrr:)r?r r#r$r¢r£r"r"r%Ú test_0_size_k÷s"" "8 zTestSolve.test_0_size_kN)rDrErFrArŸr¤r¥r"r"r"r%rÈs!rc@s$eZdZdd„Zdd„Zdd„ZdS)ÚTestInvcCs<t |¡}tt||ƒt|ƒƒttt|tƒt|tƒƒƒdS)N) rÚinvrr›rœrr'r:r)r?r#r$r>Za_invr"r"r%rAs z TestInv.doccs,dd„}xttttgD]}||fVqWdS)NcSs0tjddgddgg|d}tt |¡j|ƒdS)NrMgà?)r+)rVrrrr§r+)r+rvr"r"r%rBsz!TestInv.test_types..check)rrrr )r?rBr+r"r"r%rŸszTestInv.test_typescCs²Gdd„dtjƒ}tjdtjd |¡}t |¡}t|jj tj kƒt|j|jƒtt ||ƒƒtjdtjd |¡}t |¡}t|jj tjkƒt|j|jƒtt ||ƒƒdS)Nc@seZdZdS)z*TestInv.test_0_size..ArraySubclassN)rDrErFr"r"r"r%r sr )rrMrM)r+)rr)rVrWÚzerosÚint_rprr§rr+r,Úfloat64rrZr:Ú complex64)r?r r#Úresr"r"r%r¤s zTestInv.test_0_sizeN)rDrErFrArŸr¤r"r"r"r%r¦ sr¦c@s$eZdZdd„Zdd„Zdd„ZdS)ÚTestEigvalscCs&t |¡}t |¡\}}t||ƒdS)N)rÚeigvalsÚeigr)r?r#r$r>ÚevÚevaluesÚevectorsr"r"r%rA,s zTestEigvals.doccs,dd„}xttttgD]}||fVqWdS)NcSs`tjddgddgg|d}tt |¡j|ƒtjddgddgg|d}tt |¡jt|ƒƒdS)NrMgà?)r+r_)rVrrrr®r+r2)r+rvr"r"r%rB2sz%TestEigvals.test_types..check)rrrr )r?rBr+r"r"r%rŸ1szTestEigvals.test_typescCs²Gdd„dtjƒ}tjdtjd |¡}t |¡}t|jj tj kƒtd|jƒtt |tjƒƒtjdtjd |¡}t |¡}t|jj tjkƒtd|jƒtt |tjƒƒdS)Nc@seZdZdS)z.TestEigvals.test_0_size..ArraySubclassN)rDrErFr"r"r"r%r <sr )rrMrM)r+)rrM)rr)r)rVrWr¨r©rprr®rr+r,rªrrZr:r«)r?r r#r¬r"r"r%r¤:s zTestEigvals.test_0_sizeN)rDrErFrArŸr¤r"r"r"r%r*s rc@s$eZdZdd„Zdd„Zdd„ZdS)ÚTestEigcCsdt |¡\}}tt||ƒt |¡t |¡dddd…ft|jƒdtt t |tƒt |tƒƒƒdS)N.)Úrtol)rr¯rr›rVrr3r+rr'r:r)r?r#r$r>r±r²r"r"r%rAOs z TestEig.doccs,dd„}xttttgD]}||fVqWdS)NcSstjddgddgg|d}tj |¡\}}t|j|ƒt|j|ƒtjddgddgg|d}tj |¡\}}t|jt|ƒƒt|jt|ƒƒdS)NrMgà?)r+r_)rVrrr¯rr+r2)r+rvÚwÚvr"r"r%rBWsz!TestEig.test_types..check)rrrr )r?rBr+r"r"r%rŸVszTestEig.test_typescCsúGdd„dtjƒ}tjdtjd |¡}t |¡\}}t|jj tj kƒt|jj tj kƒt|j|jƒtd|jƒtt |tjƒƒtjdtjd |¡}t |¡\}}t|jj tjkƒt|jj tjkƒt|j|jƒtd|jƒtt |tjƒƒdS)Nc@seZdZdS)z*TestEig.test_0_size..ArraySubclassN)rDrErFr"r"r"r%r gsr )rrMrM)r+)rrM)rr)r)rVrWr¨r©rprr¯rr+r,rªrrZr:r«)r?r r#r¬Úres_vr"r"r%r¤eszTestEig.test_0_sizeN)rDrErFrArŸr¤r"r"r"r%r³Msr³c@s$eZdZdd„Zdd„Zdd„ZdS)ÚTestSVDc Cs¤d|krtttj|dƒdSt |d¡\}}}t|tt |¡t |¡dddd…ft |¡ƒt|j ƒdt tt|t ƒt|t ƒƒƒt tt|t ƒt|t ƒƒƒdS)Nzsize-0r.)r´)rrrÚsvdrr›rVrr3r+rr'r:r)r?r#r$r>ÚurxÚvtr"r"r%rA~s& z TestSVD.doccs,dd„}xttttgD]}||fVqWdS)NcSsttjddgddgg|d}t |¡\}}}t|j|ƒt|jt|ƒƒt|j|ƒtj|dd}t|jt|ƒƒdS)NrMgà?)r+F)Ú compute_uv)rVrrr¹rr+r1)r+rvrºrxZvhr"r"r%rB‹sz!TestSVD.test_types..check)rrrr )r?rBr+r"r"r%rŸŠs zTestSVD.test_typescCsdtjdtjd}ttjtj|ƒtjdtjd}ttjtj|ƒtjdtjd}ttjtj|ƒdS)N)rr)r+)rrM)rMr)rVr¨r«rrrr¹)r?r#r"r"r%r¤—szTestSVD.test_0_sizeN)rDrErFrArŸr¤r"r"r"r%r¸|s r¸c@seZdZdd„Zdd„ZdS)ÚTestCondSVDcCsZt|ƒ}d|kr&tttj|dddStj|dd}t|d|dt |¡ddddS) Nzsize-0F)r¼).r).r_rSrR)r.r/)rrrrr¹rÚcond)r?r#r$r>ršrxr"r"r%rA¤szTestCondSVD.docCsDt dddgdddgdddgg¡}tt |¡t |d¡dƒdS)Ngð?g@rgÀg@g@)N.)rVrrrr¾)r?ÚAr"r"r%Útest_stacked_arrays_explicitly®s"z*TestCondSVD.test_stacked_arrays_explicitlyN)rDrErFrArÀr"r"r"r%r½¢s r½c@seZdZdd„Zdd„ZdS)Ú TestCond2cCs\t|ƒ}d|kr&tttj|dddStj|dd}t|d|dt |d¡ddd dS) Nzsize-0F)r¼).r).r_rOrSrR)r.r/)rrrrr¹rr¾)r?r#r$r>ršrxr"r"r%rAµszTestCond2.docCsHt dddgdddgdddgg¡}tt |d¡t |dd¡dƒdS) Ngð?g@rgÀg@g@rO)N.)rVrrrr¾)r?r¿r"r"r%rÀ¿s"z(TestCond2.test_stacked_arrays_explicitlyN)rDrErFrArÀr"r"r"r%rÁ³s rÁc@seZdZdd„ZdS)ÚTestCondInfcCs6tdddgdddgdddggƒ}tt |t¡dƒdS)Ngð?rgÀg@)rrrr¾r)r?r¿r"r"r%ÚtestÆs zTestCondInf.testN)rDrErFrÃr"r"r"r%rÂÄsrÂc@seZdZdd„ZdS)ÚTestPinvcCsDt |¡}tt||ƒ |¡|dddttt|tƒt|tƒƒƒdS)NrSrR)r.r/)rÚpinvrr rr'r:r)r?r#r$r>Za_ginvr"r"r%rAÍs zTestPinv.doN)rDrErFrAr"r"r"r%rÄËsrÄc@s,eZdZdd„Zdd„Zdd„Zdd„Zd S) ÚTestDetc CsÌt |¡}t |¡\}}t|ƒjjttfkrÚdrxZldZadr°Úmr"r"r%rAÖs z TestDet.docCstt dgg¡dƒttt dgg¡ƒtƒtt dgg¡dƒttt dgg¡ƒtƒtt dgg¡dtfƒttt dgg¡dƒtƒttt dgg¡dƒtƒtt dgg¡dtfƒttt dgg¡dƒtƒttt dgg¡dƒtƒdS)NgyrrM)rrrÈr,rr rÉr)r?r"r"r%Ú test_zeroçszTestDet.test_zeroccs,dd„}xttttgD]}||fVqWdS)NcSs^tjddgddgg|d}ttj |¡j|ƒtj |¡\}}t|jt|ƒƒt|j|ƒdS)NrMgà?)r+)rVrrrrÈr+rÉr1)r+rvZphrxr"r"r%rBõs z!TestDet.test_types..check)rrrr )r?rBr+r"r"r%rŸôszTestDet.test_typescCsðtjdtjd}t |¡}t|dƒt|jjtjkƒt |¡}t|dƒt|djjtjkƒt|djjtj kƒtjdtjd}t |¡}t|dƒt|jjtjkƒt |¡}t|dƒt|djjtjkƒt|djjtjkƒdS)N)rr)r+gð?)rMrrrM)rVr¨r«rrÈrrr+r,rÉÚfloat32rª)r?r#r¬r"r"r%r¤þs zTestDet.test_0_sizeN)rDrErFrArÎrŸr¤r"r"r"r%rÆÔs rÆc@seZdZdd„ZdS)Ú TestLstsqcCsdd|krtttj||ƒdSt |¡}|j\}}t |d¡\}}} t ||¡\} }}} ||krzt|t || ƒƒt ||ƒn t ||ƒt| | |¡ƒ||krú||krút tt || ¡|ƒ¡dj dd}t |¡}t |¡jdkrød|_t |j|jƒnt g¡ t| ƒ¡}t||ƒtt |jtj¡ƒttt|tƒt| tƒƒƒttt|tƒt|tƒƒƒdS)Nzsize-0rrO)rÇrM)rM)rrrZlstsqrVrrZr¹rr rZ__array_wrap__rbÚsumrjrrpr,rZ issubdtyper+Úfloatingr'r:r)r?r#r$r>ÚarrrÍÚnrºrxr»rvZ residualsZrankZsvZ expect_residsr"r"r%rAs. & zTestLstsq.doN)rDrErFrAr"r"r"r%rÐsrÐc@sÜeZdZeddgddggƒZeddgddggƒZeddgddggƒZedd gd dggƒZedƒZ e ddd…f ¡Ze ddd…fe ddd…f<ee ddd…f<d d„Zdd„Z dd„Zdd„Zdd„Zdd„Zdd„ZdS)ÚTestMatrixPowerrrMr_rQiùÿÿÿr`é gš™™™™™¹?gš™™™™™ @g333333ó?gffffffæ?NcCs(tt|jdddddƒ|jƒdS)NrOédié rM)rrÚR90)r?r"r"r%Útest_large_power=sz TestMatrixPower.test_large_powercCs&tt|jddddƒtdƒƒdS)NrOr×irØ)rrrÙr)r?r"r"r%Útest_large_power_trailing_zeroAsz.TestMatrixPower.test_large_power_trailing_zeroccs0dd„}x"|j|j|jgD]}||fVqWdS)NcSs0t|dƒ}t|t|jdƒƒt|j|jƒdS)Nr)rrrrZr+)ÚMÚmzr"r"r%ÚtzFs z'TestMatrixPower.testip_zero..tz)ÚArb22ÚarbfloatÚlarge)r?rÞrÜr"r"r%Útestip_zeroEszTestMatrixPower.testip_zeroccs0dd„}x"|j|j|jgD]}||fVqWdS)NcSs&t|dƒ}t||ƒt|j|jƒdS)NrM)rrr+)rÜrÝr"r"r%rÞNs z&TestMatrixPower.testip_one..tz)rßràrá)r?rÞrÜr"r"r%Ú testip_oneMszTestMatrixPower.testip_oneccs0dd„}x"|j|j|jgD]}||fVqWdS)NcSs,t|dƒ}t|t||ƒƒt|j|jƒdS)NrO)rrr r+)rÜrÝr"r"r%rÞVs z&TestMatrixPower.testip_two..tz)rßràrá)r?rÞrÜr"r"r%Ú testip_twoUszTestMatrixPower.testip_twoccs4dd„}x&|j|j|j|jgD]}||fVqWdS)NcSs(t|dƒ}tt|jdƒt||ƒƒdS)Nr_r)rrrrZr )rÜrÝr"r"r%rÞ^s z)TestMatrixPower.testip_invert..tz)rÙrßràrá)r?rÞrÜr"r"r%Ú testip_invert]szTestMatrixPower.testip_invertcs$ddl}t|jjj‡fdd„ƒdS)NrcstˆjdƒS)Nr_)rÚnoninvr")r?r"r%Úgóz;TestMatrixPower.test_invert_noninvertible..)Únumpy.linalgrrr)r?Únumpyr")r?r%Útest_invert_noninvertibleds z)TestMatrixPower.test_invert_noninvertible)rDrErFrrÙrßræràrráÚcopyÚtrÚrÛrârãrärårër"r"r"r%rÕ2srÕc@seZdZdd„ZdS)Ú TestBoolPowercCs(tddgddggƒ}tt|dƒ|ƒdS)NTFrO)rrr)r?r¿r"r"r%Útest_squarelszTestBoolPower.test_squareN)rDrErFrïr"r"r"r%rîjsrîc@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „ZdS)ÚTestEigvalshcCs^t |d¡}t |¡\}}|jddt||t|jƒdt |d¡}t||t|jƒddS)NÚLr_)rÇ)r´ÚU)rÚeigvalshr¯Úsortrr3r+)r?r#r$r>r°r±r²Úev2r"r"r%rAsszTestEigvalsh.doccs,dd„}xttttgD]}||fVqWdS)NcSs:tjddgddgg|d}tj |¡}t|jt|ƒƒdS)NrMgà?)r+)rVrrrórr+r1)r+rvrµr"r"r%rBsz&TestEigvalsh.test_types..check)rrrr )r?rBr+r"r"r%rŸ~szTestEigvalsh.test_typescCsXtjddgddggtjd}tttjj|ddtttjj|dƒtttjj|dƒdS)NrMgà?)r+Úlrong)ÚUPLOÚlowerÚupper)rVrrÏrr–rró)r?rvr"r"r%Útest_invalid†szTestEigvalsh.test_invalidcCsìtjddgddggtjd}tjddgddggtjd}tjddgtjd}ttjƒ}tj |¡}t|||dtjj|dd}t|||dtjj|dd}t|||dtjj|d d}t|||dtjj|d d}t|||ddS)NrrM)r+r_)r´rñ)r÷Úlròrº)rVrrr3rrór)r?ÚKloÚKupÚtgtr´rµr"r"r%Ú test_UPLOŒs zTestEigvalsh.test_UPLOcCs²Gdd„dtjƒ}tjdtjd |¡}t |¡}t|jj tj kƒtd|jƒtt |tjƒƒtjdtjd |¡}t |¡}t|jj tjkƒtd|jƒtt |tjƒƒdS)Nc@seZdZdS)z/TestEigvalsh.test_0_size..ArraySubclassN)rDrErFr"r"r"r%r ¤sr )rrMrM)r+)rrM)rr)r)rVrWr¨r©rprrórr+r,rªrrZr:r«rÏ)r?r r#r¬r"r"r%r¤¢s zTestEigvalsh.test_0_sizeN)rDrErFrArŸrúrÿr¤r"r"r"r%rðqs rðc@s4eZdZdd„Zdd„Zdd„Zdd„Zd d „ZdS)ÚTestEighc CsÆt |¡\}}t |¡\}}|jddt||ƒtt||ƒt |¡dddd…ft |¡t |j ƒdt |d¡\}} t||ƒtt|| ƒt |¡dddd…ft | ¡t |j ƒt|ƒddS)Nr_)rÇ.)r´rò)r´Úerr_msg)rÚeighr¯rôrrr›rVrr3r+Úrepr) r?r#r$r>r°Zevcr±r²rõZevc2r"r"r%rA·s zTestEigh.doccs,dd„}xttttgD]}||fVqWdS)NcSsJtjddgddgg|d}tj |¡\}}t|jt|ƒƒt|j|ƒdS)NrMgà?)r+)rVrrrrr+r1)r+rvrµr¶r"r"r%rBËsz"TestEigh.test_types..check)rrrr )r?rBr+r"r"r%rŸÊszTestEigh.test_typescCsXtjddgddggtjd}tttjj|ddtttjj|dƒtttjj|dƒdS)NrMgà?)r+rö)r÷rørù)rVrrÏrr–rr)r?rvr"r"r%rúÓszTestEigh.test_invalidcCstjddgddggtjd}tjddgddggtjd}tjddgtjd}ttjƒ}tj |¡\}}t|||dtjj|dd\}}t|||dtjj|dd\}}t|||dtjj|d d\}}t|||dtjj|d d\}}t|||ddS)NrrM)r+r_)r´rñ)r÷rûròrº)rVrrr3rrr)r?rürýrþr´rµr¶r"r"r%rÿÙs zTestEigh.test_UPLOcCsúGdd„dtjƒ}tjdtjd |¡}t |¡\}}t|jj tj kƒt|jj tj kƒt|j|jƒtd|jƒtt |tjƒƒtjdtjd |¡}t |¡\}}t|jj tjkƒt|jj tjkƒt|j|jƒtd|jƒtt |tjƒƒdS)Nc@seZdZdS)z+TestEigh.test_0_size..ArraySubclassN)rDrErFr"r"r"r%r ñsr )rrMrM)r+)rrM)rr)r)rVrWr¨r©rprrrr+r,rªrrZr:r«rÏ)r?r r#r¬r·r"r"r%r¤ïszTestEigh.test_0_sizeN)rDrErFrArŸrúrÿr¤r"r"r"r%rµs rc@s\eZdZdZdZdd„Zdd„Zdd„Zdd „Zd d„Z dd „Z dd„Zdd„Zdd„Z dS)Ú _TestNormNcCsFttgƒdƒtttg|jdƒdƒttttg|jdƒƒdƒdS)Ng)r+)rrrÚdtr )r?r"r"r%Ú test_emptysz_TestNorm.test_emptyc Cs°t dddg¡}tjd}tjd}||}x||D]r}| |¡}t|tjƒ}tt|jj tj ƒƒt|dƒtƒ:}| td¡t|dƒ}tt|jj tj ƒƒt|dƒWdQRXt|dƒ}tt|jj tj ƒƒt|dƒt|dƒ}tt|jj tj ƒƒt|d ƒt|dƒ}tt|jj tj ƒƒt||j d ¡|j d ¡ƒt|dƒ}tt|jj tj ƒƒt||j d ¡|j d¡ƒt|tjƒ}tt|jj tj ƒƒt|d ƒq4WdS)NrMrÚ AllIntegerZAllFloatgzdivide by zero encounteredr_rOg@gà?rQgÐ?gð?)rVrÚ typecodesrÊrrrÚ issubclassr+r,rÒrr!ÚfilterÚRuntimeWarning) r?r#Úexact_typesÚ inexact_typesÚ all_typesÚeach_inexact_typesÚatÚanÚsupr"r"r%Útest_vector_return_types: z!_TestNorm.test_vector_return_typec Cs¸t dddgdddgg¡}tjd}d}||}x€|D]v}| |¡}t|tjƒ}tt|jj tj ƒƒt|dƒtƒ:}| td¡t|dƒ}tt|jj tj ƒƒt|dƒWdQRXt|dƒ}tt|jj tj ƒƒt|dƒt|d ƒ}tt|jj tj ƒƒt|d ƒt|dƒ}tt|jj tj ƒƒt|dƒt|tjƒ}tt|jj tj ƒƒt|dƒt|dƒ}tt|jj tj ƒƒt|dƒt|d ƒ}tt|jj tj ƒƒt|dddq8WdS)NrMrrZfdFDg@zdivide by zero encounteredr_gð?rOgªLXèz¶û?r`ÚfroÚnucgU&,t=Û@r()r*)rVrrrÊrrrr r+r,rÒrr!r rr-) r?r#rr rrrrrr"r"r%Útest_matrix_return_type9s@ z!_TestNorm.test_matrix_return_typecsŠddddg}ddddg}ddddg}‡fd d „}x|||fD]}||ƒq._test)r+)rr)r?r#r$ršrr¶r")r?r%Útest_vectorlsz_TestNorm.test_vectorcCsØtddgddgg|jd}tt|ƒdƒtt|dƒdƒtt|dƒd ƒtt|tƒd ƒtt|tƒdƒtt|dƒd ƒtt|dƒd ƒtt|dƒdƒtt|dƒdƒttt|dƒttt|dƒttt|dƒdS)NrMrPrSrT)r+g…¹ŽT"@rrg$@g(@g@r_g@rOg„öW³?"@r`gÑ—€Ê„ì?Znofrorar)rrrrrrr–)r?r¿r"r"r%Útest_matrix_2x2Šsz_TestNorm.test_matrix_2x2cCsÀdtjdddgdddgdddgg|jd}tt|ƒd ƒtt|d ƒd ƒtt|dƒdƒtt|tƒd ƒtt|tƒdƒtt|dƒdƒtt|dƒdƒtt|dƒdƒtt|dƒdƒdS)Ngš™™™™™¹?rMrOrPr(rrS)r+grj=Q0î?rrgþB+pcõ?gš™™™™™ñ?g333333ã?gð?r_gš™™™™™Ù?gû&Þê.dì?r`gÜ¸Õ ˆçÈ?)rVrrrrr)r?r¿r"r"r%Útest_matrix_3x3šs*z_TestNorm.test_matrix_3x3c sÈtdddgdddgg|jd‰x†ddd dddtjtjgD]h‰‡‡fd d„tˆjdƒDƒ}ttˆˆd d|ƒ‡‡fd d„tˆjd ƒDƒ}ttˆˆdd|ƒq:Wtjdd|jd ddd¡‰ˆj }xúdddddtjtjdgD]Ü‰xÖt t||ƒd¡D]¾}|\}}|d kr||7}|d kr.||7}||krLt ttˆˆ|dqþtˆˆ|d}|||‰||kr’‡‡‡fdd„tˆjˆƒDƒ}n ‡‡‡fdd„tˆjˆƒDƒ}t||ƒqþWqäWdS)NrMrOrPrQrSr()r+r_rcs$g|]}tˆdd…|fˆd‘qS)N)Úord)r)rcÚk)r¿Úorderr"r%rd±sz'_TestNorm.test_axis..)rrÇcs$g|]}tˆ|dd…fˆd‘qS)N)r)r)rcr)r¿rr"r%rd³sér`rcs*g|]"}tˆdd…j|ˆdˆd‘qS)N)rÇ)r)rÚtake)rcr)ÚBÚk_indexrr"r%rdÉscs,g|]$}tˆdd…j|ˆdjˆd‘qS)N)rÇ)r)rrÚT)rcr)r r!rr"r%rdÌs)rrrVÚInfr˜rZrrr¡rYrjrkÚcombinationsrr–) r?Z expected0Z expected1ZndrÇZrow_axisZcol_axisrÔr¢r")r¿r r!rr%Ú test_axis«s2 z_TestNorm.test_axisc Csötjdd|jd ddd¡}d}d}t|ddd }t|ddd d}tt |¡|| dd¡dd }t|j |k| |j |dd¡ƒx¨ddddddtj tj gD]Š}x„t|jƒD]v}t|||d }t|||d d}tt |¡|| ||¡dt |j ƒ}d||<t|ƒ}t|j |k| |j |||¡ƒq¬WqœWxÆdddddtj tj ddg D]¦}xžt t|jƒd¡D]ˆ}t|||d }t|||d d}tt |¡|| ||¡dt |j ƒ}d||d<d||d<t|ƒ}t|j |k| |j |||¡ƒq`WqHWdS)NrMr)r+rOrPrQzorder {0}, axis = {1}z;Shape mismatch found {0}, expected {1}, order={2}, axis={3})rrÇT)rrÇZkeepdims)r)rMrMrMr_rr`rr)rVr¡rrYrrZsqueezeÚformatrrZr#r˜rjrtrmrkÚpermutations) r?r¿Zallclose_errZ shape_errr¢ÚfoundZexpected_shaperrr"r"r%Ú test_keepdimsÐsB " z_TestNorm.test_keepdimscCsøtdddgdddgg|jd}tjdd|jd ddd¡}ttt|d d ƒttt|dd ƒtttddgd dƒtttddgddƒxcs &r>c@seZdZdd„ZdS)ÚTestCholeskycCs¶Gdd„dtjƒ}tjdtjd |¡}t |¡}t|j|jƒt |j jtjkƒt t |tjƒƒtjdtjd |¡}t |¡}t|j|jƒt |j jtjkƒt t |tjƒƒdS)Nc@seZdZdS)z/TestCholesky.test_0_size..ArraySubclassN)rDrErFr"r"r"r%r Åsr )rrMrM)r+)rMrr)rVrWr¨r©rprZcholeskyrrZrr+r,rªr:r«)r?r r#r¬r"r"r%r¤Äs zTestCholesky.test_0_sizeN)rDrErFr¤r"r"r"r%rKÁsrKcCs tjdkrd}nd}x†tjtjfD]v}tjd|d}| |¡}| d¡ ¡}t|j jdƒxrQ)r+ÚSú=)ÚsysÚ byteorderrVrÏrªr4ZnewbyteorderZbyteswaprr+rr§rÈrÅr)ZnativeZdttrÓZn_arrZsw_arrZroutiner¬r"r"r%Útest_byteorder_checkÕs rScCspt ddgddgg¡}t ddgddgg¡}t ddddg¡ddd…}||d<||d<ttjjtjj|ƒdS)NrMrOrPrQ.)rr)rVrr¨rrrr§)Z invertibleZnon_invertiblervr"r"r%Ú test_generalized_raise_multiloopêsrTcCsDd}yt ¡}Wn ttfk r0tdƒ‚YnX|dkrt d¡t d¡ddl}| |jd¡yt j j ¡Wn*t k rŠYnt tj¡YnXy0t dgg¡}t j j ddd|d||dd¡ Wn:t k r}zdt|ƒkrüt |¡Wdd}~XYnXt tj¡n$t ¡\}}t |¡|kr@tdƒ‚dS) NéÿzNot POSIX or fork failed.rrM)rrgð?zDORGQR parameter number 5zNumpy xerbla not linked in.)ÚosÚforkÚOSErrorÚAttributeErrorr ÚcloseÚresourceZ setrlimitZRLIMIT_CORErVrZlapack_liteZxerblar–Ú_exitÚ EX_CONFIGrZdorgqrr;ÚwaitÚWEXITSTATUS)Z XERBLA_OKÚpidr[r#ÚeZstatusr"r"r%Útest_xerbla_overrideøs: rbc@sDeZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Zd d„Z dS)ÚTestMultiDotc Csltj d¡}tj d¡}tj d¡}tt|||gƒ| |¡ |¡ƒtt|||gƒt |t ||¡¡ƒdS)N)r(rO)rOr()rVr9rrr )r?r¿r ÚCr"r"r%Ú(test_basic_function_with_three_arguments+s z5TestMultiDot.test_basic_function_with_three_argumentscCs\tj d¡}tj d¡}tj d¡}tj d¡}tt||||gƒ| |¡ |¡ |¡ƒdS)N)r(rO)rOr()rOrM)rVr9rrr )r?r¿r rdÚDr"r"r%Ú8test_basic_function_with_dynamic_programing_optimization5s zETestMultiDot.test_basic_function_with_dynamic_programing_optimizationcCsLtj d¡}tj d¡}tj d¡}tj d¡}tt||||gƒjdƒdS)NrO)rOr()r(rO)rOrO)rO)rVr9rrrZ)r?ÚA1dr rdrfr"r"r%Útest_vector_as_first_argument>s z*TestMultiDot.test_vector_as_first_argumentcCsLtj d¡}tj d¡}tj d¡}tj d¡}tt||||gƒjdƒdS)N)r(rO)rOr(rO)r()rVr9rrrZ)r?r¿r rdÚD1dr"r"r%Útest_vector_as_last_argumentHs z)TestMultiDot.test_vector_as_last_argumentcCsLtj d¡}tj d¡}tj d¡}tj d¡}tt||||gƒjdƒdS)NrO)rOr()r(rOr")rVr9rrrZ)r?rhr rdrjr"r"r%Ú&test_vector_as_first_and_last_argumentRs z3TestMultiDot.test_vector_as_first_and_last_argumentc Csjtj d¡tj d¡tj d¡tj d¡tj d¡tj d¡g}t ddd d ddgddd dddgddddddgddddddgddddddgddddddgg¡}tjddddddgddddddgddddddgddddddgddddddgddddddggtjd}|d8}t|dd\}}tt |dd …dd…f¡t |dd …dd…f¡ƒtt |¡t |¡ƒdS)!N)éé#)rné)rorS)rSrÖ)rÖé)rprggÃÎ@gÃ¾@g€OÂ@g€1Ç@g€ŠÍ@g‚¤@g±@gÕ»@g‚Ä@gp‡@gˆ£@gÿ´@g@@gX«@gˆ³@rrMrPrOrQrS)r+T)Zreturn_costsr_)rVr9rÚintrrrB)r?ZarraysZ m_expectedZ s_expectedrxrÍr"r"r%Útest_dynamic_programming_logic\s. z+TestMultiDot.test_dynamic_programming_logiccCs&tttgƒttttj d¡gƒdS)N)rPrP)rr–rrVr9)r?r"r"r%Útest_too_few_input_arrayszsz&TestMultiDot.test_too_few_input_arraysN) rDrErFrergrirkrlrrrsr"r"r"r%rc)s rcÚ__main__)r(r))gÚ__doc__Z __future__rrrrVrQrkr|ÚwarningsrêrVrrrrr r rrr rrrrrérrrrrZnumpy.linalg.linalgrZ numpy.testingrrrrrrrrr r!r&r'r-r1r2r3rHÚobjectr9rLr9ZseedrUrorpZrandr"r]ryrzrGr‚rƒr†r‰rŒrr“r›rœrr¦rr³r¸r½rÁrÂrÄrÆrÐrÕrîrðrrr+r0r1r2r3r=r>rKrSrTrbrcrDr"r"r"r%ÚsŠ$0 / B #/& > 8DQ ^1V