Ug4<ddlZddlZddlZddlmZmZmZmZddl m Z ddl Z ddl m Z ddlmcmZddlmcmZddlZddlZejdkZdZGddZGd d ZGd d ZdS) N) assert_equalassert_allcloseassert_almost_equalsuppress_warnings)raises)check_free_memorylctjtjtjt d|S)Ndata)ospathjoinabspathdirname__file__)basenames d/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/scipy/interpolate/tests/test_interpnd.py data_filers= 7<<(A(ABB * **ceZdZdZdZdZdZdZdZdZ dZ d Z d Z e 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)Nrrrr r(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@r r(s rtest_complex_smoketestz0TestLinearNDInterpolation.test_complex_smoketest,s HNNN: ' ' ' Iagaj 3 3 3 1H 0X *1a 0 0 3 3Ar"""""rc4tjgdtj}tj|jdtj}|d|zz }t j|}tj|||}t||dSr2) r!r"r#r$r%qhullDelaunayr&r'rr)r*r+trir,s rtest_tri_inputz(TestLinearNDInterpolation.test_tri_input6s HNNN: ' ' ' Iagaj 3 3 3 1HnQ 2X *3 2 21 5 5Ar"""""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..ipOsa%1*BB2B2B2B2B!AAYB ,ay|$ay|$AbEAYR! ,ay!b&)*ay!b&)*AbEHrrr/ r!r"r#broadcast_arrayslinspaceravelTcopyr&r'r)r)pointsrLxxyyxizirKs @r test_squarez%TestLinearNDInterpolation.test_squareAs%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@rArrTrescaler r(s rtest_smoketest_rescalez0TestLinearNDInterpolation.test_smoketest_rescalensy HBBB: ' ' ' Iagaj 3 3 3 >X *1a > > >q A AAr"""""rc|tjgdtj}tjgdtj}tjtjddddddftjddddddf\}}|}|}tj||gj}tj |||}tj ||d |}t||dS) N)r)rd) rb)rcrrr?rrcrMrbTr^rN)r)rTrKrUrVrWrX zi_rescaleds rtest_square_rescalez-TestLinearNDInterpolation.test_square_rescalews0<<>>$R[B%;%;AAAdF%C%'[C%<% >h3FF ""  B ,,,,,rctjgdtj}tj|jdtj}|d|zz }t j|}tj|j ||}tj|j |d|}t||dSNr[rrr3Tr^) r!r"r#r$r%r6r7r&r'rTrr)r*r+r9r, 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 Nr[rrr3zORescaling is not supported when passing a Delaunay triangulation as ``points``.)matchTr^) r!r"r#r$r%r6r7pytestr ValueErrorr&r'r)r*r+r9rms 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|dddSNrBy?r) r!randomseedrandr&r'pickleloadsdumpsrr)r*r+rLip2s 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'rihrccNtj||f\}}|||f}|SN)r!mgrid) interpolator slice_rows slice_colsgrid_xgrid_yress r do_interpz;TestLinearNDInterpolation.test_threading..do_interps0Xj*&<=NFF,/00CJri4)targetargs)rr!r$meshgridvstackrQrRr&r' threadingThreadslicestartr ) r)r_ticks phi_ticksr_gridphi_gridrrTrKrworker_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-r0r4r:rYr`rerjrqrrnmarkslowskipif _IS_32BITrrCrrrrs######### # # #+,+,+,Z###---$ , , , C C C 9 9 9 [ [ *>??''@?'''rrceZdZdZdZdS)TestEstimateGradients2DGlobalc tjgdt}tj|}ddfddfddfd d fg}t |D]\}\}}||ddd f|ddd f}t j||d }t|j dt|tj|dddfd |zzddd|zdS)N)r)rrBr>)r/rBrg?g333333?g?rcd|zdzSNrr/rCr*r+s rz>TestEstimateGradients2DGlobal.test_smoketest..s!A#'rrc d|zSNrrCrs rrz>TestEstimateGradients2DGlobal.test_smoketest..s !a%rr>c d|zSNrCrs rrz>TestEstimateGradients2DGlobal.test_smoketest..s "q&rr<cdd|zzd|zzSNrAL,@rCrs rrz>TestEstimateGradients2DGlobal.test_smoketest..s!ac'E!G+r)rArrr/ư>)tol)rBgh㈵>zitem %d)rtolatolerr_msg) r!r"floatr6r7 enumerater&estimate_gradients_2d_globalrr%r) r)r*r9funcsjfuncgradrJdzs rr-z,TestEstimateGradients2DGlobal.test_smoketestsC H@@@GL N N NnQ " !6 *   ( & ) + +Z 8   )// I IOA|dQqqqsVQqqqsV$$A6sA4HHHB 6 * * * 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!loadrrwryr%r6r7rfilterr&GradientEstimationWarningr)r)rTrKr9sups 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-rrCrrrrs: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|}tjdd }|s ||} n ||dddf|dddf} ||dddf|dddf} t| | fi|dS#t$r7td t| | z td |j wxYw) Nru)rr<r>r=rr)rg?rrr/r)rr_2rBz_check_accuracy: abs(a-b):zip.grad:) r!rwrxr"rr&CloughTocher2DInterpolatorryrAssertionErrorprintabsr) r)rr*r alternater_kwrLpabs r_check_accuracyz.TestCloughTocher2DInterpolator._check_accuracys t 9&&& %&&&A  P4QQqqqsVQqqqsV8L8L9=wPPPBB4a!fa!f5E59T!AAAaC&!AAAaC&5I5I9=wPPPB INN2q ! ! #1AA1QQQqS61QQQqS6""A D111Q3111Q3   Aq ' 'B ' ' ' ' '    .AE ; ; ; *bg & & &  s 3EAFc 4ddddg}t|D]}\}}||dddd|z||dddd d |z ||dddd |zd ||dddd d d|z~dS)Ncd|zdzSrrCrs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..$s1qrc d|zSrrCrs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..%s Qrc d|zSrrCrs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..&s arcdd|zzd|zzSrrCrs 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 |dzSNrBrCrs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..: Arc |dzSrrCrs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..;rrc|dz|dzz SrrCrs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..<A1rc ||zSrrCrs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..= 1r& .>g)\(?rrrTrrrs rtest_quadratic_smoketestz7TestCloughTocher2DInterpolator.test_quadratic_smoketest7s     $ $    !'' J JGAt  4d)6): ! < < <  4d)6):D ! J J J J J Jrc4tjgdtj}tj|jdtj}|d|zz }t j|}tj|||}t||dSr2) r!r"r#r$r%r6r7r&rrr8s rr:z-TestCloughTocher2DInterpolator.test_tri_inputFs HNNN: ' ' ' Iagaj 3 3 3 1HnQ 8X 0a 8 8 ; ;Ar"""""rctjgdtj}tj|jdtj}|d|zz }t j|}d}tjt|5tj ||d|ddddS#1swxYwYdSrl) r!r"r#r$r%r6r7rnrror&rrps rrqz5TestCloughTocher2DInterpolator.test_tri_input_rescaleQs 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 Irrctjgdtj}tj|jdtj}|d|zz }t j|}tj|j ||}tj|j |d|}t||dSrg) r!r"r#r$r%r6r7r&rrTrrhs rrjz;TestCloughTocher2DInterpolator.test_tripoints_input_rescale^s H???: ' ' ' Iagaj 3 3 3 1HnQ ?X 0Q ? ? B BUX8QPTUUUVWXX B +++++rrBc dddddg}tjdtjtjgdt tjd d f}t|D]B\}}|||d d d d|z|||d d d d|zdCdS)Nc |dzSrrCrs rrz;TestCloughTocher2DInterpolator.test_dense..nrrc |dzSrrCrs rrz;TestCloughTocher2DInterpolator.test_dense..orrc|dz|dzz SrrCrs rrz;TestCloughTocher2DInterpolator.test_dense..prrc ||zSrrCrs rrz;TestCloughTocher2DInterpolator.test_dense..qrrctjdtjz|ztjdtjz|zzSr)r!cospisinrs rrz;TestCloughTocher2DInterpolator.test_dense..rs3"% **26!BE'!)+<+<<ri)rr<r>r=rirBrg{Gzt?g{Gz?r)r*rrrrT)r*rrrrr_) r!rwrxr_r"rryrr)r)rgridrrs r test_densez)TestCloughTocher2DInterpolator.test_densejs     $ $   < <   tuRX:::%HHHY^^E1--./!'' J JGAt  4d)6): ! < < <  4d)6):D ! J J J J J Jrctjdd}tjd}ttt j||dS)NrvrA)r!rwrandn assert_raisesror&r)r)r*r+s rtest_wrong_ndimz.TestCloughTocher2DInterpolator.test_wrong_ndimsG IOOB " " IOOB  j("Eq!LLLLLrctjdtjdd}tjddtjdzz}t j||}t jt j|}t|dd|dddSrt) r!rwrxryr&rrzr{r|rr}s rrz*TestCloughTocher2DInterpolator.test_pickles t INN2q ! ! INN2  BINN2$6$6!6 6  0A 6 6l6<++,,BBsCLL##c3--88888rcztjdddtjddz fg}tjgd}tj||}d|jd<d }tjdtj|zdtj|zg}tjdtjtjdz |z zdtjtjdz |z zg}||}||}t||tj d tj dd} tj d} | |jj| dddfz}| || z}| || z}tj||}d|jd<||} ||} t| |t| |dS) Nrr>rrArB)r/rrr.rr/)r!r"sqrtr&rrrrrrrwrxrdotrR) r)rTrKrLalphap1p2v1v2Arw1w2s rtest_boundary_tri_symmetryz9TestCloughTocher2DInterpolator.test_boundary_tri_symmetrys 66CA+>?@@)))$$  0 @ @  XsRVE]]*C"&--,?@ A A XsRVBE!GeO444cBF257U?rs ............****** 333333---------$$$$$$$$$  [5 *** ~~~~~~~~B"J"J"J"J"J"J"J"JJu u u u u u u u u u r