a h<@sdddlZddlZddlmZmZmZddlmm Z ddl m Z ddl mZmZGdddZdS)N)assert_array_lessassert_allclose assert_equal)stats)_differentiate_EERRORINCREASEc @sLeZdZddZejddedddgdd Z ejd e j j d d Z ejd ddgejdedddgddZddZddZddZddZddZd d!Zd"d#Zd$d%Zejd&d'ejdd(d(d)gfejd*ejejejfd+d,Zd-d.Zd/d0Zejjejd d1d2dfd3d2dffd4d5Zd6S)7TestDifferentiatecCst|SN)rnormcdf)selfxrc/var/www/html/assistant/venv/lib/python3.9/site-packages/scipy/optimize/tests/test_differentiate.pyf szTestDifferentiate.fr 333333?皙? cCsVt|j|}t|}tj|j|t t |j||j |j j |j ksRJdSr ) differentiaterrr pdfnpZtestingrdfrabserrorr shape)r r resrefrrr test_basics  zTestDifferentiate.test_basiccasecCsL|\}}tt||}|d}t|j|}||}t|j|dddS)Ng?绽|=atol)getattrrZmedianrr rrr)r rdistnameparamsdistr rrrrr test_accuracys    zTestDifferentiate.test_accuracyorderr) ))r,r.cs<|rtddd|nd}t|tjfdd}fddd _d _t|d }||}d d |D}t |j |t |j j |dd |D}t |j |t |j j |dd |D} t |j| ddt |jj |dd |D} t |j| t |jj |t|jjtjs._differentiate_singlecsRjd7_j|jks(|jdkr,dn|jd7_j|g|Ri|S)Nr))nitfevalsizendimrr)r argskwargs)rnr rrr0s,z/TestDifferentiate.test_vectorization..fr2rr/cSsg|] }|jqSrr0.0rrrr :z8TestDifferentiate.test_vectorization..cSsg|] }|jqSr)rr:rrrr<>r=cSsg|] }|jqSr)rr:rrrr<Br=+br=z9TestDifferentiate.test_flags..f..cst|Sr )rexprandomr0rngrrrMcr=cSs t|Sr )rrNr0rrrrMdr=cSst|tjdS)Nr)r full_likenanr0rrrrMer=csg|]\}}||qSrr)r;r jfuncsrrr<fr=z;TestDifferentiate.test_flags..f..)r3ziprD)ZxsjsrrrQrUrr`s z'TestDifferentiate.test_flags..frr-rFr)+=r.)rtolr(r7)rrO default_rngr3ZarangeZint64rarrayeim _ECONVERGEDr _ECONVERR _EVALUEERRrr@)r r7r ref_flagsrrYr test_flags\s zTestDifferentiate.test_flagscsRtjdfdd}t|ddddd}ttjttjtj g}t |j |dS) NrKcs2|dt|t|t|tjdgS)NrLr)rrNrOrRrSr0rPrrrvs z6TestDifferentiate.test_flags_preserve_shape..fr)r[r.T)r\r(preserve_shape) rrOr]rr^r_r`rrarbrr@)r rrrcrrPrtest_flags_preserve_shapess  z+TestDifferentiate.test_flags_preserve_shapec Csdd}d}ddtd|ddtd|dtd||dddtd||dg}t||d d }t|j|dS) NcSs:|td||td|td||ddgS)Nr,rr)r.)rsinr0rrrrsz0TestDifferentiate.test_preserve_shape..frr)r,rrgr.Tre)rcosrhrrr)r rr rrrrrtest_preserve_shapes&6z%TestDifferentiate.test_preserve_shapec CsFt}d}|j}||}tdddd}|}d|d<t||fi|}tt|j |dd|d<t||fi|}tt|j |dtt|j |t|j ||}d|d<t||fi|}tt|j |dt |d|d<t||fi|}tt|j |dt |tt|j |t|j |dS) Nr)rr-)r"r\r(gMbP?r"gư>r\) rr r rdictcopyrrrrr) r r&r rrZkwargs0r8res1res2rrrtest_convergences* z"TestDifferentiate.test_convergencec Cst}d}|j}||}t||ddd}t||ddd}t|j|t|j|ks\Jt||ddd}t||ddd}t|j|t|j|ksJtddd d }t||fddd |}t||fddd |}t|j|jd d tdddd }t||fddd |}t||fdt ddd |}t|j|jd d d|d<t||fddd |}t||fdt ddd |}t|j|jd d dS)Nr)g?) initial_stepmaxiterg?r.) step_factorrrrgr-r)r(rrstep_direction)rqrsr>r\r2rt) rr r rrrrrlrrsqrt) r r&r rrrnror8rrrrtest_step_parameterss4   z&TestDifferentiate.test_step_parameterscCsldd}tddd}t|}d\||dk<||dk<t|||d }t|jt|t|jshJdS) NcSs$t|}tj||dk|dk<|S)Nrr.)rrNrS)r yrrrrs z0TestDifferentiate.test_step_direction..frr.r)r)r2rgffffff?)rt) rrBZ zeros_likerrrrNallr?)r rr rtrrrrtest_step_directions z%TestDifferentiate.test_step_directioncCsdd}dd}tgdddd}tgdddd}td d gddd}t||||fd }t||||jj}t|j|dS) NcSs||Sr rr prrrrsz@TestDifferentiate.test_vectorized_step_direction_args..fcSs|||dSNr)rr{rrrrszATestDifferentiate.test_vectorized_step_direction_args..df)r)r.r,r-r2r)r2rr)r.r,)rtr7)rr^rCrZ broadcast_torrr)r rrr hdirr|rrrrr#test_vectorized_step_direction_argssz5TestDifferentiate.test_vectorized_step_direction_argscs(d}tdfdd}d}t||dd}t|jr@Jt|j|ddd ksdJt|jksxJfd d d _ d_ t _ t||dd }| D]p}|dkr||tjksJj |tjksJ||tjksJq||j |kr||ksnJqdS)Ng~,?r,cs|}|Sr )r )r r)r&rrrs z2TestDifferentiate.test_maxiter_callback..fV瞯<)rrr\r)r.csbjd7_|_t|ds"J|jjvs2Jj|j|jtjksPJjkr^t dS)Nr)r ) iterrhasattrrdfsaddr@r_ _EINPROGRESS StopIterationr)callbackrrrrrs z9TestDifferentiate.test_maxiter_callback..callbackr2)rr\r@)rr rranyr?ryrAr3rrsetrkeysr_rarZ _ECALLBACK)r r rZ default_orderrrokeyr)rr&rrrtest_maxiter_callbacks( $  z'TestDifferentiate.test_maxiter_callbackrr~g?gffffff?rFcstj|dd}fdd}fdd}t||d||d}|jjksLJ|jjks\J|jjkslJtj}t |jt |jt |d dS) NrZrcs|jksJt|Sr )rFrrNr0rZrrrsz'TestDifferentiate.test_dtype..fcs4|jjksJ|jjks J|jjks0JdSr )r rFrrrrZrrrsz.TestDifferentiate.test_dtype..callbackr-)r(rtrru) rZasarrayrr rFrrZfinfoepsrrNrv)r rr rFrrrrrrZr test_dtypes    zTestDifferentiate.test_dtypecCsld}tjt|dtddWdn1s20Yd}tjt|dtdddWdn1sr0Yd}tjt|d"td dd d gWdn1s0Yd }tjt|d"td ddddWdn1s0Ytjt|d"tdddddWdn1s<0Ytjt|d"tdddddWdn1s~0Ytjt|d$tdddtdWdn1s0Yd}tjt|d"tdddddWdn1s0Ytjt|d"tdddddWdn1sJ0Yd}tjt|d"tdddddWdn1s0Ytjt|d"td ddddWdn1s0Yd!}tjt|d"td"ddd#d$Wdn1s0Yd%}tjt|d"td&ddd'd(Wdn1s^0YdS))Nz`func` must be callable.)matchr)z3Abscissae and function output must be real numbers.cSs|Sr rr0rrrrM1r=z9TestDifferentiate.test_input_validation..y?z6When `preserve_shape=False`, the shape of the array...cSsgdS)N)r)r.r,rr0rrrrM5r=z6Tolerances and step parameters must be non-negative...cSs|Sr rr0rrrrM9r=r2r!cSs|Sr rr0rrrrM;r=ZekkirucSs|Sr rr0rrrrM=r=)rqcSs|Sr rr0rrrrM?r=)rsz%`maxiter` must be a positive integer.cSs|Sr rr0rrrrMCr=?)rrcSs|Sr rr0rrrrMEr=rz"`order` must be a positive integercSs|Sr rr0rrrrMIr=r/cSs|Sr rr0rrrrMKr=z'`preserve_shape` must be True or False.cSs|Sr rr0rrrrMOr=Zherringriz`callback` must be callable.cSs|Sr rr0rrrrMSr=Z shrubbery)r)pytestZraises ValueErrorrobject)r messagerrrtest_input_validation(sD(,0022422222z'TestDifferentiate.test_input_validationcsdd}t|ddd}|js Jt|jdtdD]d}fd d}d |d }t||d td d }t|j|d dt|jtj t||td d}|jsJ|j d ksJt|j|d dq4dd}t|d dd}t|jddS)NcSs t|jtjsJ|ddS)Ncr))rrErFZfloatingr0rrrrZsz/TestDifferentiate.test_special_cases..fr rugl)~IQr*rcs d|S)Nr.rr0r9rrrksr.r))rrr(rr/cSs ||dSr}r)r crrrrzsr,)r7) rr?rrrangerHrrrrSr3)r rrr rrrrtest_special_casesUs$    z$TestDifferentiate.test_special_casescCs |ddS)Nr)r,rr0rrrrMr=zTestDifferentiate.cCs"t|dk|dd|ddS)Nr)r,)rwherer0rrrrMr=cCs<d}t|gd|d}t|js(Jt|jd|ddS)NgؗҜs