a ddlmZddlZddlTejjZejZej Z ej Z ej Z ej Z e gdgdgdgZgdZe gdgd gd gd gZgd Ze gd gdgdgdgdgZgdZe gdgdgdgdgdgZgdZe gdgdgdgZe gdgdgdgZgdZe d d!gd"d gd#d$ggZgd%Ze gd&gd'gd(gd)gZgd*Ze gd+gd,gd-gZgd.Ze gd/gd0gd1gZgd2Z d3d4Z!d5d6Z"d7d8Z#d9d:Z$d;d<Z%d=d>Z&d?d@Z'dAdBZ(dCdDZ)dEdFZ*e+dGdHdIZ,dJdKZ-dLdMZ.dNdOZ/dPdQZ0dRdSZ1dS)T)divisionN)*))rr)rrr)r)rr r)rrr )rrr )rrr )rr r r )rrr r r)rrrrr )r rrr)rrr i)rrr )rrrr2)gQx$@g= ףp=ggQ롿gGz@)gNbX9gRQ?g(\@g(\B!gHzG?)gRQgQ?g rпgp= ף"@gE)g)\#@g(\@g@1g;On?g\(8@)gQ"g33333sA@gp= ףS@gQ?g33333sP@)gfffff!@g ףp=4@gGz@ffffff@gףp= 7M@)rrr )r r r)rri)g ?gZB>@gr3܀o@)gsFZ*o@g#GGge,i9@)g}.AgR#@g(~Ƅ.)g7@gQ/gQ@rgrr r)rrr )rrr)r rr)r r r )rrr )rrrr )r rr )rrr )r r g@) g/)??@r)g@g@g@)g@g @g"@)?rrcCst}t}t|\}}t|||}t||}|gdks>Jdd|DgdksXJt}t}t|\}}t|||}t||}|gdksJdd|DgdksJtd}|}t|d d ||ksJdS) N)rrrrcSsg|]}t|dqSround.0ir h/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/mpmath/tests/test_linalg.py Oz"test_LU_decomp..)g-P@grhA@g(mUg^ \^v@g+6CV@)rrr rcSsg|]}t|dqSrrrr r r!r"Wr#)gw]@g_DP@gIϘ}X?gRxgE9rr) overwrite) A3copyb3 LU_decompL_solveU_solveA4b4 randmatrix)AbpyxZbakr r r!test_LU_decompHs$       r3cCs<tttfD],}t|}t||t|jddks Jq dS)Nrg+=)A1A2A5inversemnormeyerows)r.invr r r! test_inverse^sr<c Cs$dt_tt}}tt||\}}}}|ttdtdtddgdtdtdtd gdtd td td gd tdtdtdggksJ|ddtdgksJtt |dddksJgd}dd}g}g} t ddD]Z} || }tt||\}}}}t|}t|}| t |d| t t |||dqdd|DgdksVJt t|t| t dksvJdd} g} g} t ddD]n} | | }t|\}}}}| t |d| t t |ddd| d f||dd| d fdqt t| t| t dks JdS)!Nz3.0z-2.0z-1.0rgz3.333333333333333z-2.9999999999999991z2.0z-0.66666666666666674z2.8142135623730948z-2.8284271247461898rz-1.3333333333333333z-0.20000000000000018z4.2426406871192857r z-1.4142135623730949rrgv@) gJ +Y@gy&1,M@g%C;B@g(\O8@g1Z1@gn(@gMb"@g(\@gq= ףp@gʡE@gy&1, @g/$@cs>g}tddD]&|fddtd|dDqt|S)Nr csg|]}d|dqS)rrr )rjrr r!r"xr#z3test_householder..coeff..)rangeappendmatrix)nr.r r@r!coeffts$ztest_householder..coeffrcSsg|]}t|dqS)rr)rresr r r!r"r#z$test_householder..)gRmX.@g8hi\?g5Żڌ^?g?g A8C?vIh%<=cSs`td||}td|dddd}|dd}t|dddf|t|d|df}|S)Nrr rminmaxcSstdt|S)Ny?)exppir2r r r!r#z9test_householder..hilbert_cmplx..r)Zhilbertr-applydiag)rDr.vr r r! hilbert_cmplxs ,z'test_householder..hilbert_cmplxr)mpdpsA8b8 householderextendrCmpfrnormrArBresidualinfr&)r.r/Hr0r2rr1rEZ residualsZrefresrDrRZresiduals_cmplxZ refres_cmplxr r r!test_householdercsL   @r_cCs8td}t|\}}}t||||ddks4JdS)NrrgV瞯<)r-Zlur8)r.PLUr r r!test_factorizationsrccCsttttttttdks Jttttttttdks@Jtttttt t tdks`Jtttt ttdttdksJtttt ttdttdksJtttt tt dt ddksJttt tt t t ddksJttt t t t dt ddksJdS)Ng|=?go@rrg333333@) rZr[A6lu_solveb6r\A7b7rUrVZqr_solveA10b10r r r r! test_solves   $$$ rlcCsPtddgddgddgg}tdtdt g}tt|t|||dksLJdS) Nry@ry@rrrg|a2U?)rCr?rZr[rf)r.r/r r r!test_solve_overdet_complexsrmcsxdt_ddgddgg}ttd}ddg}d||fd||fd ||fd ||ffD]tttffd d qVdS) Nr=rg333333?g?g?rrzlu_solve(%s, %s)zqr_solve(%s, %s)cstSN)evalr r@r r!rNr#ztest_singular..)rSrTreprzerospytestZraisesZeroDivisionError ValueError)r.Br/r r@r! test_singulars rvcCs`ttttgdgdgdgks0Jttt}ttt|ttjdks\JdS)N)rrr)rrr)r grdr) fpZcholeskyrCA9Zcholesky_solveb9rZr[r\rMr r r! test_choleskys0 rzcCsttdksJtttddks&Jtttdks:JtttdksNJttdks^JtttdksrJttddksJdS) Nrri*iCirr) detr4rr5r%r+r6rerqr r r r!test_detsr}cCsldt_tddgddgg}t|ddtdks4Jt|d dtdksNJt|d dtd kshJdS) Nr=g6?gꕲ q?ga2U0*?gBiq?cSs t|dS)Nrr8rMr r r!rNr#ztest_cond..z327065209.73817754cSs t|tSrn)r8r\rMr r r!rNr#cSs t|dS)NFr~rMr r r!rNr#z249729266.80008656)rSrTrCZcondrYr.r r r! test_conds rrcCs,tdd}ttt||ddks(JdS)Nrrgj7֖6)r-r8r7rr r r!test_precisions rcCsdt_dt_tgdgdgdg}tgd}t||}|djdksTJ|djdksfJ|d jdksxJd |dvsJd |dvsJd |d vsJdS) Nr=)z0.1z0.3z1.0)z7.1z5.5z4.8)z3.2z4.4z5.6)4z0.6z0.5rrGrrgiE'n@gFxlbO*g'ɮ@)rSrTZivrCrfdelta)ar/cr r r!test_interval_matrixs rcCs<td}t|}|jt|ks"Jd|d<|jdus8JdS)Nr)rr)r-r(Z_LU)r.ZLUr r r! test_LU_caches rcCsptdddd}tddddd}t||tdddd d}t|||}tt|||d tt|||d kslJdS) Nrg#B ;g@xDrHrrigh㈵gh㈵>r)r-rfimprove_solutionrZr[)r.r/x1Zx2r r r!test_improve_solutions  rc CstdD]}d}d}||t_d}d}td|d}t|}tdd|D}t|dkrlt|}t|}qN||}|d||} |d||} |t_t| d d } ||t_| | } |t_t| t  dsJqdt_dS) Nrr=rrcSsg|] }t|qSr )rK)rr2r r r!r"r#z!test_exp_pade..g{Gz?r Zpade)method) rArSrTrPabsr-r|ZexpmrZr\Zae) rrTextradmNZdgrZexpamZa1e2e1dr r r! test_exp_pades,      rcCs<dt_d}d}d}d}td}tddD] }d}d }t|d }td |d t}td |d t} tt|| } tt || } t | | } |rd}d } td| D]R}td| D]B}t|||t}t|||t}t ||| ||f<qqnRd }d } td| D]>}td| D],}t|||t}t|| ||f<q$qt | ||d\}}td}t | ||}||ksJ| d krt t| |j|}||ksJt t| ||j}||ksJ| d kr(t t| |j|}||ksJt t| ||j}||ks(Jq(dS)Nr=irr Fz0.0rrfullrcomplexTreal)Zedpsz1.0E-11)rSrTrYxrangeboolZnintZrandintrJrIrCZmpcZqrrZr9T conjugate)ZlowlimitZuplimitZmaxmZflgZzerokZexdpsmodeZnum1Znum2rrDr.Zdtyper?rvalZval2QRZmaxnormZn1r r r!test_qrsV   r)2 __future__rrrZmpmathZlibmpbackendrrSr(r)r*rWrrCr4b1r5b2r%r'r+r,r6rergrhrirUrVrxryrjrkr3r<r_rcrlrmrvrzr}rZextradpsrrrrrrr r r r!s 7