ftj<ddlZddlZddlZddlZddlmZddlZddlm Z m Z ddl m Z ddl mcmZddlmcmZddlZddlZejdkZdZGddZGd d ZGd d ZdS) N)raises)xp_assert_closeassert_almost_equal)check_free_memorylctjtjtjt d|S)Ndata)ospathjoinabspathdirname__file__)basenames c/var/www/html/CA-Chatbot/venv/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)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 |usJdSr0) rr r!r"r#qhullDelaunayr$r%rtri)r'r(r)r6 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- rr r!broadcast_arrayslinspaceravelTcopyr$r%r)r'pointsrJxxyyxizirIs @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)rZ)rZ)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) r`)rarrr=rrarKr`Tr\rL)r'rRrIrSrTrUrV 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||dSNrYrrr1Tr\) rr r!r"r#r4r5r$r%rRrr'r(r)r6r* 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 NrYrrr1zORescaling is not supported when passing a Delaunay triangulation as ``points``.)matchTr\) rr r!r"r#r4r5pytestr ValueErrorr$r%r'r(r)r6rks 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|dddSNr@y?r) rrandomseedrandr$r%pickleloadsdumpsr)r'r(r)rJip2s 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'rihracNtj||f\}}|||f}|SN)rmgrid)r7 slice_rows slice_colsgrid_xgrid_yress r do_interpz;TestLinearNDInterpolation.test_threading..do_interps0Xj*&<=NFF,/00CJri4)targetargs)rrr"meshgridvstackrOrPr$r% threadingThreadslicestartr ) r'r_ticks phi_ticksr_gridphi_gridrrRrIr7worker_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.r2r8rWr^rcrhror|rlmarkslowskipif _IS_32BITrrArrrrs######### ' ' '+,+,+,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|dS)N)r)rr@r<)r-r@rg?g333333?g?rcd|zdzSNrr-rAr(r)s rz>TestEstimateGradients2DGlobal.test_smoketest..s!A#'rrc d|zSNrrArs rrz>TestEstimateGradients2DGlobal.test_smoketest..s !a%rr<c d|zSNrArs rrz>TestEstimateGradients2DGlobal.test_smoketest..s "q&rr:cdd|zzd|zzSNr?L,@rArs rrz>TestEstimateGradients2DGlobal.test_smoketest..s!ac'E!G+r)r?rrr-ư>)tol)r@gh㈵>zitem )rtolatolerr_msg) rr floatr4r5 enumerater$estimate_gradients_2d_globalr#r) r'r(r6funcsjfuncgradrHdzs rr+z,TestEstimateGradients2DGlobal.test_smoketests> H@@@GL N N NnQ " !6 *   ( & ) + +Z 8   )//  OA|dQqqqsVQqqqsV$$A6sA4HHHB8v%%%% BHTNN47+ad2D#         rctjtd}tj|jd}t j|}tj 5tj ddtj tj ||dddddS#1swxYwYdS)Nzestimate_gradients_hang.npyrignorez$Gradient estimation did not converger-)maxiter)rloadrrurwr#r4r5warningscatch_warningsfilterwarningsr$GradientEstimationWarningr)r'rRrIr6s rtest_regression_2359z2TestEstimateGradients2DGlobal.test_regression_2359s#@AABB Q00nV$$ $ & & J J  #62     1#vq I I I I  J J J J J J J J J J J J J J J J J Js38B88B<?B<N)rrrr+rrArrrrs70JJJJJrrceZdZ 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) Nrs)rr:r<r;rr)rg?rrr-r)rr]2r@z_check_accuracy: abs(a-b):zip.grad:) rru RandomStater rr$CloughTocher2DInterpolatorrwrAssertionErrorprintabsr) r'rr(r alternater]kwrngrJpabs 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|||dddd d | ||dddd |d ||dddd d d|~dS)Ncd|zdzSrrArs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..,s1qrc d|zSrrArs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest..-s Qrc d|zSrrArs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest...s arcdd|zzd|zzSrrArs rrzFTestCloughTocher2DInterpolator.test_linear_smoketest../sQqS57*rgvIh%<=gHz> Function rrrrTzFunction (alternate) )rrrrrzFunction (rescaled) rrrrr]zFunction (alternate, rescaled) )rrrrr]rrrr'rrrs rtest_linear_smoketestz4TestCloughTocher2DInterpolator.test_linear_smoketest)s !     * *  !''  GAt  %d1 !     %d333 !     %d2q22D !     %dt=!== !      rc ddddg}t|D]@\}}||dddd| ||dddd|d AdS) Nc |dzSNr@rArs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..G Arc |dzSrrArs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..Hrrc|dz|dzz SrrArs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..IA1rc ||zSrrArs rrzITestCloughTocher2DInterpolator.test_quadratic_smoketest..J 1r& .>g)\(?rrrTrrrs rtest_quadratic_smoketestz7TestCloughTocher2DInterpolator.test_quadratic_smoketestDs     $ $    !''  GAt  $T?q?? !     $T?q??TX !       rc4tjgdtj}tj|jdtj}|d|zz }t j|}tj|||}t||dSr0) rr r!r"r#r4r5r$rr)r'r(r)r6r*s rr8z-TestCloughTocher2DInterpolator.test_tri_inputUs HNNN: ' ' ' Iagaj 3 3 3 1HnQ 8X 0a 8 8 ; ;Ar"""""rctjgdtj}tj|jdtj}|d|zz }t j|}d}tjt|5tj ||d|ddddS#1swxYwYdSrj) rr r!r"r#r4r5rlrrmr$rrns rroz5TestCloughTocher2DInterpolator.test_tri_input_rescale`s 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 Irpctjgdtj}tj|jdtj}|d|zz }t j|}tj|j ||}tj|j |d|}t||dSre) rr r!r"r#r4r5r$rrRrrfs rrhz;TestCloughTocher2DInterpolator.test_tripoints_input_rescalems 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||||d d d d|dCdS)Nc |dzSrrArs rrz;TestCloughTocher2DInterpolator.test_dense..}rrc |dzSrrArs rrz;TestCloughTocher2DInterpolator.test_dense..~rrc|dz|dzz SrrArs rrz;TestCloughTocher2DInterpolator.test_dense..rrc ||zSrrArs rrz;TestCloughTocher2DInterpolator.test_dense..rrctjdtjz|ztjdtjz|zzSr)rcospisinrs rrz;TestCloughTocher2DInterpolator.test_dense..s3"% **26!BE'!)+<+<<ri)rr:r<r;rir@rg{Gzt?g{Gz?r)r(rrrrT)r(rrrrr]) rrurr_r rrwrr)r'rrgridrrs r test_densez)TestCloughTocher2DInterpolator.test_denseys      $ $   < <  i##D))uRX:::%HHHXXeQ''()!''  GAt  $ToRSoo !     $T'A !       rctjdd}tjd}ttt j||dS)Nrtr?)rrurandn assert_raisesrmr$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|dddSrr) rrurrwr$rrxryrzr)r'rr(r)rJr{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?r@)r-rrr.rr-)rr sqrtr$rrrrrrrurrdotrP)r'rRrIrJalphap1p2v1v2rArw1w2s rtest_boundary_tri_symmetryz9TestCloughTocher2DInterpolator.test_boundary_tri_symmetrys 66CA+>?@@)))$$  0 @ @  XsRVE]]*C"&--,?@ A A XsRVBE!GeO444cBF257U?rsu ****** FFFFFFFF333333.........$$$$$$$$$  [5 *** @@@@@@@@F'J'J'J'J'J'J'J'JT@ @ @ @ @ @ @ @ @ @ r