g<ddlZddlZddlZddlmZddlmZddlZddl m Z m Z ddl m Z ddlmcmZddlmcmZddlZddlZejdkZdZGdd ZGd d ZGd d ZdS)N)suppress_warnings)raises)xp_assert_closeassert_almost_equal)check_free_memorylctjtjtjt d|S)Ndata)ospathjoinabspathdirname__file__)basenames [/granary/edev/quantum/lib/python3.11/site-packages/scipy/interpolate/tests/test_interpnd.py data_filers= 7<<(A(ABB * **ceZdZdZdZdZdZdZdZdZ dZ d Z d Z e jje jje jed d ZdS)TestLinearNDInterpolationctjgdtj}tj|jdtj}t j|||}t||dS)Nrr)r)r?)rr)?333333?dtypernparrayfloat64arangeshapeinterpndLinearNDInterpolatorrselfxyyis rtest_smoketestz(TestLinearNDInterpolation.test_smoketestst HNNN: ' ' ' Iagaj 3 3 3 0X *1a 0 0 3 3Ar"""""rcRtjgdtj}tj|jdtj}t j|dddf|dddff||dddf|dddf}t||dS)Nrrrrr's rtest_smoketest_alternatez2TestLinearNDInterpolation.test_smoketest_alternate#s HNNN: ' ' ' Iagaj 3 3 3 ?X *AaaacFAaaacF+;Q ? ?!!!A#!!!A# O OAr"""""rc tjgdtj}tj|jdtj}|d|zz }t j|||}t||dSNrrr@rr's rtest_complex_smoketestz0TestLinearNDInterpolation.test_complex_smoketest,s HNNN: ' ' ' Iagaj 3 3 3 1H 0X *1a 0 0 3 3Ar"""""rcNtjgdtj}tj|jdtj}|d|zz }t j|}tj||}||}t|||j |usJdSr1) r r!r"r#r$qhullDelaunayr%r&rtri)r(r)r*r7 interpolatorr+s rtest_tri_inputz(TestLinearNDInterpolation.test_tri_input6s HNNN: ' ' ' Iagaj 3 3 3 1HnQ4S!<< \!__Ar"""3&&&&&&rc\tjgdtj}tjgdtjfd}tjtjddddddftjddddddf\}}|}|}tj||gj}tj ||}t||||dS)N)rrr.r.r.r.rrg?g@gg@c&||zdk}|}||}||}||}||}d|z} dd|z |z z d|zz d|zz||< d||zdz z dd|z zz dd|z zz||<|S)Nr.r) r)r*t1t2x1y1x2y2zvaluess ripz1TestLinearNDInterpolation.test_square..ipQsa%1*BB2B2B2B2B!AAYB ,ay|$ay|$AbEAYR! ,ay!b&)*ay!b&)*AbEHrrr. r r!r"broadcast_arrayslinspaceravelTcopyr%r&r)r(pointsrKxxyyxizirJs @r test_squarez%TestLinearNDInterpolation.test_squareCs%666bjIII+++2:>>>     *$R[Ar%:%:111T6%B%'[Ar%:%:46%BDDB XXZZ XXZZ Xr2h   ! & & ( ( :X *66 : :2 > >B2r +++++rctjgdtj}tj|jdtj}t j||d|}t||dS)Nr)r[)r[)r\r\)g@r@rrTrescalerr's rtest_smoketest_rescalez0TestLinearNDInterpolation.test_smoketest_rescalepsy HBBB: ' ' ' Iagaj 3 3 3 >X *1a > > >q A AAr"""""rc|tjgdtj}tjgdtj}tjtjddddddftjddddddf\}}|}|}tj||gj}tj |||}tj ||d |}t||dS) N)r)rd) ra)rbrrr>rrbrLraTr]rM)r(rSrJrTrUrVrW zi_rescaleds rtest_square_rescalez-TestLinearNDInterpolation.test_square_rescaleys0<<>>$R[B%;%;AAAdF%C%'[C%<% >h3FF ""  B ,,,,,rctjgdtj}tj|jdtj}|d|zz }t j|}tj|j ||}tj|j |d|}t||dSNrZrrr2Tr]) r r!r"r#r$r5r6r%r&rSrr(r)r*r7r+ yi_rescales rtest_tripoints_input_rescalez6TestLinearNDInterpolation.test_tripoints_input_rescales H???: ' ' ' Iagaj 3 3 3 1HnQ 9X *3:q 9 9! < <X23:q!! B +++++rctjgdtj}tj|jdtj}|d|zz }t j|}d}tjt|5tj ||d|ddddS#1swxYwYdS NrZrrr2zORescaling is not supported when passing a Delaunay triangulation as ``points``.)matchTr]) r r!r"r#r$r5r6pytestr ValueErrorr%r&r(r)r*r7rls rtest_tri_input_rescalez0TestLinearNDInterpolation.test_tri_input_rescales H???: ' ' ' Iagaj 3 3 3 1HnQ9 ]:U 3 3 3 C C ?H )#q$ ? ? ? B B B C C C C C C C C C C C C C C C C C C!B55B9<B9ctjdtjdd}tjddtjdzz}t j||}t jt j|}t|dd|dddSNrAy?r) r randomseedrandr%r&pickleloadsdumpsr)r(r)r*rKip2s r test_picklez%TestLinearNDInterpolation.test_pickles t INN2q ! ! INN2  BINN2$6$6!6 6  *1a 0 0l6<++,,BBsCLL##c3--88888rzit fails on 32-bit)reasonc ptdtjddd}tjddd}tj||\}}d}tj||fj}||z}tj||}tj ||tddtddf} tj ||tddtddf} tj ||tddtddf} tj ||tddtddf} | | | | | | | | dS)Ni'rihrbcNtj||f\}}|||f}|SN)r mgrid)r8 slice_rows slice_colsgrid_xgrid_yress r do_interpz;TestLinearNDInterpolation.test_threading..do_interps0Xj*&<=NFF,/00CJri4)targetargs)rr r#meshgridvstackrPrQr%r& threadingThreadslicestartr ) r(r_ticks phi_ticksr_gridphi_gridrrSrJr8worker_thread_1worker_thread_2worker_thread_3worker_thread_4s rtest_threadingz(TestLinearNDInterpolation.test_threadings2 %   )AtR((Iar** ;w ::    FLLNNHNN,<,<=>>@8#**,,4VVDD #*aa?AAA$*dD 1 15D>>BDDD$*adD0A0ABDDD$*dD 1 15t3D3DEGGG rN)__name__ __module__ __qualname__r,r/r3r9rXr_rdrirpr}rmmarkslow thread_unsafeskipif _IS_32BITrrBrrrrs######### ' ' '+,+,+,Z###---$ , , , C C C 9 9 9 [ [ [ *>??''@?'''rrceZdZdZdZdS)TestEstimateGradients2DGlobalc tjgdt}tj|}ddfddfddfd d fg}t |D]\}\}}||ddd f|ddd f}t j||d }|jdksJt|tj|dddfd |zzddd|zdS)N)r)rrAr=)r.rArg?g333333?g?rcd|zdzSNrr.rBr)r*s rz>TestEstimateGradients2DGlobal.test_smoketest..s!A#'rrc d|zSNrrBrs rrz>TestEstimateGradients2DGlobal.test_smoketest..s !a%rr=c d|zSNrBrs rrz>TestEstimateGradients2DGlobal.test_smoketest..s "q&rr;cdd|zzd|zzSNr@L,@rBrs rrz>TestEstimateGradients2DGlobal.test_smoketest..s!ac'E!G+r)r@rrr.ư>)tol)rAgh㈵>zitem %d)rtolatolerr_msg) r r!floatr5r6 enumerater%estimate_gradients_2d_globalr$r) r(r)r7funcsjfuncgradrIdzs rr,z,TestEstimateGradients2DGlobal.test_smoketestsD H@@@GL N N NnQ " !6 *   ( & ) + +Z 8   )// I IOA|dQqqqsVQqqqsV$$A6sA4HHHB8v%%%% BtAAAv 62 =!%D)a- I I I I I  I Irctjtd}tj|jd}t j|}t5}| tj dtj ||dddddS#1swxYwYdS)Nzestimate_gradients_hang.npyrz$Gradient estimation did not converger.)maxiter) r loadrrvrxr$r5r6rfilterr%GradientEstimationWarningr)r(rSrJr7sups rtest_regression_2359z2TestEstimateGradients2DGlobal.test_regression_2359s#@AABB Q00nV$$  JC JJx9= ? ? ?  1#vq I I I I J J J J J J J J J J J J J J J J J Js.8B33B7:B7N)rrrr,rrBrrrrs:III, J J J J JrrceZdZ ddZdZdZdZdZd Ze j d d Z d Z d ZdZdS)TestCloughTocher2DInterpolatorNrFc tjd}|tjgdt}|s7t j|||dddf|dddfd|}nLt j|dddf|dddff||dddf|dddfd|}|dd } |s || } n || dddf| dddf} || dddf| dddf} t| | fi|dS#t$r7td t| | z td |j wxYw) Nrt)rr;r=r<rr)rg?rrr.r)rr^2rAz_check_accuracy: abs(a-b):zip.grad:) r rv RandomStater!rr%CloughTocher2DInterpolatorrxrAssertionErrorprintabsr) r(rr)r alternater^kwrngrKpabs r_check_accuracyz.TestCloughTocher2DInterpolator._check_accuracysi##D)) 9&&& %&&&A  P4QQqqqsVQqqqsV8L8L9=wPPPBB4a!fa!f5E59T!AAAaC&!AAAaC&5I5I9=wPPPB HHROO #1AA1QQQqS61QQQqS6""A D111Q3111Q3   Aq ' 'B ' ' ' ' '    .AE ; ; ; *bg & & &  s )D99AE:c 4ddddg}t|D]}\}}||dddd|z||dddd d |z ||dddd |zd ||dddd d d|z~dS)Ncd|zdzSrrBrs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..(s1qrc d|zSrrBrs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..)s Qrc d|zSrrBrs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..*s arcdd|zzd|zzSrrBrs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..+sQqS57*rgvIh%<=gHz> Function %drrrrTzFunction (alternate) %d)rrrrrzFunction (rescaled) %drrrrr^z!Function (alternate, rescaled) %d)rrrrr^rrrr(rrrs rtest_linear_smoketestz4TestCloughTocher2DInterpolator.test_linear_smoketest%s !     * *  !'' R RGAt  5t$)6): ! < < <  5t$+/)BQ)F ! H H H  5t$)AA)Et ! U U U  5t$+/)Lq)P ! R R R R R Rrc ddddg}t|D]@\}}||dddd|z ||dddd|zd AdS) Nc |dzSNrArBrs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..> Arc |dzSrrBrs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..?rrc|dz|dzz SrrBrs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..@A1rc ||zSrrBrs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..A 1r& .>g)\(?rrrTrrrs rtest_quadratic_smoketestz7TestCloughTocher2DInterpolator.test_quadratic_smoketest;s     $ $    !'' J JGAt  4d)6): ! < < <  4d)6):D ! J J J J J Jrc4tjgdtj}tj|jdtj}|d|zz }t j|}tj|||}t||dSr1) r r!r"r#r$r5r6r%rr)r(r)r*r7r+s rr9z-TestCloughTocher2DInterpolator.test_tri_inputJs HNNN: ' ' ' Iagaj 3 3 3 1HnQ 8X 0a 8 8 ; ;Ar"""""rctjgdtj}tj|jdtj}|d|zz }t j|}d}tjt|5tj ||d|ddddS#1swxYwYdSrk) r r!r"r#r$r5r6rmrrnr%rros rrpz5TestCloughTocher2DInterpolator.test_tri_input_rescaleUs H???: ' ' ' Iagaj 3 3 3 1HnQ9 ]:U 3 3 3 I I EH /Q E E Ea H H H I I I I I I I I I I I I I I I I I Irqctjgdtj}tj|jdtj}|d|zz }t j|}tj|j ||}tj|j |d|}t||dSrf) r r!r"r#r$r5r6r%rrSrrgs rriz;TestCloughTocher2DInterpolator.test_tripoints_input_rescalebs H???: ' ' ' Iagaj 3 3 3 1HnQ ?X 0Q ? ? B BUX8QPTUUUVWXX B +++++rr\c dddddg}tjd}tjtjgdt |d d f}t|D]B\}}|||d d d d|z|||d d d d|zdCdS)Nc |dzSrrBrs rrz;TestCloughTocher2DInterpolator.test_dense..rrrc |dzSrrBrs rrz;TestCloughTocher2DInterpolator.test_dense..srrc|dz|dzz SrrBrs rrz;TestCloughTocher2DInterpolator.test_dense..trrc ||zSrrBrs rrz;TestCloughTocher2DInterpolator.test_dense..urrctjdtjz|ztjdtjz|zzSr)r cospisinrs rrz;TestCloughTocher2DInterpolator.test_dense..vs3"% **26!BE'!)+<+<<ri)rr;r=r<rirArg{Gzt?g{Gz?r)r)rrrrT)r)rrrrr^) r rvrr_r!rrxrr)r(rrgridrrs r test_densez)TestCloughTocher2DInterpolator.test_densens     $ $   < <  i##D))uRX:::%HHHXXeQ''()!'' J JGAt  4d)6): ! < < <  4d)6):D ! J J J J J Jrctjdd}tjd}ttt j||dS)Nrur@)r rvrandn assert_raisesrnr%r)r(r)r*s rtest_wrong_ndimz.TestCloughTocher2DInterpolator.test_wrong_ndimsG IOOB " " IOOB  j("Eq!LLLLLrctjd}|dd}|dd|dzz}t j||}t jt j|}t|dd|dddSrs) r rvrrxr%rryrzr{r)r(rr)r*rKr|s rr}z*TestCloughTocher2DInterpolator.test_picklesi##D)) HHROO HHRLL2chhrll? *  0A 6 6l6<++,,BBsCLL##c3--88888rcRtjdddtjddz fg}tjgd}tj||}d|jd<d }tjdtj|zdtj|zg}tjdtjtjdz |z zdtjtjdz |z zg}||}||}t||tj d } | dd} | d} | |jj| dddfz}| || z}| || z}tj||}d|jd<||} ||} t| |t| |dS) Nrr=rr@rA)r.rrr.rr.)r r!sqrtr%rrrrrrrvrrdotrQ)r(rSrJrKalphap1p2v1v2rArw1w2s rtest_boundary_tri_symmetryz9TestCloughTocher2DInterpolator.test_boundary_tri_symmetrys 66CA+>?@@)))$$  0 @ @  XsRVE]]*C"&--,?@ A A XsRVBE!GeO444cBF257U?rs~ ++++++****** FFFFFFFF333333.........$$$$$$$$$  [5 *** AAAAAAAAH"J"J"J"J"J"J"J"JJv v v v v v v v v v r