L iXdZddlmZmZddlZddlmZmZddl Z ddl m Z GddZ y)z5 Unit tests for optimization routines from _root.py. )assert_ assert_equalN)raiseswarnsrootceZdZdZdZdZdZejjdZ ejjdgddZ y ) TestRootc d}d}dD]}|dvr|dvr|}nd}t|ddg|d|}t|ddg|d |}|d ||jd ||j}t|j|t|j|tt ||jj t ||jj k|y) NcR|\}}tj|dzdz |dzdz gSNnparrayzxys e/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/scipy/optimize/tests/test__root.pyfuncz)TestRoot.test_tol_parameter..func/DAq88QTAXq!tax01 1cZ|\}}tjd|dzzdgdd|dzzggS)Nrrrrs rdfuncz*TestRoot.test_tol_parameter..dfuncs7DAq88a1fa[1a1f+67 7r)hybrlmbroyden1broyden2anderson diagbroydenkrylov) linearmixingexcitingmixing)rr皙?g-C6?)jactolmethod?z: z vs. )rrrsuccessabsmax)selfrrr*r(sol1sol2msgs rtest_tol_parameterzTestRoot.test_tol_parameter s 2 80 F;;'s3iSd6JDs3iSc&IDHBtDFF|nE$tvv,@C DLL# & DLL# & CTVV %))+c$tvv,.?.C.C.EE ! rc|d}dD]4}ttjtjd|d|i6y)Nct|dSNr)r-rs rnormz$TestRoot.test_tol_norm..norm,sqt9 r)r&r#r%r"r r!r$rtol_norm)r*options)rr zeros_likezeros)r/r8r*s r test_tol_normzTestRoot.test_tol_norm*s< ! ,F  F#T* , ,rc,dd}t|ddgdy)NcR|\}}tj|dzdz |dz|z gSr r)rfrrs rrz=TestRoot.test_minimize_scalar_coerce_args_param..func<rrr'g?)argsrr)r/rs r&test_minimize_scalar_coerce_args_paramz/TestRoot.test_minimize_scalar_coerce_args_param:s 2 TC:C(rcGdd}|}tt5t|ddgddddy#1swYyxYw)NceZdZdZdZy)!TestRoot.test_f_size..funcd|_yr6count)r/s r__init__z*TestRoot.test_f_size..fun.__init__Fs  rc|xjdz c_|jdzs|dd|d|dz dzzzdz }|S|dd|d|dz dzzzdz d|d|dz dzz|dzg}|S)Nrrr+rg?rH)r/rrets r__call__z*TestRoot.test_f_size..fun.__call__Is a  QA$!qt '9!99C?C  aD3!A$1+!);#;;cA!A$1+!!33ad:.fun]sIadFT!A$'\)B.AaD!G a!f0Dq0HI Irc*dd|dzgd|dzdggS)Nrr+rg?rrLrRr\s rgradz$TestRoot.test_gh_10370..grad`s%ad OcAaDj!_5 5rc(||||fS)NrR)rr]rSr_s rfun_gradz(TestRoot.test_gh_10370..fun_gradcsq'?DG$44 4rrrBr$)rAr*z'Method krylov does not use the jacobian)match)rAr*r(T)rr<r assert_warnsRuntimeWarningrrr,) r/rax0refmessageres1res2rSr_s @@r test_gh_10370zTestRoot.test_gh_10370Xs  J 6 5XXa[3h7; . 8 GRd8FD G . 8 L"4dKD L TVVSUU#TVVSUU#||t||Bs{{BdBBBBB G G L LsC;0D;DDr*) rrr r!r"r%r#r&r$zdf-sanecJd}t|dg|}|j|k(sJy)Nc |dz S)NrrRr7s rrz,TestRoot.test_method_in_result..funcxs q5Lrr)rer*)rr*)r/r*rress rtest_method_in_resultzTestRoot.test_method_in_resultss+  4QC/zzV###rN) rOrPrQr3r=rCrXpytestmark thread_unsaferj parametrizernrRrrr r s`:, )-. [[CC4 [[X(=>$ >$rr )__doc__ numpy.testingrrrorrUrrcnumpyrscipy.optimizerr rRrrrws'0 Ap$p$r