bggDdZddlmZddlZddlmcmZddl m Z ddl m Z m Z mZmZejdgdz Zejddgdz Zejgdd z Zejgd d z Zejgd d z Zejgddz Zejgddz ZeeeeeeegZdZGddZGddZGddZGddZGddZGddZGdd Z Gd!d"Z!Gd#d$Z"Gd%d&Z#dS)'zTests for laguerre module. )reduceNpolyval)assert_almost_equal assert_raises assert_equalassert_)r r )i r r)iHir r)xiiXi8r r)i iiiir rc.tj|dS)Ngư>)tol)laglagtrimxs w/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib64/python3.11/site-packages/numpy/polynomial/tests/test_laguerre.pytrimrs ;qd # # ##c&eZdZdZdZdZdZdS) TestConstantsc>ttjddgdS)Nrr )rr lagdomainselfs rtest_lagdomainzTestConstants.test_lagdomainsS]QF+++++rc<ttjdgdS)Nr)rrlagzeror"s r test_lagzerozTestConstants.test_lagzero!sS[1#&&&&&rc<ttjdgdSNr )rrlagoner"s r test_lagonezTestConstants.test_lagone$sSZ!%%%%%rc>ttjddgdS)Nr r )rrlagxr"s r test_lagxzTestConstants.test_lagx'sSX2w'''''rN)__name__ __module__ __qualname__r$r'r+r.rrrrsP,,,'''&&&(((((rrcVeZdZejdddZdZdZdZdZ dZ d Z d S) TestArithmeticdctdD]}tdD]}d|d|}tjt||dz}||xxdz cc<||xxdz cc<t jdg|zdgzdg|zdgz}t t|t||dSNAt i=, j=r rerr_msg)rangenpzerosmaxrlagaddrrr#ijmsgtgtress r test_lagaddzTestArithmetic.test_lagadd.q @ @A1XX @ @(a((Q((hs1ayy1}--A! A! j!Q!qc!eqck::T#YYS 3?????  @ @ @rctdD]}tdD]}d|d|}tjt||dz}||xxdz cc<||xxdzcc<t jdg|zdgzdg|zdgz}t t|t||dSr9)r?r@rArBrlagsubrrrDs r test_lagsubzTestArithmetic.test_lagsub8rKrcJttjdgdgttjdgddgtddD]H}dg|zdgz}dg|dz z| d|zdz|dz gz}t tj||IdS)Nrr r r:r )rrlagmulxr?r)r#rEserrHs r test_lagmulxzTestArithmetic.test_lagmulxBsS[!%%s+++S[!%%2w///q! 7 7A#a%1#+C#q1u+!QqS1WAh 77C  C 0 0# 6 6 6 6 7 7rctdD]}dg|zdgz}tj|j|}tdD]}d|d|}dg|zdgz}tj|j|}tj||}tj|j|} t t |||zdzk|t| ||z|dS)Nr:rr r;r<r=)r?rlagvalrlagmulr lenr) r#rEpol1val1rFrGpol2val2pol3val3s r test_lagmulzTestArithmetic.test_lagmulJsq B BA3q5A3;D:dfd++D1XX B B(a((Q((s1us{z$&$//z$--z$&$//D QUQY.444#D$t)SAAAAA B B BrctdD]}tdD]}d|d|}dg|zdgz}dg|zdgz}tj||}tj||\}}tjtj|||} t t | t ||dS)Nr:r;r<rr r=)r?rrClagdivrUrr) r#rErFrGcicjrHquoremrIs r test_lagdivzTestArithmetic.test_lagdivXsq G GA1XX G G(a((Q((SUaS[SUaS[jR((:c2..SjC!4!4c::#DIItCyy#FFFFF G G Grc rtdD]}tdD]}d|d|}tj|dz}ttj|g|ztjdg}t j||}tt|t||dS)Nr:r;r<r r=) r?r@arangerrrUarraylagpowrr)r#rErFrGcrHrIs r test_lagpowzTestArithmetic.test_lagpowcsq @ @A1XX @ @(a((Q((Ia!e$$SZ!Q! >>jA&&T#YYS 3?????  @ @ @rN) r/r0r1r@linspacerrJrNrRr]rdrjr2rrr4r4+s B3A@@@@@@777 B B B G G G@@@@@rr4ceZdZejgdZejdeeZejdeeeZej ddzdz Z e e gdZ dZ d Zd Zd Zd Zd S)TestEvaluation)g"@g,g@i,j->ij i,j,k->ijkr6r:r r )?g@g@cttjgdgjdt jddfdt D}tdD]>}d|}||}tjdg|zdgz}t|||?tdD]}d g|z}t j |ttjdgj |ttjddgj |ttjgd j |dS) Nr rr c0g|]}t|Sr2r).0rirs r z.TestEvaluation.test_lagval..}s! * * *qWQ]] * * *rr;r=r6r )r rr) rrrTsizer@rkLlistr?rrAshape)r#yrErGrHrIdimsrs @r test_lagvalzTestEvaluation.test_lagvalwsQSZQC((-q111 KA   * * * *E * * *q 7 7A!++CA$C*QA ,,C S# 6 6 6 6 6q ? ?A3q5DA As++14 8 8 8 A1v..4d ; ; ; Ayyy117 > > > >  ? ?rc~|j\}}}|j\}}}tttj||dd|j||z}t j|||j}t||tj d} t j| | |j}t|j dkdSNr r r6) rrzr ValueErrorrlagval2dc2drr@onesr ry r#x1x2x3y1y2y3rHrIzs r test_lagval2dzTestEvaluation.test_lagval2dsV BV B j#,BrrFDHEEEel2r48,,C%%% GFOOl1a** V#$$$$$rc |j\}}}|j\}}}tttj|||dd|j||z|z}t j||||j}t||tj d} t j| | | |j}t|j dkdSr~) rrzrrrlagval3dc3drr@rr ryrs r test_lagval3dzTestEvaluation.test_lagval3dsV BV B j#,B2A2IIIeBhl2r2tx00C%%% GFOOl1aDH-- V#$$$$$rcB|j\}}}|j\}}}tjd||}t j|||j}t||tjd} t j| | |j}t|j dkdS)Nrnr)r r6r r6) rrzr@einsumr laggrid2drrrr ryrs rtest_laggrid2dzTestEvaluation.test_laggrid2dsV BV Bi 2r**mBDH--C%%% GFOOmAq$(++ X%&&&&&rcH|j\}}}|j\}}}tjd|||}t j||||j}t||tjd} t j| | | |j}t|j dkdS)Nror)r r6r r6r r6) rrzr@rr laggrid3drrrr ryrs rtest_laggrid3dzTestEvaluation.test_laggrid3dsV BV Bi b"b11mBB11C%%% GFOOmAq!TX.. X%&&&&&rN)r/r0r1r@rgc1drrrrandomrrrzr|rrrrr2rrrmrmms "(>>> " "C ")IsC ( (C ")L#sC 0 0C   "Q&A<<<  A???*%%%"%%%" ' ' ' ' ' ' ' 'rrmceZdZdZdZdS) TestIntegralc T tttjdgdtttjdgdtttjdgdddgtttjdgdgtttjdgdgtttjdgdt dd D]9}dg|dz zdgz}tjdg|| }t |ddg:t d D]}|dz}dg|zdgz}|gdg|zzd|z gz}tj|}tj|d|g }tj|}t t|t|t d D]a}|dz}dg|zdgz}tj|}tj|d|gd }t tj d||bt d D]}|dz}dg|zdgz}|gdg|zzd|z gz}tj|}tj|d|gd }tj|}t t|t|t d D]}t dd D]~} dg|zdgz}|dd}t | D]}tj|d }tj|| }t t|t|t d D]}t dd D]} dg|zdgz}|dd}t | D]}tj|d|g }tj|| tt |  }t t|t|t d D]}t dd D]} dg|zdgz}|dd}t | D]}tj|d|gd }tj|| tt | d }t t|t|t d D]}t dd D]} dg|zdgz}|dd}t | D]}tj|d|gd }tj|| tt | d }t t|t|dS)Nr?r r )lbnd)sclaxisr r:)mk)rrr)rrrr) r TypeErrorrlagintrr?rpoly2laglag2polyrrTlist) r#rErrIrpolrHlagpolrrFs r test_lagintzTestIntegral.test_lagintsiaS"555j#*qc2666j#*qc1q!f===j#*qc<<<<j#*qcs;;;;iaSr::::q! . .AQU qc!A*aSA+++C aW - - - -q 6 6Aa%C#a%1#+C#A+3'C\#&&FZ!s333F,v&&C S 499 5 5 5 5q ; ;Aa%C#a%1#+C\#&&FZ!s<<>>CCjT%((^^"EEE#DIItCyy9999  :q : :A1a[[ : :c!eqck!!!fq==A*SA!!<<.$ 6661A666rrrc6g|]}tj|Sr2rrs rruz1TestIntegral.test_lagint_axis..( 4441A444rr c:g|]}tj|dS)r6)rrrs rruz1TestIntegral.test_lagint_axis..,s'999A+++999rr6)rr)r@rvstackTrrrr#rrHrIs rtest_lagint_axiszTestIntegral.test_lagint_axis siv&&i66666779j1%%%C%%%i4444455j1%%%C%%%i99S999::j***C%%%%%rN)r/r0r1rrr2rrrrs6Q:Q:Q:f&&&&&rrceZdZdZdZdS)TestDerivativec Vtttjdgdtttjdgdt dD]L}dg|zdgz}tj|d}t t|t|Mt dD]s}t ddD]`}dg|zdgz}tjtj|||}tt|t|att dD]u}t ddD]b}dg|zdgz}tjtj||d|d}tt|t|cvdS) Nrrr r:r rr )rr) rrrlagderrr?rrrr)r#rErHrIrFs r test_lagderzTestDerivative.test_lagder3siaS"555j#*qc2666q / /A#a%1#+C*SA&&&C cDII . . . .q : :A1a[[ : :c!eqckjC1!5!5!5;;;#DIItCyy9999 : q : :A1a[[ : :c!eqckjC1!!.Prrrrc6g|]}tj|Sr2rrs rruz3TestDerivative.test_lagder_axis..Trrr )r@rrrrrrrs rtest_lagder_axiszTestDerivative.test_lagder_axisLsiv&&i66666779j1%%%C%%%i4444455j1%%%C%%%%%rN)r/r0r1rrr2rrrr1s2:::2 & & & & &rrc`eZdZejddzdz ZdZdZdZdS) TestVanderrpr r c8tjd}tj|d}t |jdkt dD]7}dg|zdgz}t|d|ftj||8tj ddgddgdd gg}tj|d}t |jd kt dD]7}dg|zdgz}t|d|ftj||8dS) Nr6rrrr .r r:r)r6r r) r@rfr lagvanderr ryr?rrTrg)r#rvrEcoefs rtest_lagvanderzTestVander.test_lagvander]s) IaLL M!Q  6!"""q @ @A3q5A3;D #q& 3:a+>+> ? ? ? ? Hq!fq!fq!f- . . M!Q  9$%%%q @ @A3q5A3;D #q& 3:a+>+> ? ? ? ? @ @rcn|j\}}}tjd}tj||ddg}tj|||}tj||j}t||tj|g|gddg}t|j dkdS)Nrr r )r r:r) rr@rr lagvander2drdotflatrr ryr#rrrrivanrHrIs rtest_lagvander2dzTestVander.test_lagvander2dnsV B I  V $ $ob"q!f--l2r1%%fS!&!!C%%%ordRD1a&11 Y&'''''rcv|j\}}}tjd}tj|||gd}tj||||}tj||j}t||tj|g|g|ggd}t|j dkdS)N)r r6r)r r r6)r r:r) rr@rr lagvander3drrrrr ryrs rtest_lagvander3dzTestVander.test_lagvander3d{sV B I  Y ' 'ob"b)))44l2r2q))fS!&!!C%%%ordRD2$ :: Z'(((((rN) r/r0r1r@rrrrrr2rrrrYsc   "Q&A@@@" ( ( ( ) ) ) ) )rrceZdZdZdS) TestFittingc  d}tttjdgdgdtttjdggdgdtttjgdgdtttjdgdgggdtttjddgdgdtttjdgddgdtttjdgdgddggtttjdgdgdddgtttjdgdgdgtttjdgdggdtttjdgdggt jdd}||}tj||d}tt|d ttj |||tj||gd }tt|d ttj |||tj||d }tt|d ttj |||tj||gd }tt|d ttj |||tj|t j ||gj d}t|t j ||gj tj|t j ||gj gd }t|t j ||gj t j |}|}d|ddd<d|ddd<tj||d|} t| |tj||gd |} t| |tj|t j ||gj d|} t| t j ||gj tj|t j ||gj gd |} t| t j ||gj gd }ttj||dddgttj||ddgddgdS)Nc||dz z|dz zS)Nr r r2rs rfz"TestFitting.test_lagfit..fsa!e9a!e$ $rr r rr )w)r r rr6r)rr r r6r:)rr r r6r)r y?r y)rrrlagfitrr@rkrrVrrTrgr zeros_likecopy) r#rrrzcoef3coef4coef2drywwcoef3wcoef2ds r test_lagfitzTestFitting.test_lagfits % % % j#*qcA3;;;iqcUQC;;;iR!a888iaSQC5'1===iaVaS!<<<iaS1a&!<<<iaS1#qaSEBBBBiaS1#qQFCCCCj#*qcA3>>>j#*qcA3 CCCiaS1#r::: K1   AaDD 1a##SZZ###CJq%00!444 1a..SZZ###CJq%00!444 1a##SZZ###CJq%00!444 1a11SZZ###CJq%00!444ArxA//1155FBHeU^$<$<$>???ArxA//1<<<@@FBHeU^$<$<$>??? M!   VVXX!$Q$!$Q$Ar1***FE***Ar<<<1555FE****Q"b 2 2 4a1===GRXuen%=%=%?@@@*Q"b 2 2 4lllaHHHGRXuen%=%=%?@@@   CJq!Q//!R999CJq!aV44q"g>>>>>rN)r/r0r1rr2rrrrs#9?9?9?9?9?rrc eZdZdZdZdZdS) TestCompanionctttjgtttjdgdSr))rrr lagcompanionr"s r test_raiseszTestCompanion.test_raisess4j#"2B777j#"2QC88888rctddD]8}dg|zdgz}ttj|j||fk9dS)Nr r:r)r?r rrry)r#rErs rtest_dimensionszTestCompanion.test_dimensionss\q! < r&s/'''''''''//////RXqc]]1_ RXq"gq RXjjj! RXoooq  RX###$$R' RX,,,--c1 RX44455c9 RRR$$$$ ( ( ( ( ( ( ( (?@?@?@?@?@?@?@?@D['['['['['['['['|c&c&c&c&c&c&c&c&L%&%&%&%&%&%&%&%&P-)-)-)-)-)-)-)-)`;?;?;?;?;?;?;?;?| 7 7 7 7 7 7 7 7********&0&0&0&0&0&0&0&0&0&0&r