o `^h|@s<ddlZddlZddlZddlZddlmZddlmZm Z ddl Z ddl m Z m Z ddlmZmZmZmZmZmZmZmZddlmZddlmZGdd d Zd d ZGd d d eZeddgddgdZeddgddgdZeddgddgdZ GdddeZ!e!dgdgde"dgdgdgdgdgdgdgdgdgdgg e"gd d!Z#e!d"gdgdge"dgge"dgd!Z$Gd#d$d$eZ%e%d%d&gd'd(gd)d*Z&Gd+d,d,eZ'e'd-gd.gd/d0d*Z(Gd1d2d2eZ)e)d3d3gd4gd5d6gd7Z*Gd8d9d9eZ+d:Z,e-e.d;gd:dd?d?eZ1e1d@gdAdBe2dAdCd7Z3GdDdEdEeZ4e4d-d-gdFgdGdGgd7Z5GdHdIdIeZ6e6d%dJgddd7Z7e j89dK   PdfdQdRZ:GdSdTdTZ;GdUdVdVZGd[d\d\Z?d]d^Z@e j8Ad_d`daZBdbdcZCdddeZDdS)gN)Pool)assert_allcloseIS_PYPY)raiseswarns)shgoBoundsminimize_scalarminimizerosen rosen_der rosen_hessNonlinearConstraint)new_constraint_to_old)SHGOc@seZdZ  dddZdS)StructTestFunctionNcCs"||_||_||_||_||_dSNbounds expected_x expected_fun expected_xl expected_funl)selfrrrrrr]/home/air/shanriGPT/back/venv/lib/python3.10/site-packages/scipy/optimize/tests/test__shgo.py__init__s  zStructTestFunction.__init__)NNN)__name__ __module__ __qualname__rrrrrrsrcCsRg}|dur%t|ttfs|f}n |D] }|d|dqt|}|Sd}|S)Nineqtypefun) isinstancetuplelistappend)gconsrrrwrap_constraintss r*c@s$eZdZddZddZeeZdS) StructTest1cC|dd|ddSNrrrxrrrf+z StructTest1.fcCstj|ddd S)Nraxisg@npsumr1rrrr(.z StructTest1.gNrrrr2r(r*r)rrrrr+*s r+))rrrr/NNc@s(eZdZdZddZddZeeZdS) StructTest2zN Scalar function with several minima to test all minimiser retrievals cCs|dt|S)N)r7sinr0rrrr2Az StructTest2.fcCsdtj|ddS)N:rr4r6r9rrrr(DrCz StructTest2.gNrrr__doc__r2r(r*r)rrrrr@<s  r@)r<g⡽4$?0_rE@gqŤCQ4@g8$B@go.n:@g^5 >@) rHgs8g^V&+6g+?-2g+!/gN|1(gc=C#g^\gN g自ݿr)rg@c@sDeZdZdZddZddZddZdd Zeed e j Z e fZ d S) StructTest3a Hock and Schittkowski 18 problem (HS18). Hoch and Schittkowski (1981) http://www.ai7.uni-bayreuth.de/test_problem_coll.pdf Minimize: f = 0.01 * (x_1)**2 + (x_2)**2 Subject to: x_1 * x_2 - 25.0 >= 0, (x_1)**2 + (x_2)**2 - 25.0 >= 0, 2 <= x_1 <= 50, 0 <= x_2 <= 50. Approx. Answer: f([(250)**0.5 , (2.5)**0.5]) = 5.0 cCsd|dd|ddS)Ng{Gz?rr.r/rr0rrrr2{z StructTest3.fcCs|d|ddS)Nrr/9@rr9rrrg1~r:zStructTest3.g1cCs|dd|dddS)Nrr.r/rKrr9rrrg2rJzStructTest3.g2cCs0|d|dd|dd|dddfS)Nrr/rKr.rr9rrrr(0z StructTest3.grN) rrrrFr2rLrMr(rr7inf_StructTest3__nlcr)rrrrrIis rI)r.2)rrQg'In/@gS[:XL?@)rrrc@LeZdZdZddZddZddZdd Zd d ZeeeefZ e e Z d S) StructTest4z Hock and Schittkowski 11 problem (HS11). Hoch and Schittkowski (1981) NOTE: Did not find in original reference to HS collection, refer to Henderson (2015) problem 7 instead. 02.03.2016 cCs|dddd|ddd|ddd|dd dd|dd d |dd|d dd|d|d d|dd |d S) Nr r.r/  r=rr0rrrr2s&.  z StructTest4.fcCsJd|ddd|dd|dd|ddd|dd S)Nr.rrYr/rXrVrr9rrrrLs 6 zStructTest4.g1cCs>d|dd|dd|dd|d|dd S) Nr[rrYr/rUr.rXgq@rr9rrrrMs>zStructTest4.g2cCs:d|d|ddd|ddd|dd S) Nrr/r.r=rVr\rr9rrrg3s:zStructTest4.g3cCsZd|dd|ddd|d|dd|ddd|dd|d S) NrXrr.r/rYrVrZr=rr9rrrg4s >  zStructTest4.g4N rrrrFr2rLrMr`rar(r*r)rrrrrTs  rT)irUr[)gܤ@g$H8?g ޿g0ev@gn8)gk_@/?g0g+?gP{[ E@c@ eZdZddZdZeeZdS) StructTest5c Csb|dd ttt|dd|dd|dttt|d|ddSNr/gG@r@r7rBsqrtabsr0rrrr24*z StructTest5.fNr;rrrrrd rd)iig g@gyCy@)rrrc@s$eZdZdZddZdZeeZdS) StructTestLJzZ LennardJones objective function. Used to test symmetry constraints settings. cGstd||d|_t|jd}d}t|dD]T}t|d|D]J}d|}d|}||||} ||d||d} ||d||d} | | | | | | } | | | } | dkrn|d| d| 7}q$q|S) Nzx = rrYr/r.?rf)printNintrange)rr1argsksijabxdydzdedudrrrr2s$   zStructTestLJ.fNrErrrrrls  rlr=g@c@rc) StructTestScCs@|ddd|ddd|ddd|dddS)Nr?r.r/rYrr0rrrr2s z StructTestS.fNr;rrrrrs r)rrfrXrmrc@rc)StructTestTablecCs |ddkr|ddkrdSdS)Nr@r/rQdrr0rrrr2 szStructTestTable.fNr;rrrrr rkrrQrc@rS) StructTestInfeasiblez0 Test function with no feasible domain. cGr,r-r)rr1rsrrrr2r3zStructTestInfeasible.fcCs|d|ddSNrr/rr9rrrrL!r:zStructTestInfeasible.g1cCs|d|dd Srrr9rrrrM$zStructTestInfeasible.g2cCs|d |ddSrrr9rrrr`'rzStructTestInfeasible.g3cCs|d |dd Srrr9rrrra*r3zStructTestInfeasible.g4Nrbrrrrrs  r)r<r/ Not a testrh㈵>rsobolr/c Cst|j|j||j||||||| d } td| td| |jdur2tj j | j |j||d|j durBtj j | j |j |d|jdurRtj j | j|j|d|jdurbtj j | j|j|ddS)N) rs constraintsniterscallbackminimizer_kwargsoptionssampling_methodworkerszres = rtolatolr)rr2rr)rologginginforr7testingrr1rr#rxlrfunl) testrs test_atolrrrrrrrresrrrrun_test7s8       rc@s`eZdZdZddZddZddZdd Zd d Zd d Z e j j ddZ ddZddZdS)TestShgoSobolTestFunctionsz8 Global optimisation tests with Sobol sampling: cC ttdS)VMultivariate test function 1: x[0]**2 + x[1]**2 with bounds=[(-1, 6), (-1, 6)]Nrtest1_1rrrrtest_f1_1_sobol` z*TestShgoSobolTestFunctions.test_f1_1_sobolcCr)zUMultivariate test function 1: x[0]**2 + x[1]**2 with bounds=[(0, 1), (0, 1)]Nrtest1_2rrrrtest_f1_2_sobolerz*TestShgoSobolTestFunctions.test_f1_2_sobolcCsddi}tt|ddS)z_Multivariate test function 1: x[0]**2 + x[1]**2 with bounds=[(None, None),(None, None)]dispT)rNrtest1_3rrrrrtest_f1_3_soboljsz*TestShgoSobolTestFunctions.test_f1_3_sobolcCr)RUnivariate test function on f(x) = (x - 30) * sin(x) with bounds=[(0, 60)]Nrtest2_1rrrrtest_f2_1_sobolprz*TestShgoSobolTestFunctions.test_f2_1_sobolcCr)NUnivariate test function on f(x) = (x - 30) * sin(x) bounds=[(0, 4.5)]Nrtest2_2rrrrtest_f2_2_sobolurz*TestShgoSobolTestFunctions.test_f2_2_sobolcCr)%NLP: Hock and Schittkowski problem 18Nrtest3_1rrrr test_f3_sobolzs z(TestShgoSobolTestFunctions.test_f3_sobolcCddi}ttd|ddS)?NLP: (High dimensional) Hock and Schittkowski 11 problem (HS11)infty_constraintsFirrNrtest4_1rrrr test_f4_sobol~sz(TestShgoSobolTestFunctions.test_f4_sobolcCsttdddS)NLP: Eggholder, multimodalrG)rNrtest5_1rrrrtest_f5_1_sobolsz*TestShgoSobolTestFunctions.test_f5_1_sobolcCttddddS)rrGrV)rrNrrrrrtest_f5_2_sobolz*TestShgoSobolTestFunctions.test_f5_2_sobolN)rrrrFrrrrrrpytestmarkslowrrrrrrrrZs  rc@seZdZdZddZddZddZdd Zd d Zd d Z e j j ddZ ddZddZddZe j dddZe j dddZe j dddZdS)TestShgoSimplicialTestFunctionsz= Global optimisation tests with Simplicial sampling: cCr)rr/ simplicialrrNrrrrrtest_f1_1_simplicialrz4TestShgoSimplicialTestFunctions.test_f1_1_simplicialcCr)zTMultivariate test function 1: x[0]**2 + x[1]**2 with bounds=[(0, 1), (0, 1)]r/rrNrrrrrtest_f1_2_simplicialrz4TestShgoSimplicialTestFunctions.test_f1_2_simplicialcCr)z_Multivariate test function 1: x[0]**2 + x[1]**2 with bounds=[(None, None),(None, None)]rVrrNrrrrrtest_f1_3_simplicialrz4TestShgoSimplicialTestFunctions.test_f1_3_simplicialcCddi}ttdd|dddS)rminimize_every_iterFr[rrrrrNrrrrrtest_f2_1_simplicials  z4TestShgoSimplicialTestFunctions.test_f2_1_simplicialcCr)rr/rrNrrrrrtest_f2_2_simplicialrz4TestShgoSimplicialTestFunctions.test_f2_2_simplicialcCr)rr/rrNrrrrrtest_f3_simplicialsz2TestShgoSimplicialTestFunctions.test_f3_simplicialcCr)rr/rrNrrrrrtest_f4_simplicialrz2TestShgoSimplicialTestFunctions.test_f4_simplicialcCs&ddd}d}tt|d|ddddS) z&LJ: Symmetry-constrained test functionTsymmetryrr=,r/rrsrrrrNrtestLJrrrsrrrtest_lj_symmetry_olds z4TestShgoSimplicialTestFunctions.test_lj_symmetry_oldcCs,dgddd}d}tt|d|ddd d S) z&LJ: Symmetry constrained test functionrr=Trrrr/rrNrrrrrtest_f5_1_lj_symmetrys z5TestShgoSimplicialTestFunctions.test_f5_1_lj_symmetrycCs$ddgdd}ttd|ddddS) z"Symmetry constrained test functionrTrrr/rrrrrNrrrrrtest_f5_2_cons_symmetrys z7TestShgoSimplicialTestFunctions.test_f5_2_cons_symmetryrUcCs$gddd}ttd|ddddS) z(Asymmetrically constrained test function)rrrrYTri'r/rrN)rtest_srrrrtest_f5_3_cons_symmetrys z7TestShgoSimplicialTestFunctions.test_f5_3_cons_symmetryrcsVddfddd}ttjddg|d}|jsJt|jd d d t|jdd S) zOReturn a minimum on a perfectly symmetric problem, based on gh10429reqcst|Sr)r7meanr9avgrrszFTestShgoSimplicialTestFunctions.test_f0_min_variance..r!r=r>rrrV瞯.funr>)rrư>)rN)rr rrr#r1)rr#rrrefrrrtest_f0_min_variance_1Ds  z7TestShgoSimplicialTestFunctions.test_f0_min_variance_1DN)rrrrFrrrrrrrrrrrrr fail_slowrskiprrrrrrrs&     rc@sreZdZddZddZddZddZd d Zd d Zd dZ e j j ddZ e j j ddZddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Ze j jeoaejd'kd(d)d*d+Zd,d-Ze j jd.d)d/d0Z d1d2Z!d3d4Z"d5d6Z#d7d8Z$d9d:Z%d;d<Z&e j 'd=d>d?Z(d@dAZ)dBdCZ*e j +dDgdEdFdGZ,dHdIZ-dJdKZ.dLS)MTestShgoArgumentscCttdddddS)z>Iterative simplicial sampling on TestFunction 1 (multivariate)Nr.rrrrrrrrrtest_1_1_simpl_iterz%TestShgoArguments.test_1_1_simpl_itercCsddi}ttdd|dddS)z3Iterative simplicial on TestFunction 2 (univariate)rFN rrrrrrrtest_1_2_simpl_iter  z%TestShgoArguments.test_1_2_simpl_itercCr)z9Iterative Sobol sampling on TestFunction 1 (multivariate)Nr/rrrrrrrtest_2_1_sobol_iter rz%TestShgoArguments.test_2_1_sobol_itercCsLttjtjtjdddd}tjj|jtj dddtjj|j tj dddS)z7Iterative Sobol sampling on TestFunction 2 (univariate)Nr/r)rrrrrrr) rrr2rr)r7rrr1rr#rrrrrrtest_2_2_sobol_iters z%TestShgoArguments.test_2_2_sobol_iterc CsRdd}ttfD]}t|j|jdd|ddidt|j|jdd|ddidqd S) zKIterative sampling on TestFunction 1 and 2 (multi and univariate) cS tddSNz Local minimization callback testror9rrr callback_func zATestShgoArguments.test_3_1_disp_simplicial..callback_funcr/rrTrrrrrrrrNrrrr2rrrrrrrtest_3_1_disp_simplicials  z*TestShgoArguments.test_3_1_disp_simplicialc CsRdd}ttfD]}t|j|jdd|ddidt|j|jdd|ddid qd S) zAIterative sampling on TestFunction 1 and 2 (multi and univariate)cSrrrr9rrrr)rz.callback_funcr/rrTr rr Nr r rrrtest_3_2_disp_sobol&s z%TestShgoArguments.test_3_2_disp_sobolcCsFtdddgdd}tdddgd}t|j|jt|j|jdS) zMUsing `args` used to cause `shgo` to fail; see #14589, #15986, #16506cSs |||Srr)r1yzrrrr6 z5TestShgoArguments.test_args_gh14589..)rrY)r/r.)funcrrscSs d|dS)Nr.r/rr9rrrr8r)rrN)rrr#r1)rrrrrrtest_args_gh145893sz#TestShgoArguments.test_args_gh14589cCs$tjddd}ttdd|dddS)z,Test known function minima stopping criteriarT)f_minf_tolrNrrrrrrrrrrrrrtest_4_1_known_f_min<s  z&TestShgoArguments.test_4_1_known_f_mincCs&tjdddd}ttdd|dddS) z+Test Global mode limiting local evaluationsrTr/)rrr local_iterNrrrrrrrrtest_4_2_known_f_minGs  z&TestShgoArguments.test_4_2_known_f_minc CsJtjddddd}ttjtjtjdd|dd}tjj|j tj d d d dS) z8Test Global mode limiting local evaluations for 1D funcsrTr/FrrrrrNrrrrrrrr) rrrr2rr)r7rrr1rrrrrrrtest_4_4_known_f_minTsz&TestShgoArguments.test_4_4_known_f_mincCs0ttjtjtjd}tjj|jtj ddddS)z;Test Default simplicial sampling settings on TestFunction 1rrrN rrr2rr)r7rrr1rrrrrtest_5_1_simplicial_arglesscsz-TestShgoArguments.test_5_1_simplicial_arglesscCs2ttjtjtjdd}tjj|jtj ddddS)z6Test Default sobol sampling settings on TestFunction 1rrrrrNr rrrrtest_5_2_sobol_arglesshsz(TestShgoArguments.test_5_2_sobol_arglesscCRddi}ttjtjtj|dd}tjj|jtj dddtjj|j tj dddS) :Test that maximum iteration option works on TestFunction 3max_iterr.rrrrrrrN rrr2rr)r7rrr1rr#rrrrrtest_6_1_simplicial_max_itern z.TestShgoArguments.test_6_1_simplicial_max_itercCr$) r%min_iterr.rr'rrrNr(rrrrtest_6_2_simplicial_min_itervr*z.TestShgoArguments.test_6_2_simplicial_min_itercCs,dD]}|tjd}ttdd|ddqdS)z@Test the minimizer_kwargs arguments for solvers with constraints)COBYLACOBYQASLSQP)methodrrMbP?rrrrrN)rr)r)rsolverrrrrtest_7_1_minkwargs~sz$TestShgoArguments.test_7_1_minkwargscCs0ddi}ddi}ttjtjtjd||ddS)z'Test the minimizer_kwargs default initsftolrrTr)rrrNrrr2rr))rrrrrrtest_7_2_minkwargss  z$TestShgoArguments.test_7_2_minkwargscCsVdD]&}dd}dd}|||d}td|tdttd d |d d qd S)z?Test minimizer_kwargs arguments for solvers without constraints) z Nelder-MeadPowellCGBFGS Newton-CGzL-BFGS-BTNCdoglegz trust-ncgz trust-exactz trust-krylovcSs td|dd|dgjS)Nr.rr/)r7arrayTr9rrrjacs z1TestShgoArguments.test_7_3_minkwargs..jaccSstddgddggSNr.r)r7r>r9rrrhessrz2TestShgoArguments.test_7_3_minkwargs..hess)r0r@rBz Solver = zd====================================================================================================rr1rr2N)rrrr)rr3r@rBrrrrtest_7_3_minkwargss z$TestShgoArguments.test_7_3_minkwargscCs ddd}ttdd|dddS)Nr/T)minhgrdrrrrrrrrtest_8_homology_group_diffs   z,TestShgoArguments.test_8_homology_group_diffcCsttjtjtjdddS)'Test single function constraint passingrrNr6rrrr test_9_cons_gszTestShgoArguments.test_9_cons_gwin32z2Failing and fix in PyPy not planned (see gh-18632))reasoncCs4ddi}dd}t|tjd|d}|jdksJdS) rFmaxtimercSstddS)Ng+=rm)timesleepr9rrrr2s z0TestShgoArguments.test_10_finite_time..frV)rrr/N)rrrnit)rrr2rrrrtest_10_finite_timesz%TestShgoArguments.test_10_finite_timecCsNddd}ttjtjdd|dd}tjd|jdtjd|jd dS) *Test to cover the case where f_lowest == 0rmT)rrrUNrrrr/)rrr2rr7r assert_equalr1rrrrtest_11_f_min_0sz!TestShgoArguments.test_11_f_min_0z no way of currently testing thiscCs6ddd}ttjtjdd|dd}tjd|jdS)rOrrm)rJrr/Nrr)rrr2rr7rrPr#rrrrtest_12_sobol_inf_conssz(TestShgoArguments.test_12_sobol_inf_conscCs0dd}dgd}t||dd}|ddd S) z-Test init of high-dimensional sobol sequencescSdSNrrr9rrrr2z/TestShgoArguments.test_13_high_sobol..fr?)rrr.rQN)rsampling_function)rr2rSHGOcrrrtest_13_high_sobols z$TestShgoArguments.test_13_high_sobolcCr)z6Test limited local iterations for a pseudo-global moderrXrGrNrrrrrtest_14_local_itersz$TestShgoArguments.test_14_local_itercCr)z9Test minimize every iter options and cover function cacherTr/r[rrNrrrrrtest_15_min_every_iterrz(TestShgoArguments.test_15_min_every_itercCs$ddi}ddi}ttd||ddS)z:Test disp=True with minimizers that do not support bounds rTr0z nelder-meadr)rrrNr)rcapsysrrrrrtest_16_disp_bounds_minimizers  z/TestShgoArguments.test_16_disp_bounds_minimizercCsdd}ttd|ddS)=Test the functionality to add custom sampling methods to shgocSstjj||fdS)N)size)r7randomuniform)rdrrrsamplerCz9TestShgoArguments.test_17_custom_sampling..samplerArNr)rrdrrrtest_17_custom_samplingsz)TestShgoArguments.test_17_custom_samplingc Csdd}gd}gd}tt||}t||}t||}t||}|j|jks*J|j|jks2J|j|jks:Jtgd}tj |j |tj |j |j dS)NcSst|Sr)r7squarer8r9rrrr2sz1TestShgoArguments.test_18_bounds_class..f)grng)rrrR)rrnrm) r&ziprrnfevmessagerr7r>rrr1) rr2lbub bounds_old bounds_newres_old_boundsres_new_boundsx_optrrrtest_18_bounds_classs   z&TestShgoArguments.test_18_bounds_classrUcCstd}ttd|jdWdn1swYttdtdtd}ttd|jdWdn1s:wYttdtddS)r_r.rA)rrN)rrrmapr)rprrrtest_19_parallelization s  z)TestShgoArguments.test_19_parallelizationcCszdd}dd}dgd}dd}d d }d d }d |ddd |dd|df}t||dd|dt||dd|dddS)z0Test that constraints can be passed to argumentsc Ssb|dd ttt|dd|dd|dttt|d|ddSrergr9rrr eggholderrjz=TestShgoArguments.test_20_constrained_args..eggholdercSs0d|dd|dd|dd|dS) Nǧ8@rg:@r/'r.g@D@rYrr9rrrr2!rNz5TestShgoArguments.test_20_constrained_args..frrnrXcSs<|d|d|d|dd|dd|dd S) Ngffffff@rgffffff@r/g333333&@r.g?rYrVr)r1rvrrr g1_modified&s .z?TestShgoArguments.test_20_constrained_args..g1_modifiedcSs~d|dd|dd|dd|dd d td |ddd |ddd |ddd|ddS)NrWrg'@r/gfffffD@r.g J@rYgRQ?gQ?gRQ?g4@gףp= ?)r7rhr9rrrrM*s 2>z6TestShgoArguments.test_20_constrained_args..g2cS$|d|d|d|ddSNrr/r.rYrr9rrrh12$z6TestShgoArguments.test_20_constrained_args..h1r )r)r"r#rsr!rrr/)rrrr)rrrrN)r)rrur2rrxrMr|r)rrrtest_20_constrained_argss    z*TestShgoArguments.test_20_constrained_argsc Csdd}t|ddgddggdddd d d d d d}ddgddgddgddgddgg}t||dd d d d}t|gd|d d}|jsFJt|j|jt|j|jdddS)z|Test that shgo can handle objective functions that return the gradient alongside the objective value. Fixes gh-13547cSstt|dd|fSNr.)r7r8powerr9rrrrBsz2TestShgoArguments.test_21_1_jac_true..funcr<r/r.rrVrr/T)r0r@)rrrrrcSst|dd|fSrr6r9rrrrNrrrY)rrr)r/r/r/r/r/)x0rr@rrN)rr rrr#r1)rrrrrrrrtest_21_1_jac_true>s& "  z$TestShgoArguments.test_21_1_jac_true derivativer@rBhesspc Csdd}dd}dd}dd}|||d }|||i}d d i}d g} t|| ||d } t|fdg| d||} | js>Jtj| j| jtj| j| jdS)zzshgo used to raise an error when passing `options` with 'jac' # see gh-12963. check that this is resolved cSs$d|d|dd|ddS)NrYrr.rVrr9rrr objectivear}zATestShgoArguments.test_21_2_derivative_options..objectivecSsd|ddS)Nr=rr.rr9rrrgradientdz@TestShgoArguments.test_21_2_derivative_options..gradientcSrSNr=rr9rrrrBgrUz.hesscSsd|Srr)r1rsrrrrjsz=TestShgoArguments.test_21_2_derivative_options..hessprr0 trust-constr)irrrr)rrN)rr rr7rrr#r1) rrrrrBrderivative_funcsrrrrrrrrtest_21_2_derivative_options[s"   z.TestShgoArguments.test_21_2_derivative_optionscCstgd}ttd}ddi}tt|||d}tttdfddi|}|js(Jt|j |j t|j |j ddd S) zsEnsure the Hessian gets passed correctly to the local minimizer routine. Previous report gh-14533. )rg?rrgffffff?rr)r@rBr0r;rrVrrN) r r rr r r7zerosrrr#r1)rrrrrrrrrtest_21_3_hess_options_rosen{s  z.TestShgoArguments.test_21_3_hess_options_rosencCspdd}dddd}dg}t||d|d d }t|td |d|d }|js(Jt|j|jt|j|jd S)zwshgo used to raise an error when passing `args` with Sobol sampling # see gh-12114. check that this is resolvedcSs |d|SrTr)r1rtrrrr#rz6TestShgoArguments.test_21_arg_tuple_sobol..funr cSs |ddSrrr9rrrrrz;TestShgoArguments.test_21_arg_tuple_sobol..r!rrU)r/r)rsrrr/)rrsrN)rr r7rrrr#r1)rr#rrrrrrrtest_21_arg_tuple_sobols  z)TestShgoArguments.test_21_arg_tuple_sobolN)/rrrrrrrr rrrrrrrrr!r#r)r,r4r7rCrErGxfailrsysplatformrNrQrrRrZr[r\r^rerqrrtr~r parametrizerrrrrrrrsV            &  rc@sleZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ e j jddZddZdS)TestShgoFailurescCsDddi}ttjtjdd|dd}tjd|jtjd|jdS)z'Test failure on insufficient iterationsmaxiterr.NrrFrX) rrr2rr7rrPrtnevrrrrtest_1_maxiters zTestShgoFailures.test_1_maxitercCsttttjtjdddS)z$Rejection of unknown sampling method not_SobolrWN) assert_raises ValueErrorrrr2rrrrrtest_2_samplings z TestShgoFailures.test_2_samplingcCsDddd}ttjtjd|dd}tjd|jtjd|jd S) zoCheck that the routine stops when no minimiser is found after maximum specified function evaluationsrUT)maxfevrrYrrrrFrWN) r test_tabler2rr7rrPrrhrrrrtest_3_1_no_min_pool_sobolsz+TestShgoFailures.test_3_1_no_min_pool_sobolcCs4ddd}ttjtjd|dd}tjd|jdS) zoCheck that the routine stops when no minimiser is found after maximum specified sampling evaluationsrUTmaxevrrYrrFN)rrr2rr7rrPrrrrrtest_3_2_no_min_pool_simplicialsz0TestShgoFailures.test_3_2_no_min_pool_simplicialcCddg}ttttj|dS)zSpecified bounds ub > lb)r=rYrYrVNrrrrr2rrrrrtest_4_1_bound_errz#TestShgoFailures.test_4_1_bound_errcCr)z)Specified bounds are of the form (lb, ub))rYrVrVrNrrrrrtest_4_2_bound_errrz#TestShgoFailures.test_4_2_bound_errcCs8ddd}ttjtjtjd|dd}tjd|jdS)zxEnsures the algorithm terminates on infeasible problems after maxev is exceeded. Use infty constraints optionrTrrrrrrFN rtest_infeasibler2rr)r7rrPrrrrrtest_5_1_1_infeasible_sobol z,TestShgoFailures.test_5_1_1_infeasible_sobolcCs:dddd}ttjtjtjd|dd}tjd|jdS)zEnsures the algorithm terminates on infeasible problems after maxev is exceeded. Do not use infty constraints optionrTF)rrrrrNrrrrrtest_5_1_2_infeasible_sobols z,TestShgoFailures.test_5_1_2_infeasible_sobolcCs8ddd}ttjtjtjd|dd}tjd|jdS)z[Ensures the algorithm terminates on infeasible problems after maxev is exceeded.iFrrrrNrrrrrtest_5_2_infeasible_simplicialrz/TestShgoFailures.test_5_2_infeasible_simplicialcCsPtjdddddd}tjtjf}tjdd|dd }tttg|Ri|dS) z.fun)scipy.optimizer rrhror1r#)rrrresultrrrrs  zTestShgoFailures.testN)rrrrrrrrrrrrrr thread_unsaferrrrrrrs       rc@seZdZddZddZdS)TestShgoReturnscs:gd}fddd_t|}tjj|jdS)Nrcjd7_t|Srrhr r9r#rrr#rz3TestShgoReturns.test_1_nfev_simplicial..funrrhrr7rrPrrrrrrtest_1_nfev_simplicials   z&TestShgoReturns.test_1_nfev_simplicialcs>gd}fddd_t|dd}tjj|jdS)Nrcrrrr9rrrr#$rz.TestShgoReturns.test_1_nfev_sobol..funrrrWrrrrrtest_1_nfev_sobol!s  z!TestShgoReturns.test_1_nfev_sobolN)rrrrrrrrrrs rcCsdd}t|dgdg}t|tddg}ttddg|dd}tt|jd dks/Jtt|jd d ks>J|j sCJdS) NcSt|}t|dgSrr7asarrayr8r9rrrquad0 z$test_vector_constraint..quadg@rYrnrrr"r.r) rrr7r>rr allr8r1r)rnlcoldcrrrrtest_vector_constraint.srzignore:delta_gradcCsdd}t|dgdg}ddi}ttddg|d|d }tt|jd dks*Jtt|jd d ks9J|js>JdS) NcSrrrr9rrrr?rztest_trust_constr..quadg@rYr0rrr)rrrr.r)rrr r7rr8r1r)rrrrrrrtest_trust_constr=srcCsdgd}dd}t|dd}tt||d}tt|jddd}d|d }tt||d}tt|jdd gd }d d}d|d }td d||dd}tt|jddS)N)g?rr.cSs|d|dSrrr9rrrfaultyXrz)test_equality_constraints..faulty333333@rcSs|d|ddS)Nrr/rrr9rrrr_r:rr!rwrXcSrzr{rr9rrrrhr}cSs t| Sr)r7prodr9rrrrnrz+test_equality_constraints..r)rrrrn)rrr rr7r8r1)rrrrrrrrtest_equality_constraintsTs&     rcCsjdd}|dd}ddddd d }ttd gd ||d }|jddks(J|jdddks3JdS)NcSst|ddSrAr6r9rrrr)vrCztest_gh16971..consr )r#r"r-rVrg?)rhobegtolcatol)r0rrr.)rrr0cobylarr)rr rlower)r)crrurrr test_gh16971us  r) rrrNNNNrr/)Errnumpyr7rKmultiprocessingr numpy.testingrrrrrrrrrr r r r r rscipy.optimize._constraintsrscipy.optimize._shgorrr*r+rrrr@r>rrrIrrTrrdrrlrpr&rgboundsLJrronesrrrrrrrrrrrrrrfilterwarningsrrrrrrrs (       % "      ":m v   !