o ?HhXn@s<ddlZddlZddlmZmZmZddlmZddl m Z ddl m mZddlmZmZmZmZmZmZmZmZmZddlmZddlmZmZmZddlm Z m!Z!ddl"m#Z#ej$j%Z%Gd d d Z&Gd d d Z'Gd ddZ(e GdddZ)GdddZ*dddddddZ+GdddZ,GdddZ-dS)N) assert_equalassert_almost_equalassert_allclose)given) romb newton_cotescumulative_trapezoid trapezoidquadsimpson fixed_quadqmc_quadcumulative_simpson)%_cumulative_simpson_unequal_intervals)statsspecial integrate)array_api_compatibleskip_xp_invalid_arg)xp_assert_closec@seZdZddZddZdS) TestFixedQuadcs>ddd}tfddddd\}}t||dd dS) Ncs|ddS)Nrrxnre/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/scipy/integrate/tests/test_quadrature.pysz+TestFixedQuad.test_scalar..rr-q=rtol)r r)selfexpectedgot_rrr test_scalars zTestFixedQuad.test_scalarcsNd}tdd|dd}tfdddd|d\}}t||dd dS) Nrrrcs|dddfSNrrprrr "z+TestFixedQuad.test_vector..rrr!r")nparanger r)r$rr%r&r'rr*r test_vectors  zTestFixedQuad.test_vectorN)__name__ __module__ __qualname__r(r/rrrrrs rc@sVeZdZddZddZddZddZd d Zd d Ze j d ddgddZ dS)TestQuadraturecCstr))NotImplementedError)r$rabargsrrrr 'szTestQuadrature.quadcCstttdddS)N)rrr-r.r$rrr test_romb*szTestQuadrature.test_rombcCsPtd}td|}t|}tdd||\}}t||ddddS)Nr8皙?cSstd|S)Nr<)r-cosrrrrr 2z2TestQuadrature.test_romb_gh_3731..g:0yE>rr#atol)r-r.r=rr minmaxr)r$ryvalval2errrrrtest_romb_gh_3731-s z TestQuadrature.test_romb_gh_3731cCsd}t|d\}}t||tddgt||d dd}t|d\}}t||tgddt||d d d}t|d\}}t||tgd d t||d d d }t|d\}}t||tgddt||d ddS)z5Test the first few degrees, for evenly spaced points.r?(@r)?@rK@g@)rK@rOrK @gP@r)@@@rJrRrQgV@g=AN)rrr-arrayr)r$rwtserrcoffrrrtest_newton_cotes5s z TestQuadrature.test_newton_cotescCs|tgd}|d}t|\}}d}t||}t||tgd}|d}t|\}}d}t||}t||dS)z9Test newton_cotes with points that are not evenly spaced.)g?@rgUUUUUU@)rXgffffff?g@rO"@N)r-rTrdotr)r$rrCrUrVexact_integralnumeric_integralrrrtest_newton_cotes2Ks     z!TestQuadrature.test_newton_cotes2c Cstd}tt|dtt|dddtt|tddddd td dd}d d }tt|||dd td dd}tt||dddtddd}tdddd}||}tdD]C}t|||d}tj|dgd}|D]/}t |j } | |t d|t | ddd|t | ddd} t||j | qyqftddd}||}t||dd}d|ddd f|dddf|ddd f|dddf} t|| tddd}tdddd}||}tdD]E}t|||d}tj|dgd}|D]0}t |j } | |t d|t | ddd|t | ddd} t||j | qqtdg}t|d}tt||dddtt||dddtgd}t|d}tt||dddtt||dddtgdgdgdg}t|d}gd } d!gd} tt||dd| tt||dd| tgdgdgd"g}t|d}gd#} gd$} tt||dd| tt||dd| dS)%Nr8r9rHdx@rrr rcSs|dS)Nrrrrrrfdsz&TestQuadrature.test_simpson..f5@rSrYrP@rIraxis multi_index)flagsrrN}rX)rIrIrIrI)rrrrm)rXrXrXrX竪Je@)rrmrfrb)rXga@g@g@)rprpg*U@)r-r.rr linspacerreshaperangenditerlistrjinsertslicetuplerTpower) r$rCrrcr5iritr'idxintegral zero_axis default_axisrrr test_simpson\st   0D   0      zTestQuadrature.test_simpsondroplastFTcCs`tgdgdg}|r|ddddf}t|dd}ttj|tjddd}t||dS)N)rrrrrmrmrN)rrrr rrlridtype)r-rTr float64r)r$rrCresultr%rrrtest_simpson_2d_integer_no_xs  z+TestQuadrature.test_simpson_2d_integer_no_xN) r0r1r2r r;rGrWr^rpytestmark parametrizerrrrrr3&sNr3c@sNeZdZddZddZddZddZej d d d gd d Z ddZ dS)TestCumulative_trapezoidcCsXtjdddd}|}t||dd}gd}t||t||dd}t||dddS) NrrrNnumrinitial)rXrrXrr-rqrr)r$rrCy_int y_expectedrrrtest_1ds z TestCumulative_trapezoid.test_1dcCstdddd}|}t||dd}tgdgdggd gd ggd gd gg}t||gd }tgd|D] \}}t||d|d}t|jdt||d|d}t|j|q=dS)NrIrrrr)rXrHrY@)rXr$@0@)rX!@2@g<@)rX)@g:@@D@)rXrgA@g@J@)rX4@gE@g P@))rrr)rIrr)rIrrIrrr)rri)rIrr) r-r.rrrrTrziprshape)r$rrCrrshapesrirrrrtest_y_nd_x_nds*   z'TestCumulative_trapezoid.test_y_nd_x_ndc Cstdddd}tdd}tgdgdggdgdggtgd ggd ggd ggtgd gd ggdgdggdgdggf}tgd|D]\}}t||d|j||dd}t||qUdS)NrrIrr)rL@rMrQ)rPrZr&@)gD@gF@gH@gJ@)gL@gN@rggQ@)rYrOrLr)rrrJ*@)rg3@4@rd)rHrg1@)rrdgJ@)rgB@g`V@)rgJ@g`_@)rg@Q@g0d@)rg@U@gh@r)rrir)r-r.rrrTrrrr)r$rCr ys_expectedrirrrrrtest_y_nd_x_1ds<    z'TestCumulative_trapezoid.test_y_nd_x_1dcCstjdddd}t|}gd}t||t|dd}gd}t||t|d d }gd }t||t|d dd }gd }t||dS)NrrrNr)rrrrXrr)rrrrrXrIr_)rrX)r`r)rrrrrXr)r$rCrrrrr test_x_nones     z$TestCumulative_trapezoid.test_x_nonerrrHcCsPtjdddd}tjtddt||dWddS1s!wYdS)z4If initial is not None or 0, a ValueError is raised.rrrz `initial`matchrN)r-rqrraises ValueErrorr)r$rrCrrrtest_initial_errors"z+TestCumulative_trapezoid.test_initial_errorcCs>tjtddtgdWddS1swYdS)NAt least one point is requiredrrC)rrrrr:rrrtest_zero_len_y s "z(TestCumulative_trapezoid.test_zero_len_yN) r0r1r2rrrrrrrrrrrrrrs   rc@seZdZddZeddgdejdddZeddgdejdd d Z e d d Z ed dgdejdddZ dS) TestTrapezoidcCsR|ddd}t|d|d|d||jdd}t||ddS)Nrg?grr_rK)r.r expsqrtasarraypir)r$xprr{rrr test_simples0zTestTrapezoid.test_simplez jax.numpyz)JAX arrays do not support item assignment)reasonsskip_xp_backendsc Cs,|ddd}|ddd}|ddd}|||d|d}|dd<|dd<|||d|d}|dd<|dd<|||d|d}|dd<|dd<|ddddf|ddddf|ddddf}|j||ddddfdd} |j||ddddfdd} |j||ddddfdd} t||ddddfdd } t| | t||ddddfdd } t| | t||ddddfdd } t| | t||dd } t| | t||dd } t| | t||dd } t| | dS) NrrrIrrm rlrrh)rq ones_likesumr r) r$rrrCzwxwywzqqxqyqzr{rrr test_ndims86        zTestTrapezoid.test_ndimcCs(||dddd}||dddd}|ddd}tt||dd|tt||gd dd|tt||gd dddfdd|ttt||gd dddfddWdn1smwY|gd }tt||d d|tt||dd dd d|dS) Nr)rIri|rrh)rrr)rgP`@gk@r )rrrqrr rrr Exception)r$rrrCout0out1rrr test_gh21908@s$" &zTestTrapezoid.test_gh21908cCstd}||}|dk}tjj||d}d}tt|||tjj||d}tt|||tjj||d}tt|||dS)NrNr)maskr)r-r.marTrr )r$rrrCrymr{xmrrr test_masked[s zTestTrapezoid.test_maskedTz,array-likes only supported for NumPy backend)np_onlyrcCs\ttd}dd|D}|j||jd}|j||jd}t||}t||}t||dS)NrNcSsg|]}||qSrr).0trrr qr,z1TestTrapezoid.test_array_like..r)rursrrr r)r$rrrCxarryarrresresarrrrrtest_array_likels   zTestTrapezoid.test_array_likeN) r0r1r2rrrr usefixturesrrrrrrrrrrs&  %    rc@seZdZejjddZdddZejddd gejd dd gd d Z ejdddgddgddgddggddZ ejjejdddgddZ ddZ dS) TestQMCQuadcCsvd}tjt|dtdddgddgWdn1swYd}tjt|dtddddgddgWdn1sCwYd d }d }tjt|dt|ddgddgWdn1sjwYd }tjt|dtd dddgddgddWdn1swYd}tjt|dtddddgddgddWdn1swYd}tjt|dtddddgddgddWdn1swYd}tjt|dtddddgddgtj ddWdn 1s wYd}tjt|dtddddgddgddWddS1s4wYdS)Nz`func` must be callable.rza duckrrz/`func` must evaluate the integrand at points...cSdSNrrrrrrr z3TestQMCQuad.test_input_validation..cSs|jdksJt|Sr)ndimr-rrrrrfuncs z/TestQMCQuad.test_input_validation..funcz8Exception encountered when attempting vectorized call...z`n_points` must be an integer.cSrrrrrrrr rg@)n_pointsz!`n_estimates` must be an integer.cSrrrrrrrr rr) n_estimatesz8`qrng` must be an instance of scipy.stats.qmc.QMCEngine.cSrrrrrrrr r)qrngz8`qrng` must be initialized with dimensionality equal to cSrrrrrrrr rz,`log` must be boolean \(`True` or `False`\).cSrrrrrrrr rrlog) rr TypeErrorr rwarns UserWarningrqmcSobol)r$messagerrrrtest_input_validationzsB&$z!TestQMCQuad.test_input_validationrmNc sj|dur td}d}t|t|fddtjd}tjj||d}t|}t||}t |||||d} tj j ||d} t |dd | j} t| j| | d t|| jd ksiJtjd}tjj||d}t fd d ||||d|d} tt| j| jddt| jt|d krtjnd ksJtt| j| jddddS)Nrcstj|jSr))rmultivariate_normalpdfTr)covmeanrrrsz$TestQMCQuad.basic_test..funcl^^@E9?)seed)rrr) lower_limitrgףp= ?r@rcst|Sr))r-r)r7)rrrr r>z(TestQMCQuad.basic_test..T)rrrr+=r"缉ؗҜ.rrr)rrrr r~r-infr )r$rrrrrr test_zeros zTestQMCQuad.test_zerocCsFdd}t|dd}tjjdddtjjddd}t|j|ddS)NcSstjj|ddS)Nrscale)rnormrrrrrrsz-TestQMCQuad.test_flexible_input..funcrrrrg{Gz?)r rrr rr~)r$rrrrrrtest_flexible_inputs  zTestQMCQuad.test_flexible_input)rrmN) r0r1r2rr thread_unsaferrrrrrrrrrrrys  $ $   rrlrr`rric sd|j|dkr|durt||||ddS|t||||ddSt||d}t|dkr4t||dn|}t|dkrDt||dn|}t|dkrTt||dn|}|jd|durf|tn|}|du}|rpdn|}t||j}t||jddd}tj|||fdd}fdd }t|d|}|r|d ddfn|}t|d|}|S) NrI)rr`rirrrlr)rrcs.t|d|d|dddS)Nr)rr)r)rrrrrcs.z*cumulative_simpson_nd_reference..f.) rrr-moveaxisrr. broadcast_to concatenateapply_along_axis) rCrr`rriinitial_was_nonerrcrrrrcumulative_simpson_nd_references(      r#c@seZdZedZedZej ddej ddddZ ej ded d ej d d ej d dej ddej ddddZ ej dde gddfde egdgeegdfde eeegeeegddfde ggdfde ddfd e ed!dfd"e ed!d#fd$e d%ed!d&fgd'd(Z d)d*Zd+d,Zejjejjeejejejdd d-d.ejd/d-d0d1d2d3d4d5d6d7Zejjejjeejejejdd d-d.ejd/d-d0d1d8d3d4d5d9d:Zd%S);TestCumulativeSimpsonrruse_dx)FT use_initialcCsptjd}d}|r dnd}|}|dkrt||n t|||}t|dddtjf}||dddtjf} tj| ||dd} tj| ||d|ddd} |re| n | | ddd} |ru| dnd} |r}d|ind |i}t| fi|d | i}|st|| d d dS|rdnd}t|| d d t||dd| |ddd d dS)NF*BGrrIrrrrr`rrgV瞯|j|dn|}n|dkrRtj|j|d|d n t|j||d} |durcdn|j| d} t| | || |d } t| | || |d }tjj| |d d dS)Nr'rNrlrrIrsize)NNrrV瞯absrrrrr  zTestCumulativeSimpson.)elementsrcCstdd}t|dd}||}|j|t|jdd}tjj|ddd f|ddd f|ddd fd d d S) eTheoretically, the output of `cumulative_simpson` will be identical to `simpson` at all even indices and in the last index. The first index will not match as `simpson` uses the trapezoidal rule when there are only two data points. Odd indices after the first index are shown to match with a mathematically-derived correction.cs,tjfddtdjddDddS)Ncs$g|]}tdd|fddqS).NrKr_r rrzrrrrs$z|TestCumulativeSimpson.test_cumulative_simpson_against_simpson_with_default_dx..simpson_reference..rrlrrr-stackrsrrrrrsimpson_references"zhTestCumulativeSimpson.test_cumulative_simpson_against_simpson_with_default_dx..simpson_referencerKr_rlr.rNrr)rrJr-r.rr8r)r$rCrZrrrIrrr7test_cumulative_simpson_against_simpson_with_default_dx~s . zMTestCumulativeSimpson.test_cumulative_simpson_against_simpson_with_default_dxcCrOrPrQrrrrr rScCsd|jdd}tjdd|jdd}|ddd|tjddt|d|dd<dd }t||d }|||}|||}tj |d ddf|d ddf|d ddfdS) rUrrlrrrNr<cs.tjfddtdjddDddS)Ncs0g|]}tdd|fdd|fdqS).NrrVrWr=rrrs0zlTestCumulativeSimpson.test_cumulative_simpson_against_simpson..simpson_reference..rrlrrrX)rCrrr=rrZs"zXTestCumulativeSimpson.test_cumulative_simpson_against_simpson..simpson_referencer.) rr-rqruniformlenrrJr8r)r$rCintervalrrZrrrIrrr'test_cumulative_simpson_against_simpsons4  ,z=TestCumulativeSimpson.test_cumulative_simpson_against_simpson)r0r1r2r-r.r@r?rrrrr;r>rBrCrJrslowrhyp_numarraysr array_shapesstfloatsfilterr[r_rrrrr$sZ    !       r$c @sNeZdZddZddZejdee dddee d d d d d Z dS) TestLebedevcCsBd}tjt|dtdWddS1swYdS)NzOrder n=-1 not available...rrl)rrr4r lebedev_rule)r$rrrrrs "z!TestLebedev.test_input_validationcCsNdd}td\}}|||}d}t||ddtt|dtjdS)NcSst|dS)Nr)r-rrrrrrcsz&TestLebedev.test_quadrature..fg9-@rr"r)rrhrr-rr)r$rcrwrrrrrtest_quadratures  zTestLebedev.test_quadraturer*rIrbr#r3cCsvt|\}}tjdddtj|jdd}t|jdddWdn1s)wYtt |dtj dS)Nignore)divideinvalidrrr6rr) rrhr-errstaterdirectional_statsrrmean_resultant_lengthrr)r$r*rrjrrrrtest_propertiess zTestLebedev.test_propertiesN) r0r1r2rrkrrrrursrtrrrrrgs ( rg).rnumpyr- numpy.testingrrr hypothesisrhypothesis.strategies strategiesrdhypothesis.extra.numpyextrarascipy.integraterrrr r r r r rscipy.integrate._quadraturerscipyrrrscipy.conftestrrscipy._lib._array_api_no_0drrrrr3rrrr#r$rgrrrrs.  ,  Yhb%B