xf d dl Z d dlmZ d dlmZmZmZmZmZm Z d dl
Z d dl
m Z m
Z
mZmZmZmZmZmZ d dlmZmZmZmZ d dlmZ d dl m Z m Z d e j e! jD z Z# e e! jD Z$ejJ ejL ejN ejP ejR gZ*ej. gZ+e*e+z Z,d Z-d
Z.d Z/d Z0d
Z1d Z2d Z3ed Z4 G d d Z5 G d de5 Z6 G d de5 Z7d Z8d Z9d Z: G d d Z;d Zd Z?d Z@d! ZAe j j d"d#ej fd$ej fg d% ZFd& ZGe j j d'g d( d) ZHe j j d*g d( d+ ZIe j j d'g d( e j j d,d-d.g d/ ZJd0 ZKe j j d'g d( e j j d,d-d.g d1 ZLd2 ZMd3 ZNe j j d*e, d4 ZOy)5 N) lru_cache) assert_warnsassert_assert_allclose assert_equalassert_array_equalsuppress_warnings)finfopowernanisclosesqrtexpsincos) _zeros_pynewton root_scalarOptimizeResult)getfullargspec_no_self) get_tests functions c | dz d| z z
dz
S N xs A/usr/lib/python3/dist-packages/scipy/optimize/tests/test_zeros.pyf1r" ! s 6AE>A c d| z dz
S Nr r r s r! f1_1r& % s q519r# c dd| z z S N @r r r s r! f1_2r* ) s Q;r# c B t | t | t | fS N)r" r& r* r s r! f1_and_p_and_ppr- - s
a5$q'47 ""r# c 0 t | t | z
S r, r r r s r! f2r0 2 q6CF?r# c 0 t | t | z S r, )r r r s r! f2_1r3 6 r1 r# c 0 t | t | z S r, r/ r s r! f2_2r5 : r1 r# c | S r, r r s r! f_lrucachedr7 ? s Hr# c e Zd Zd ej e j z Zd ej e j z Z ddZ ddZ
ddZ y)TestScalarRootFindersr Nc g }|xs g D ]0 }||vrddddj || }|j || 2 t d
i |}|j ddd |xs g D ]
}|| ||< |j d } |j d d
}
||d |
i|\ } } | | |fS # t $ r0 | t j t d d t
j |fcY S w xY w) Nx0x1f)abfuncTF full_outputdisprootargsr ) getappenddictupdate Exceptionzeros RootResultsr
_EVALUEERR)
selftcmethod
sig_args_keyssig_kwargs_keyskwargs method_argsk
method_kwargsrD func_argsrrrs
r!
_run_one_testz#TestScalarRootFinders._run_one_testJ s
$" &A{ T37;;AqA r!u % & v
T5 AB &B %A !!uM! % vvf~FF62& NKIiI=IEArR< N**3B8H8HI2M M Ns B$ $6C C c t | }t |j t |j }t |j
|z
}|j
d| } g }
|dv r:|dv r&|
j
d |dv r|
j
d | j |d< n | j |d< | j |d < |D cg c]" } t | j | |f| |
d
| $ } } |xs g }| D
cg c] }
|
d j r|
}}
|D
cg c] }
|
d d
|vs|
}}
|D
cg c]
}
|
d d
}}
t t | |gdg g | j | j d} |j di | |d }|j d|d }| D
cg c] }
|
d j s|
}}
|D
cg c] }
|
d j }}
|D
cg c] }
|
d }}
t! ||| D
cg c]% \ }}}
t# |||| s|
d d
|vr|g|
z ' }}}}
|D cg c]+ \ }}}} | d |g| j dt% - }}}}} t! || D
cg c] \ } }
| dk7 s | g|
z }} }
t |t | gg dg yc c} w c c}
w c c}
w c c}
w c c}
w c c}
w c c}
w c c}
}}w c c} }}}w c c}
} w )zRun test-cases using the specified method and the supplied signature.
Extract the arguments for the method call from the test case
dictionary using the supplied keys for the method's signature.N)secantr halley)r r^ fprime)r^ fprime2tolxtolrtol)rR rS r rF IDr rb rc )rc atolr= rE r )_getfullargspecr
kwonlyargslendefaultsrE rH rb rc listr[ convergedr rJ rG rD zipr
tuple) rO testsrQ name
known_failrT sig nDefaults nRequiredrR rS rP resultseltnotcvgd notcvged_IDStolsrc rf cvgdapproxcorrectr> cnotclosearootfulloutfvsfvs r! run_testsz TestScalarRootFinders.run_testsa s f%CNN"# % MI- ),
1 1++ &&x0:%#**95 IIF5M !YYF6N !YYF6N CHI<> *** 7&3 +7/57 8 I I
%2
")B3Q1A1A3BB")M3SWT]*-L3MM189#B
9 9c,' 6B@ 4995 f F|xxtF|, &;#a&*:*:;;)-.##a&++..%)*c3q6**/267D/I 9 9)!Q 14d;B
Z7 C#I 9 9
-56 6(5!Wb r#wu7rvvfeg 67 6 6.1#x.@L72sB!GRD3JLLhH
.Q89I CM9 <.*96LsT 'J%3J*J*J/"J/,J4 J94J9>J> K9*K.0K
2
K Kc L t || } | j |||fd|i| y)zuRun a collection of tests using the specified method.
The name is used to determine some optional arguments.
smoothnessrq N)r r )rO
collectionrQ rp r rq rT ro s r! run_collectionz$TestScalarRootFinders.run_collection s,
*<ufdLzLVLr# )NNr, ) __name__
__module__ __qualname__npr
floatepsrb rc r[ r r r r# r! r9 r9 D sZ
xrxx"" "D xrxx"" "D6:&*N.49l CG"&Mr# r9 c e Zd Zej j de ej j de d Zej j de ej j de d Z ej j de d Z
ej j de j e j e j g d Zej j de d Zy) TestBracketMethodsrQ functionc dt d }}t ||j ||g|| j | j }|j
sJ t
|j d| j | j y )N ? )rQ bracketr; rb rc ?rf rc )r r r rb rc rl r rD )rO rQ r r> r? rY s r! test_basic_root_scalarz)TestBracketMethods.test_basic_root_scalar s^ 471 1a&Q !YYTYY
8{{{$))$))Dr# c dt d }} ||||| j | j d \ }}|j sJ t |d| j | j y )Nr r T)rb rc rB r r )r rb rc rl r )rO rQ r r> r? rD rY s r! test_basic_individualz(TestBracketMethods.test_basic_individual sX 4711adiidii%)+a {{{c Br# c B | j d||j d y )Napsr r )r r )rO rQ s r! test_aps_collectionz&TestBracketMethods.test_aps_collection s E66??qIr# c r |t j k( rdhni }| j d||j | y )Nzfun7.4 chandrupatla)rq )rL ridderr r )rO rQ rq s r! test_chandrupatla_collectionz/TestBracketMethods.test_chandrupatla_collection s8 $*U\\#9hZr
NFFOO'1 3r# c h d\ }} |t ||d \ }}|j sJ t |d y )N)rF r TrB r )r7 rl r )rO rQ r> r? rD rY s r! test_lru_cached_individualz-TestBracketMethods.test_lru_cached_individual s7 1a=a{{{a r# N)r r r pytestmark parametrizebracket_methodststutils_functionsr r r rL bisectr toms748r r r r# r! r r s [[ X7 [[ Z);< E = 8 E [[ X7 [[ Z);< C = 8 C [[ X7J 8J [[ X ell(-
(7 8383
[[ X7! 8!r# r c e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd Zej* j- dddg d Zy)
TestNewtonc n dg}|ddgz
}dD ]& }| j |t j dd| ( y )N aps.13.00z aps.12.05 aps.12.17r complexr r r rq r rL r rO rq r s r! test_newton_collectionsz"TestNewton.test_newton_collections sL !]
{K00
, EJ
ELL(+,
E Er# c b g d}dD ]& }| j |t j dd| ( y )N)z aps.12.06z aps.12.07z aps.12.08z aps.12.09z aps.12.10z aps.12.11z aps.12.12z aps.12.13z aps.12.14z aps.12.15z aps.12.16r z aps.12.18r r r^ r r r r s r! test_halley_collectionsz"TestNewton.test_halley_collections s? 0
- EJ
ELL(+,
E Er# c t t t ft t t
ffD ] \ }}}t
j |dd }t || dd t
j |ddd }t || dd t
j |d|d }t || dd t
j |d||d }t || dd y )
Nr ư>)ra r rf )r< ra )r_ ra )r_ r` ra ) r" r& r* r0 r3 r5 rL r r )rO r= f_1f_2r s r! test_newtonzTestNewton.test_newton s t ,r4.>? 0KAsC Qt,A AaD!$ / QaT2A AaD!$ / Q#48A AaD!$ / Q#sEA AaD!$ / 0r# c f t t t ft t t
ffD ]4 \ }}}t
|dd|d }t ||j dd 6 t t t ft t t
ffD ]3 \ }}}t
|ddd }t ||j dd 5 y) z#Invoke newton through root_scalar()r r r )rQ r; r_ rb r r rQ r; rb N r" r& r* r0 r3 r5 r r rD rO r= r r rY s r! test_newton_by_namez TestNewton.test_newton_by_name s t ,r4.>? 5KAsCAh1StLA AaffIqt 4 5 t ,r4.>? 5KAsCAh14@A AaffIqt 4 5r# c t t t ft t t
ffD ]b \ }}}t
|dddd }t ||j dd t
|dddd }t ||j dd d t t t ft t t
ffD ]3 \ }}}t
|ddd }t ||j dd 5 y
) z#Invoke secant through root_scalar()r] r r r rQ r; r< rb r r r r Nr r s r! test_secant_by_namez TestNewton.test_secant_by_name s t ,r4.>? 5KAsCAh1FA AaffIqt 4Ah1FA AaffIqt 4 5
t ,r4.>? 5KAsCAh14@A AaffIqt 4 5r# c t t t ft t t
ffD ]5 \ }}}t
|dd||d }t ||j dd 7 y)z#Invoke halley through root_scalar()r^ r r )rQ r; r_ r` rb r r Nr r s r! test_halley_by_namez TestNewton.test_halley_by_name sV t ,r4.>? 5KAsCAh1#&$@A AaffIqt 4 5r# c , d}t j t | 5 t t dt
dd d d d d}t j t | 5 t t dt dd d d d y # 1 sw Y HxY w# 1 sw Y y xY w) Nz$fprime2 must be specified for halleymatchr^ r r )rQ r_ r; rb z#fprime must be specified for halley)rQ r` r; rb )r raises
ValueErrorr r" r& r* )rO messages r! test_root_scalar_failz TestNewton.test_root_scalar_fail s 8
]]:W
5 K 8DQT J K7
]]:W
5 L 8Tad K L L K K L Ls A> B
>B
Bc d }d }d }t j g d }t j t d dz dz }||dd dd
f}dgdz }t j
|||| }d } t
|| t j
||||| }t
|| t j
|||
}t
|| y)ztest newton with arrayc |d | |d z z }|d |d t j ||d z dz
z z
||d z z
| z
S )Nr r r r r r r r r r r> r? s r! r" z(TestNewton.test_array_newton..f1 sW !q1Q4xAQ4!A$"&&QqT"2S"899A!HDqH Hr# c |d |d z }|d t j |d |d z | |z z z |z |d |d z z
dz
S )Nr r r r r r r r s r! r& z*TestNewton.test_array_newton..f1_1 s] !qt AaD5266!A$1+A"566:QqTAaD[H1L Lr# c z |d |d z }|d t j |d |d z | |z z z |dz z S )Nr r r r r r s r! r* z*TestNewton.test_array_newton..f1_2 sK !qt AaD5266!A$1+A"566A= =r# )
g
4O@gNk@g]0J@g]Qݚt@g~EO5@g$J ?g~
5,@gXCڭ@gͮ9@@gӍ@
r g @g& .>gMbp?gn2d?)
gԩ˰@g9~4b'@gaq(@g0p @gk"z?g4se?gU+F@gwQu%@gd6)@g)i!@)r` rE N)r arrayr rangerL r r )
rO r" r& r* a0a1rE r; r
x_expecteds
r! test_array_newtonz TestNewton.test_array_newton s I M > XX
ffU2Y #%
,Bub'2URZ LLRt ,
:& LLRtT ::& LLRd +:&r# c ^ d }d }t j dd }t j ||| }t || d t j
d }t j ||| }t || d t j || }t || d y )Nc | dz dz S )Nr ?r r s r! r= z/TestNewton.test_array_newton_complex..f7 s q58Or# c yNr r r s r! r_ z4TestNewton.test_array_newton_complex..fprime: s r# r r )r_ )r fullrL r r ones)rO r= r_ tr s r! test_array_newton_complexz$TestNewton.test_array_newton_complex6 s
GGArN LLAf -!b!
GGAJ LLAf -!b! LLA !b!r# c ~ t j d ddgt j ddg g }t |d y) z8test secant doesn't continue to iterate zero derivativesc | | z |d z
S )Nr r r r> s r! z>TestNewton.test_array_secant_active_zero_der..K s qsQqTz r# gˡE}@r )r; rE )f~@g @N)rL r r r r rO r s r! !test_array_secant_active_zero_derz,TestNewton.test_array_secant_active_zero_derI s7 LL1uaj !xxR 1 2
434r# c t j d dgdz ddgf }t |d t j d d gdz d
d gf }t |d y ) Nc || dz z
S r% r yzs r! r z7TestNewton.test_array_newton_integers..Q a!q&j r# @r g .@g 1@r )gNO@r c || dz z
S r% r r s r! r z7TestNewton.test_array_newton_integers..U r r# r r )rL r r r s r! test_array_newton_integersz%TestNewton.test_array_newton_integersO sY LL03%!) $d|o
/AB LL01#'"b LABr# c t t t j d ddgd t j
t 5 t j d ddgd d }t
|j d |j j sJ |j j rJ d d d y # 1 sw Y y xY w) Nc | dz dz
S r% r r s r! r z@TestNewton.test_array_newton_zero_der_failures..[ s q!tax r# r c d| z S r% r r s r! r z@TestNewton.test_array_newton_zero_der_failures..[ s
QU r# c | dz dz
S r% r r s r! r z@TestNewton.test_array_newton_zero_der_failures..^ s QTAX r# c d| z S r% r r s r! r z@TestNewton.test_array_newton_zero_der_failures.._ s
QqS r# Tr r ) r RuntimeWarningrL r r warnsr rD zero_derallrl any)rO ru s r! #test_array_newton_zero_der_failuresz.TestNewton.test_array_newton_zero_der_failuresX s ^U\\'"b? D \\.
) /ll#5Bx#0dDG GLL! ,##'') )) ((,,. ... / / /s A,B55B>c d }d }d }d }t |dd| }t |ddd }t |j |j d
t d |j z |j t |d d||
}t |d dd }t |j |j d
t d|j z |j y )Nc | dz d| z z
dz
S r r r s r! r" z+TestNewton.test_newton_combined..f1e s 6AE>A% %r# c d| z dz
S r% r r s r! r& z-TestNewton.test_newton_combined..f1_1g s q519 r# c dd| z z S r( r r s r! r* z-TestNewton.test_newton_combined..f1_2i s Q; r# c 0 | dz d| z z
dz
d| z dz
dfS )Nr r r) r r s r! r- z8TestNewton.test_newton_combined..f1_and_p_and_ppl s' a4!A#:a<1Q+ +r# r r )rQ r; r_ T:0yE>r r r^ )rQ r; r_ r` )rQ r; r` )r r rD r function_calls)rO r" r& r* r- sol0sols r! test_newton_combinedz TestNewton.test_newton_combinedd s & , 2h1TB/(qN 388$7Qs)))4+>+>?2h1T4P/(q$O 388$7Qs)))4+>+>?r# c d}g d}t d D ] }ddd}dt gdt ggd | D ]
\ }}|||< t j t
|fdd i|\ }}t
|j t ||j t |j |j f|| |d
k( r |j |j d z k s(J t |j |d z |j z |j d z
} t j t
|f| d d |\ }}t
|j t ||j t |j | |d k( s[t j t d
| z 5 t j t
|f| dd |\ }}d d d y # 1 sw Y xY w)Nr )) )r r )r r T)ra rB r_ r` rC Fr r )maxiterrC z3Failed to converge after %d iterations, value is .*r )r r& r* rL r r" r rl r rD
iterationsr r r RuntimeError)
rO r; expected_countsderivsrT rV vr rY iterss
r! test_newton_full_outputz"TestNewton.test_newton_full_outputy s
3Ah TF !$9F"D)It+<=gvF
1 q
<<B=U=f=DAq AKK AFF # !,,(8(89?6;R S{''1<.func s 6C< r# c d| z S r% r r s r! dfuncz1TestNewton.test_deriv_zero_warning..dfunc s q5Lr# r FrC zDerivative was zeror )r r rL r r r r )rO r@ r! s r! test_deriv_zero_warningz"TestNewton.test_deriv_zero_warning sV ^U\\4e%P
]].f s b519 r# r )r; rQ g;f?)r rl r absrD dtyper float64)rO rQ r= ress r! test_int_x0_gh19280z TestNewton.test_int_x0_gh19280 sX
!&1}} }CHH
w/xx~~"**!5555r# N)r r r r r r r r r r r r r r r r r r# r( r. r r r r6 r r# r! r r s EE 05 55L%'N"&5 C
/@*#TJ+(8: [[ X(';< 6 = 6r# r c dfd} t j t j g}t x}}|D ], } || dd|| }t |||d|j
z . y )Nr% c | z
S r, r )r rD s r! r= ztest_gh_5555..f s 4xr# g חg cAre z method %s)rf rc err_msg)rL r r TOLr r )r= methodsrb rc rQ r5 rD s @r! test_gh_5555r< sh D ||U\\*GD4 ?Qc48c4 +foo = ??r# c d } d}dt z }t j t j g}|D ] } || dd|| }t d||| y ) Nc | dk ry| dz
S )Nr g333333?r r s r! r= ztest_gh_5557..f s s7s7Nr# gR Q?r r r re r? r )
_FLOAT_EPSrL brentqbrenthr )r= rf rc r; rQ r5 s r! test_gh_5557rC sX D z>D||U\\*G 8Q14d3St$78r# c d} d}| |fD ]Y \ }}}t j | t j t j fD ] } |fd|| }t || [ y )N)g |g ug y)g u@g |@g y@c 4 t j | z
S r, r )r r} s r! r z9test_brent_underflow_in_root_bracketing.. s 266!9Q; r# )r r rL rB rA r )underflow_scenariooverflow_scenarior> r? rD rQ r5 r} s @r! 'test_brent_underflow_in_root_bracketingrH sn
2-)+<= '
1d FF4L ||U\\2 'F.15C D# & ''r# c D e Zd Z ej dddd Zd Zd Zy) TestRootResultsr , . r )rD r r flagc F d}t t | j | y )Nzm converged: True
flag: converged
function_calls: 46
iterations: 44
root: 1.0)r reprrY )rO
expected_reprs r! test_reprzTestRootResults.test_repr s 0
T$&&\=1r# c < t | j t sJ y r, )
isinstancerY r )rO s r! test_typezTestRootResults.test_type s $&&.111r# N)r r r rL rM rY rQ rT r r# r! rJ rJ s$ sr"1MA2 2r# rJ c d } d }d }t dd }d}t j | ||||d }t | |g| d d
|gd z }d}t j | ||||d }t | |g| d d
y )
z&Test Halley's works with complex rootsc 6 |d | dz z |d | z z |d z S )Nr r r r r s r! r= z test_complex_halley..f s* tad{QqTAX%!,,r# c $ d|d z | z |d z S )Nr r r r r s r! r z test_complex_halley..f_1! s 1Q4x!|ad""r# c X d|d z } t | }|g|z S # t $ r |cY S w xY w)Nr r )ri TypeError)r r> retvalsizes r! r z test_complex_halley..f_2$ sA QqT #q6D 8d? " M s ))r r) )r) g @r r )rE r_ r` ra r r r N)r rL r r )r= r r r coeffsr s r! test_complex_halleyr] s -## SA
F
QsCTJAAaM&M140
bA
F
QsCTJAAaM&M140r# c t j t j dz } d| z
d| z z }t 5 }|j t d t j d |gdz }ddd t d gdz d
}t 5 }|j t d t j d |d
}ddd t |d t j t d 5 t j d |d }ddd d}t 5 }|j t d t j d |d
}ddd t |d t j t d 5 t j d |d }ddd y# 1 sw Y 5xY w# 1 sw Y xY w# 1 sw Y xY w# 1 sw Y xxY w# 1 sw Y yxY w)zBTest secant method with a non-zero dp, but an infinite newton stepg Q ?g i@r) zRMS ofc | dz
dz S )Ng Y@r r r s r! r z%test_zero_der_nz_dp..E s AI> r# r r; Nd g.ЗK.? Tolerance ofc | dz
dz S Nr r r r s r! r z%test_zero_der_nz_dp..K AG> r# F)r; rC r r c | dz
dz S rd r r s r! r z%test_zero_der_nz_dp..N re r# Tg.ЗK.c | dz dz S rd r r s r! r z%test_zero_der_nz_dp..R re r# rF c | dz dz S rd r r s r! r z%test_zero_der_nz_dp..U re r# )
r r
r r r filterr rL r r r r r )dxp0supr s r! test_zero_der_nz_dprm 8 s
% $B "*r "B A
>8, LL1rdRi @A Aurz" $B F
>>2 LL1bu EF Aq |> : E LL1bt DE %B F
>>2 LL1bu EF Ar |> : E LL1bt DE E#A A F FE EF FE Es; 3F"0F/2F; 0G?G"F,/F8;GGG c d} dd}d}d}||z | z |z }fd}t j t 5 t j |g dd|| gd
}|j
j
rJ d d d t j t 5 t j |d gdz d|| gd
}d d d y # 1 sw Y LxY w# 1 sw Y y xY w)
z(Test that array newton fails as expectedr% ga2U0*#?g@g\mJ A?gCl@c dt j | z dt j dz |z d|z t j | z z z z S )Nr r g
@gGz@)r r log10)darcy_frictionredia roughnesss r!
colebrook_eqnz1test_array_newton_failures..colebrook_eqne s[ BGGN++BHHY_s 2 !BY )@ @ A B BB Cr# ){Gz?g?g vÖ?g333333?r T)r; r rE rB Nrv ) r r r rL r rl r r r )diameterrhomuureynolds_numberru resultrt s @r! test_array_newton_failuresr} X s
HI
C B
AAg(2-OC
n % * 7!8,$
##'')))) *
| $
tfqj!!8,$
* *
s ;B<!C<CCc f d } t j | d }t |dt j t j t j | dgdz }t |dt j t j d }d }t j | d| }t |dt j t j t j | d|| }t |dt j t j t j | dgdz | }t |dt j t j t j | dgdz || }t |dt j t j t j | d
| }t |dt j t j t j | d
gdz | }t |dt j t j y ) z@Test that Newton or Halley don't warn if zero derivative at rootc | dz | dz z
S Nr r r r s r! f_zeroder_rootz9test_gh8904_zeroder_at_root_fails..f_zeroder_root~ s !tad{r# r r` r r c d| dz z d| z z
S r r r s r! fderz/test_gh8904_zeroder_at_root_fails..fder s 1a4x!a% r# c d| z dz
S )Nr r r r s r! fder2z0test_gh8904_zeroder_at_root_fails..fder2 sQwr# )r; r_ )r; r_ r` r N)rL r r _xtol_rtol)r rY r r s r! !test_gh8904_zeroder_at_root_failsr z su ^*AAqu{{=
^B/AAqu{{= ^$7AAqu{{=
^$ " $AAqu{{=
^BtAAqu{{=r# c dfd} fd}fd}d}t | ||d \ }}|j sJ t | |||d \ }}|j sJ y )
zzTest that Halley's method realizes that the 2nd order adjustment
is too big and drops off to the 1st order adjustment.r c B t | dz t dz z
S r r r ns r! r= ztest_gh_8881..f s" QAq#a% 00r# c . t | dz
z z S r r r s r! fpztest_gh_8881..fp s QQ "1$$r# c L t | ddz z
z dz z dz
z z S rd r r s r! fppztest_gh_8881..fpp s3 QQqS! $A.#a%8::r# r% T)r_ rB r_ r` rB N)r rl )r= r r r; rtrY r s @r! test_gh_8881r sf
A1%;
B
1b 6EB ;;;
1bSd CEB ;;;r# c d } d }d }t j dgt j }t | |||d \ }}|j sJ t j ddgt j }t j t 5 t j | |||d }d d d d
}t j | |||d }|j j sJ y # 1 sw Y CxY w) z_
Test that shape is preserved for array inputs even if fprime or fprime2 is
scalar
c | dz S r% r r s r! r= z,test_gh_9608_preserve_array_shape..f s !t r# c d| z S r% r r s r! r z-test_gh_9608_preserve_array_shape..fp 1u r# c yr% r r s r! r z.test_gh_9608_preserve_array_shape..fpp s r# r1 r3 Tr Nc t t j t j | dt j S )Nr r )r r r* float32r s r! fpp_arrayz4test_gh_9608_preserve_array_shape..fpp_array s! wwrxx{ARZZ88r# )
r r r r rl r r
IndexErrorrL r ) r= r r r; r rY x0_arrayr| r s r! !test_gh_9608_preserve_array_shaper s
2$bjj )B 1bSd CEB ;;;xxR
3H z "
xCT
9 \\ 8B tF !! !
s C..C7z maximum_iterations,flag_expectedr ra c
t j d ddddd| dd }|d j |k( sJ |t j k( r|d j | k( sJ y
|t j
k( r|d j | k sJ y
y
) z]
Test that if the maximum iterations is exceeded that the flag is not
converged.
c * d| z dz
| z dz | z dz
S )Ng333333?gffffff@g333333 @g @r r s r! r z6test_gh9254_flag_if_maxiter_exceeded.. s! CECK?S(!+c1 r# i r r TFrA r N)rL rA rM CONVERRr CONVERGED)maximum_iterations
flag_expectedr| s r! $test_gh9254_flag_if_maxiter_exceededr s \\1 RT4!3u&F !9>>] ** *
%ay##'9999 %// )ay##&8888
*r# c R d } d }t t t j | d|d t j
t d 5 t j | d| ddd t j | t d d | }t |t d
d y# 1 sw Y AxY w) zBTest that if disp is true then zero derivative raises RuntimeErrorc | | z dz S Nr r r s r! r= z/test_gh9551_raise_error_if_disp_true..f r r# c d| z S r% r r s r! f_pz1test_gh9551_raise_error_if_disp_true..f_p s s
r# r Fr" zY^Derivative was zero\. Failed to converge after \d+ iterations, value is [+-]?\d*\.\d+\.$r Ng $@r ) r r rL r r r r r r )r= r rD s r! $test_gh9551_raise_error_if_disp_truer s q#sG n
p " QS!" <<74. 4DD'#s+, " "s B B& solver_name)rA rB r r r c d }t t | }t j t d 5 ||dd d d d y # 1 sw Y y xY w)Nc " t j S r, )r r r s r! r= ztest_gh3089_8394..f s vv
r# zThe function value at x...r r r )getattrrL r r r )r r= solvers r! test_gh3089_8394r
sD
UK
(F z)E F q!Q s AA rQ c fdd_ t d| }|j du sJ |j j d sJ |j
j k( sJ t
|j |j v sJ y )Nc N xj dz
c_ t j S r )_countr r r r= s r! r= ztest_gh18171..f s A
vv
r# r )r r )r rQ FzThe function value at x)r r rl rM
startswithr strrD )rQ r5 r= s @r! test_gh18171r s| AH
a
7C ==E !! ! 88 8 99 9 )) ) sxx=CHH $$ $r# rs_interfaceTFc |rd nt t | }fdd_ |ddd }|r |j j k( sJ y |d j j k( sJ y )Nc t | ||f S N)r r r= r> r? rT s r! r z%test_function_calls..2 QA)G r# c > xj dz
c_ | dz dz
S )Nr r )callsr s r! r= z test_function_calls..f5 s 1 !taxr# r r Tr r )r rL r r )r r r r5 r= s @r! test_function_callsr , sr H#*5+#> AG
Art
,C!!QWW,,,1v$$///r# c d } t j t d 5 t | ddd }ddd j rJ |j
d k( sJ t j t d 5 t
| ddd
d d
}ddd |j rJ |j
d k( sJ y# 1 sw Y xY w# 1 sw Y 5xY w)zDTest that zero slope with secant method results in a converged=Falsec @ | t j | | z z dz
S )NgQ ?r r s r! lhsz*test_gh_14486_converged_false..lhsD s 2661"Q$< $&&r# rb r r] g333333ÿr )rQ r; r< Nzconvergence errorFT)r; r< rC rB r )r r r r rl rM r )r r5 s r! test_gh_14486_converged_falser B s '
nN ; B#h5SAB}} 88* ** * nN ; MSUsDI!LM}} 88* ** *B B
M Ms B-1B9-B69Cc |rd nt t | }d }t j t d 5 ||ddd d d d ||dd d }|r|n|d
}|j
sJ t
|j d d
||dt d d }|r|n|d
}|j
sJ t
|j d d
y # 1 sw Y xY w)Nc t | ||f S r r r s r! r z test_gh5584..X r r# c d| z S )NgN~hr r s r! r= ztest_gh5584..f[ s axr# z ...must have different signsr g gٿTr g?r r r
r z-0.0) r rL r r r rl r rD r )r r r r= r5 s r! test_gh5584r R s H#*5+#>
z)G H 0q$$/0 D#4
0C #3q6C ===CHHad+ D%-T
:C #3q6C ===CHHad+0 0s
C Cc d } d}t j t j }t j
| dd|d|z } | | }t j
| dd|d|z } | | }||k sJ d|d z d
d |d
d }t
j t |
5 t j
| dd||d z d d d y # 1 sw Y y xY w)Nc | dz d| z z
dz
S )Nr r r r r s r! r= ztest_gh13407..fu s !taczA~ r# gY ng|=g _Br re r zrtol too small \(r gz < z\)r ) r r
r r rL r r r r )r= rb r r< r" x4f4r s r! test_gh13407r o s D
((5/
C q%Dqu =B
2B q%Dqu =B
2B
7N7 #3q5)3s1gR8G z 1 =
a4c!e<= = =s # C
Cc d } t | d }t |dd t | ddd }t |j dd y ) Nc | dz
S r r )r s r! r= z&test_newton_complex_gh10103..f r r# y ? ?r g-q=r y @ ?r] )r; r< rQ )r r r rD )r= r5 s r! test_newton_complex_gh10103r s>
D/CC'
aDVH
=CCHHae,r# c d}t j t | 5 | t ddd d d d y # 1 sw Y y xY w)Nz2'float' object cannot be interpreted as an integerr r r g R@)r )r r rY r" )rQ r s r! test_maxiter_int_check_gh10236r s; CG y 0 ,r3U+, , ,s 8A)Pr functoolsr
numpy.testingr r r r r r numpyr r
r r r
r r r r scipy.optimizer rL r r r scipy._lib._utilr rg scipy.optimize._tstutilsr r r r r r: r@ r r rA rB r r gradient_methods all_methodsr" r& r* r- r0 r3 r5 r7 r9 r r r< rC rH rJ r] rm r} r r r r r r r r r r r r r r r r r r r# r! r s
. .
A A A, , G P
5\
<< u||U\\=="LL> 00 #
YM YMx-!. -!`G6& G6T ? 80 '
2
218 E@
D,>`4 "D &%--3013939$-$ LNN LN
%N
% LN $ 70 8N0&
+ LN $ 7, 8N,4=.
- ;/, 0,r#