o 4Bi<@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)selfxrg/var/www/html/Trade-python/venv/lib/python3.10/site-packages/scipy/optimize/tests/test_differentiate.pyf szTestDifferentiate.fr 333333?皙? cCsVt|j|}t|}tj|j|t t |j||j |j j |j ks)JdSr ) differentiaterrr pdfnptestingrdfrabserrorr shape)r r resrefrrr test_basics zTestDifferentiate.test_basiccasecCsL|\}}tt||}|d}t|j|}||}t|j|dddS)Ng?绽|=atol)getattrrmedianrr rrr)r r distnameparamsdistr rrrrr test_accuracys    zTestDifferentiate.test_accuracyorderr) ))r.r0cs6|r tddd|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|jjtjsJdd |D} t |j| t |jj |t|jjtjsJdd |D} t |j| t t|jjt |jj |j j t|jjtjsJdd |D} t |j| t t|jjt |jj |j j t|jjtjsJdS)Nrrr-rcstj|dS)Nr*)rrr )r*r rr_differentiate_single,szCTestDifferentiate.test_vectorization.._differentiate_singlecsRjd7_j|jks|jdkrdn|jd7_j|g|Ri|S)Nr+)nitfevalsizendimrr)r argskwargs)rnr rrr0s,z/TestDifferentiate.test_vectorization..fr4rr1cSg|]}|jqSrr2.0rrrr :z8TestDifferentiate.test_vectorization..cSr<r)rr=rrrr?>r@cSr<r)rr=rrrr?Br@+bz9TestDifferentiate.test_flags..f..cst|Sr )rexprandomr2rngrrrYcr@cSs t|Sr )rr[r2rrrrYds cSst|tjdS)Nr)r full_likenanr2rrrrYer@csg|] \}}||qSrr)r>r jfuncsrrr?fsz;TestDifferentiate.test_flags..f..)r5ziprH)xsjsrrr^rbrr`s z'TestDifferentiate.test_flags..frr/rJr++=r0)rtolr*r9)rr\ default_rngr5arangeint64rarrayeim _ECONVERGEDr _ECONVERR _EVALUEERRrrC)r r9r ref_flagsrrgr test_flags\s zTestDifferentiate.test_flagscsRtjdfdd}t|ddddd}ttjttjtj g}t |j |dS) NrWcs2|dt|t|t|tjdgS)NrXr)rr[r\r_r`r2r]rrrvs z6TestDifferentiate.test_flags_preserve_shape..fr+rir0T)rjr*preserve_shape) rr\rkrrnrorprrqrrrrC)r rrrsrr]rtest_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+r0)rsinr2rrrrs:z0TestDifferentiate.test_preserve_shape..frr+r.rrwr0Tru)rcosrxrrr)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#rjr*gMbP?r#gư>rj) rr r rdictcopyrrrrr) r r(r rrkwargs0r:res1res2rrrtest_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|ksNJtddd 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?r0) step_factorrrwr/r)r*rstep_direction)rrrArjr4r) rr r rrrrr|rrsqrt) r r(r rrrrr:rrrrtest_step_parameterss4   z&TestDifferentiate.test_step_parameterscCsldd}tddd}t|}d\||dk<||dk<t|||d }t|jt|t|js4JdS) NcSs$t|}tj||dk|dk<|S)Nrr0)rr[r`)r yrrrrs z0TestDifferentiate.test_step_direction..frr0r)r+r4rgffffff?)r) rrE zeros_likerrrr[allrB)r rr rrrrrtest_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+rrrrrrszATestDifferentiate.test_vectorized_step_direction_args..df)r+r0r.r/r4r+r4rr+r0r.)rr9)rrnrFr broadcast_torrr)r rrr hdirrrrrrr#test_vectorized_step_direction_argssz5TestDifferentiate.test_vectorized_step_direction_argscs(d}tdfdd}d}t||dd}t|jr Jt|j|ddd ks2Jt|jks.fV瞯<)rrjr+r0csbjd7_|_t|dsJ|jjvsJj|j|jtjks(Jjkr/t dS)Nr+r ) iterrhasattrrdfsaddrCro _EINPROGRESS StopIterationr)callbackrrrrs z9TestDifferentiate.test_maxiter_callback..callbackr4)rrjrC)rr rranyrBrrDr5rrsetrkeysrorqr _ECALLBACK)r r r default_orderrrkeyr)rr(rrtest_maxiter_callbacks* $  ,z'TestDifferentiate.test_maxiter_callbackrrg?gffffff?rJcstj|dd}fdd}fdd}t||d||d}|jjks&J|jjks.J|jjks6Jtj}t |jt |jt |d dS) Nrhrcs|jksJt|Sr )rJrr[r2rhrrrs z'TestDifferentiate.test_dtype..fcs4|jjksJ|jjksJ|jjksJdSr )r rJrrrrhrrrsz.TestDifferentiate.test_dtype..callbackr/)r*rrr) rasarrayrr rJrrfinfoepsrr[r)r rr rJrrrrrrhr test_dtypes    "zTestDifferentiate.test_dtypecCshd}tjt|d tddWdn1swYd}tjt|dtdddWdn1s9wYd}tjt|dtd dd d gWdn1s[wYd }tjt|dtd ddddWdn1s}wYtjt|dtdddddWdn1swYtjt|dtdddddWdn1swYtjt|dtdddtdWdn1swYd}tjt|dtdddddWdn 1swYtjt|dtdddddWdn 1s"wYd}tjt|dtdddddWdn 1sEwYtjt|dtd ddddWdn 1sfwYd!}tjt|dtd"ddd#d$Wdn 1swYd%}tjt|dtd&ddd'd(WddS1swYdS))Nz`func` must be callable.)matchr+z3Abscissae and function output must be real numbers.cS|Sr rr2rrrrY1z9TestDifferentiate.test_input_validation..y?z6When `preserve_shape=False`, the shape of the array...cSsgdS)N)r+r0r.rr2rrrrY5rZz6Tolerances and step parameters must be non-negative...cSrr rr2rrrrY9rr4r"cSrr rr2rrrrY;rekkircSrr rr2rrrrY=r)rcSrr rr2rrrrY?r)rz%`maxiter` must be a positive integer.cSrr rr2rrrrYCr?)rcSrr rr2rrrrYErrz"`order` must be a positive integercSrr rr2rrrrYIrr1cSrr rr2rrrrYKrz'`preserve_shape` must be True or False.cSrr rr2rrrrYOrherringryz`callback` must be callable.cSrr rr2rrrrYSr shrubbery)r)pytestraises ValueErrorrobject)r messagerrrtest_input_validation(s^ $z'TestDifferentiate.test_input_validationcsdd}t|ddd}|jsJt|jdtdD]Ld}fd d}d |d }t||d td d }t|j|d dt|jtj t||td d}|jsWJ|j d ks^Jt|j|d dqdd}t|d dd}t|jddS)NcSs t|jtjs J|ddS)Ncr+)rrIrJfloatingr2rrrrZs z/TestDifferentiate.test_special_cases..fr!rgl)~IQr,rcs d|S)Nr0rr2r;rrrk r0r+)rr*rr1cSs ||dSrr)r crrrrzrr.)r9) rrBrrrangerMrrrr`r5)r rrr rrrrtest_special_casesUs$     z$TestDifferentiate.test_special_casescCs |ddS)Nr+r.rr2rrrrYs zTestDifferentiate.cCs"t|dk|dd|ddS)Nr+r.)rwherer2rrrrYs"cCs<d}t|gd|d}t|jsJt|jd|ddS)NgؗҜs