
    gej                        d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZm	Z	 d dl
Z
d dl
mZmZ d dlmZmZmZmZmZmZmZmZ d dlmZ d dlmZ  G d d	          Zd
 Z G d de          Z eddgd d g          Z eddgd d g          Z eddgd d g          Z  G d de          Z! e!dgdgd ej"        dgdgdgdgdgdgdgdgdgdgg
           ej"        g d                     Z# e!d!gdgdg ej"        dgg           ej"        dg                     Z$ G d" d#e          Z% e%d$d%gd&d'gd()          Z& G d* d+e          Z' e'd,gd-z  g d.d/)          Z( G d0 d1e          Z) e)d2d2gd3gd4d5g6          Z* G d7 d8e          Z+d9Z, e- e.d:gd9z  d;gd9z                      Z/ e+e/d<gd6          Z0 G d= d>e          Z1 e1d?gd@z  dA ej2        d@          dBz
  6          Z3 G dC dDe          Z4 e4d,d,gdEgdFdFg6          Z5 G dG dHe          Z6 e6d$dIgdd6          Z7e
j8        9                    dJ          	 	 	 d`dP            Z: G dQ dR          Z; G dS dT          Z< G dU dV          Z= G dW dX          Z> G dY dZ          Z?d[ Z@e
j8        A                    d\          d]             ZBd^ ZCd_ ZDdS )a    N)Pool)assert_allcloseIS_PYPY)raiseswarns)shgoBoundsminimize_scalarminimizerosen	rosen_der
rosen_hessNonlinearConstraint)new_constraint_to_old)SHGOc                       e Zd Z	 	 ddZdS )StructTestFunctionNc                 L    || _         || _        || _        || _        || _        d S Nbounds
expected_xexpected_funexpected_xlexpected_funl)selfr   r   r   r   r   s         ?lib/python3.11/site-packages/scipy/optimize/tests/test__shgo.py__init__zStructTestFunction.__init__   s.    $(&*    )NNN)__name__
__module____qualname__r    r   r   r   r      s.        8<15+ + + + + +r   r   c                     g }| ^t          |           t          urt          |           t          ur| f} n	 | D ]} |                    d| d           t          |          }nd }|S )Nineqtypefun)r'   tuplelistappend)gconss     r   wrap_constraintsr.      s    D}GG5  tAwwd':':AA 	$ 	$AKK !# # $ $ $ $T{{Kr   c                   0    e Zd Zd Zd Z ee          ZdS )StructTest1c                 0    |d         dz  |d         dz  z   S Nr         r#   r   xs     r   fzStructTest1.f,       tqy1Q419$$r   c                 6    t          j        | d          dz
   S )Nr   axisg      @numpysumr6   s    r   r,   zStructTest1.g/   s    11%%%+,,r   Nr    r!   r"   r7   r,   r.   r-   r#   r   r   r0   r0   +   s>        % % %- - - ADDDr   r0   )   )r   r   r   r4   NNc                   4    e Zd ZdZd Zd Z ee          ZdS )StructTest2zN
    Scalar function with several minima to test all minimiser retrievals
    c                 6    |dz
  t          j        |          z  S )N   )r=   sinr5   s     r   r7   zStructTest2.fB   s    B%)A,,&&r   c                 4    dt          j        | d          z
  S )N:   r   r:   r<   r?   s    r   r,   zStructTest2.gE   s    EIaa(((((r   Nr    r!   r"   __doc__r7   r,   r.   r-   r#   r   r   rF   rF   =   sH         ' ' ') ) ) ADDDr   rF   )r   <   g⡽4$?0_r<g GK@g
]<@gI{_H@gi<&,@g>E@gqŤCQ4@g8$B@go.n:@g^5	>@)
rO   gs8g^V&+6g+?-2g+!/gN|1(gc=C#g^\gNg自ݿr   )r   g      @c                   T    e Zd ZdZd Zd Zd Zd Z eede	j
                  ZefZ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


    c                 6    d|d         dz  z  |d         dz  z   S )Ng{Gz?r   r3   r4   r#   r5   s     r   r7   zStructTest3.f|   s"    qtk!QqTaK//r   c                 *    | d         | d         z  dz
  S )Nr   r4         9@r#   r?   s    r   g1zStructTest3.g1   s    tad{T!!r   c                 6    | d         dz  | d         dz  z   dz
  S )Nr   r3   r4   rT   r#   r?   s    r   g2zStructTest3.g2   s"    tqy1Q419$t++r   c                 ^    | d         | d         z  dz
  | d         dz  | d         dz  z   dz
  fS )Nr   r4   rT   r3   r#   r?   s    r   r,   zStructTest3.g   s8    tad{T!1Q419qtqy#84#???r   r   N)r    r!   r"   rM   r7   rU   rW   r,   r   npinf_StructTest3__nlcr-   r#   r   r   rQ   rQ   j   sv         "0 0 0" " ", , ,@ @ @  1bf--E8DDDr   rQ   )r3   2   )r   r\   g'In/@gS[:XL?      @)r   r   r   c                   R    e Zd ZdZd Zd Zd Zd Zd ZeeeefZ	 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
    c                 ,   |d         dz
  dz  d|d         dz
  dz  z  z   |d         dz  z   d|d         d	z
  dz  z  z   d|d         d
z  z  z   d|d         dz  z  z   |d
         dz  z   d|d         z  |d
         z  z
  d|d         z  z
  d|d
         z  z
  S )Nr   
   r3      r4               rB         r#   r5   s     r   r7   zStructTest4.f   s    1q 1!r	a'7#77!A$!)CqtbyQ&&')+adai8:;adai-HJKKK ad(QqT/" %'1I. 12AaD9 	r   c                     d| d         dz  z  d| d         dz  z  z   | d         z   d| d         dz  z  z   d| d         z  z   dz
   S )Nr3   r   re   r4   rd   rb      r#   r?   s    r   rU   zStructTest4.g1   s^    QqTQYQqTQY.15AaDAIEqt8!" # 	#r   c                 z    d| d         z  d| d         z  z   d| d         dz  z  z   | d         z   | d         z
  dz
   S )	Nrg   r   re   r4   ra   r3   rd   g     q@r#   r?   s    r   rW   zStructTest4.g2   sH    QqTA!H$rAaDAI~5!<qtCeKLLr   c                 n    d| d         z  | d         dz  z   d| d         dz  z  z   d| d         z  z
  dz
   S )	N   r   r4   r3   rB   rb   rh      r#   r?   s    r   g3zStructTest4.g3   sC    adQqTQY&QqTQY6QqTACGHHr   c                     d| d         dz  z  | d         dz  z   d| d         z  | d         z  z
  d| d         dz  z  z   d| d         z  z   d| d         z  z
   S )	Nrd   r   r3   r4   re   rb   rf   rB   r#   r?   s    r   g4zStructTest4.g4   sn    QqTQY1*Q1X!_<q1Q419}Lqt8 1Q4i( ) 	)r   Nr    r!   r"   rM   r7   rU   rW   ro   rq   r,   r.   r-   r#   r   r   r_   r_      s           # # #M M MI I I) ) ) 
RRAADDDr   r_   )ira   rg   )gܤ@g$H8?g	޿g0ev@gn8)gk_@/?g0g+?gP{[
E@c                   .    e Zd Zd ZdZ ee          ZdS )StructTest5c                 J   |d         dz    t          j        t          j        t          |d         dz  |d         dz   z                                 z  |d         t          j        t          j        t          |d         |d         dz   z
                                z  z
  S Nr4   g     G@r          @r=   rI   sqrtabsr5   s     r   r7   zStructTest5.f   s    A$+)EJs1Q4#:1+E'F'FGGHHIA$5:c!A$!A$+2F.G.G#H#HIIIJ 	r   Nr@   r#   r   r   rt   rt      s4           	AADDDr   rt   )i i   g g      @gyCy@)r   r   r   c                   2    e Zd ZdZd ZdZ ee          ZdS )StructTestLJzZ
    LennardJones objective function. Used to test symmetry constraints
    settings.
    c                    t          d|            |d         | _        t          | j        dz            }d}t          |dz
            D ]}t          |dz   |          D ]x}d|z  }d|z  }||         ||         z
  }	||dz            ||dz            z
  }
||dz            ||dz            z
  }|	|	z  |
|
z  z   ||z  z   }||z  |z  }|dk    r|d|z  dz
  |z  z  }y|S )	Nzx = r   re           r4   r3         ?rw   )printNintrange)r   r6   argsksijabxdydzdeduds                 r   r7   zStructTestLJ.f   s   jQjja
OOq1u 
	/ 
	/A1q5!__ 	/ 	/EEqTAaD[q1uX!a%(q1uX!a%("WrBw&b0"Wr\88#(S.B..A	/ r   NrL   r#   r   r   r|   r|      s>         
  ( 	AADDDr   r|   rB   g            @      c                   .    e Zd Zd ZdZ ee          ZdS )StructTestSc                 x    |d         dz
  dz  |d         dz
  dz  z   |d         dz
  dz  z   |d         dz
  dz  z   S )Nr         ?r3   r4   re   r#   r5   s     r   r7   zStructTestS.f   sS    1!QqTCZA$55Q4#:!#$'(tcza&78 	9r   Nr@   r#   r   r   r   r      s4        9 9 9 	AADDDr   r   )r   rw   rd   r~   r   c                   .    e Zd Zd ZdZ ee          ZdS )StructTestTablec                 :    |d         dk    r|d         dk    rdS dS )Nr         @r4   r\   d   r#   r5   s     r   r7   zStructTestTable.f  s%    Q43;;1Q43;;23r   Nr@   r#   r   r   r   r   
  s4           	AADDDr   r   r\   r   c                   R    e Zd ZdZd Zd Zd Zd Zd ZeeeefZ	 e
e	          ZdS )StructTestInfeasiblez0
    Test function with no feasible domain.
    c                 0    |d         dz  |d         dz  z   S r2   r#   )r   r6   r   s      r   r7   zStructTestInfeasible.f  r8   r   c                 *    | d         | d         z   dz
  S Nr   r4   r#   r?   s    r   rU   zStructTestInfeasible.g1"  s    tad{Qr   c                 ,    | d         | d         z   dz
   S r   r#   r?   s    r   rW   zStructTestInfeasible.g2%  s    1!q!!r   c                 ,    | d          | d         z   dz
  S r   r#   r?   s    r   ro   zStructTestInfeasible.g3(  s    !uqt|ar   c                 .    | d          | d         z   dz
   S r   r#   r?   s    r   rq   zStructTestInfeasible.g4+  s    A$1!""r   Nrr   r#   r   r   r   r     s         % % %  " " "     # # # 
RRAADDDr   r   )rA   r4   
Not a testr#   h㈵>r   sobolr4   c
                 D   t          | j        | j        || j        |||||||	          }
t	          d|
            t          j        d|
            | j        -t          j	        
                    |
j        | j        ||           | j        ,t          j	        
                    |
j        | j        |           | j        ,t          j	        
                    |
j        | j        |           | j        ,t          j	        
                    |
j        | j        |           d S )N)	r   constraintsniterscallbackminimizer_kwargsoptionssampling_methodworkerszres = rtolatolr   )r   r7   r   r-   r   logginginfor   r=   testingr   r6   r   r(   r   xlr   funl)testr   	test_atolr   r   r   r   r   r   r   ress              r   run_testr   8  sG    tvt{49%( 0'.A A AC
 
.3..L#   "%%ceT_+4+4 	& 	6 	6 	6
 $%%cg&*&7+4 	& 	6 	6 	6 #%%cf&*&6+4 	& 	6 	6 	6 %%%ch&*&8+4 	& 	6 	6 	6 Fr   c                   l    e Zd ZdZd Zd Zd Zd Zd Zd Z	e
j        j        d             Zd	 Zd
 ZdS )TestShgoSobolTestFunctionsz8
    Global optimisation tests with Sobol sampling:
    c                 .    t          t                     dS )VMultivariate test function 1:
        x[0]**2 + x[1]**2 with bounds=[(-1, 6), (-1, 6)]Nr   test1_1r   s    r   test_f1_1_sobolz*TestShgoSobolTestFunctions.test_f1_1_sobola       	r   c                 .    t          t                     dS )zUMultivariate test function 1:
         x[0]**2 + x[1]**2 with bounds=[(0, 1), (0, 1)]Nr   test1_2r   s    r   test_f1_2_sobolz*TestShgoSobolTestFunctions.test_f1_2_sobolf  r   r   c                 :    ddi}t          t          |           dS )z_Multivariate test function 1:
        x[0]**2 + x[1]**2 with bounds=[(None, None),(None, None)]dispT)r   Nr   test1_3r   r   s     r   test_f1_3_sobolz*TestShgoSobolTestFunctions.test_f1_3_sobolk  s&     4.'******r   c                 .    t          t                     dS )RUnivariate test function on
        f(x) = (x - 30) * sin(x) with bounds=[(0, 60)]Nr   test2_1r   s    r   test_f2_1_sobolz*TestShgoSobolTestFunctions.test_f2_1_sobolq  r   r   c                 .    t          t                     dS )NUnivariate test function on
        f(x) = (x - 30) * sin(x) bounds=[(0, 4.5)]Nr   test2_2r   s    r   test_f2_2_sobolz*TestShgoSobolTestFunctions.test_f2_2_sobolv  r   r   c                 .    t          t                     dS )%NLP: Hock and Schittkowski problem 18Nr   test3_1r   s    r   test_f3_sobolz(TestShgoSobolTestFunctions.test_f3_sobol{  s    r   c                 <    ddi}t          t          d|           dS )?NLP: (High dimensional) Hock and Schittkowski 11 problem (HS11)infty_constraintsFi  r   r   Nr   test4_1r   s     r   test_f4_sobolz(TestShgoSobolTestFunctions.test_f4_sobol  s)     '.GW555555r   c                 2    t          t          d           dS )NLP: Eggholder, multimodalrN   )r   Nr   test5_1r   s    r   test_f5_1_sobolz*TestShgoSobolTestFunctions.test_f5_1_sobol  s     	Br   c                 4    t          t          dd           dS )r   rN   rb   )r   r   Nr   r   s    r   test_f5_2_sobolz*TestShgoSobolTestFunctions.test_f5_2_sobol  s     	Ba((((((r   N)r    r!   r"   rM   r   r   r   r   r   r   pytestmarkslowr   r   r   r#   r   r   r   r   [  s         
  
  
+ + +  
  
   [6 6 6     
) ) ) ) )r   r   c                      e Zd ZdZd Zd Zd Zd Zd Zd Z	e
j        j        d             Zd	 Zd
 Zd Zd Ze
j                            d          d             Ze
j                            d          d             ZdS )TestShgoSimplicialTestFunctionsz=
    Global optimisation tests with Simplicial sampling:
    c                 4    t          t          dd           dS )r   r4   
simplicialr   r   Nr   r   s    r   test_f1_1_simplicialz4TestShgoSimplicialTestFunctions.test_f1_1_simplicial       	A|<<<<<<r   c                 4    t          t          dd           dS )zTMultivariate test function 1:
        x[0]**2 + x[1]**2 with bounds=[(0, 1), (0, 1)]r4   r   r   Nr   r   s    r   test_f1_2_simplicialz4TestShgoSimplicialTestFunctions.test_f1_2_simplicial  r   r   c                 4    t          t          dd           dS )z_Multivariate test function 1: x[0]**2 + x[1]**2
        with bounds=[(None, None),(None, None)]rb   r   r   Nr   r   s    r   test_f1_3_simplicialz4TestShgoSimplicialTestFunctions.test_f1_3_simplicial  r   r   c                 @    ddi}t          t          dd|d           dS )r   minimize_every_iterF   rg   r   r   r   r   r   Nr   r   s     r   test_f2_1_simplicialz4TestShgoSimplicialTestFunctions.test_f2_1_simplicial  s:     )%0Cq'!-	/ 	/ 	/ 	/ 	/ 	/r   c                 4    t          t          dd           dS )r   r4   r   r   Nr   r   s    r   test_f2_2_simplicialz4TestShgoSimplicialTestFunctions.test_f2_2_simplicial  r   r   c                 4    t          t          dd           dS )r   r4   r   r   Nr   r   s    r   test_f3_simplicialz2TestShgoSimplicialTestFunctions.test_f3_simplicial  s    A|<<<<<<r   c                 4    t          t          dd           dS )r   r4   r   r   Nr   r   s    r   test_f4_simplicialz2TestShgoSimplicialTestFunctions.test_f4_simplicial  r   r   c                 H    ddd}d}t          t          |d|dd           dS )	z&LJ: Symmetry-constrained test functionTsymmetryr   rB   ,  r4   r   r   r   r   r   r   Nr   testLJr   r   r   s      r   test_lj_symmetry_oldz4TestShgoSimplicialTestFunctions.test_lj_symmetry_old  sH    #! !dc !-	/ 	/ 	/ 	/ 	/ 	/r   c                 P    dgdz  dd}d}t          t          |d|dd	           d
S )z&LJ: Symmetry constrained test functionr   rB   Tr  r  r	  r4   r   r
  Nr  r  s      r   test_f5_1_lj_symmetryz5TestShgoSimplicialTestFunctions.test_f5_1_lj_symmetry  sN     !uqy! !dc !-	/ 	/ 	/ 	/ 	/ 	/r   c                 F    ddgdd}t          t          d|dd           dS )	z"Symmetry constrained test functionr   Tr  r   r4   r   r   r   r   r   Nr   r   s     r   test_f5_2_cons_symmetryz7TestShgoSimplicialTestFunctions.test_f5_2_cons_symmetry  sG     !1v! ! 	C !-	/ 	/ 	/ 	/ 	/ 	/r   c                 F    g ddd}t          t          d|dd           dS )	z)Assymmetrically constrained test function)r   r   r   re   Tr  i'  r4   r   r  N)r   test_sr   s     r   test_f5_3_cons_symmetryz7TestShgoSimplicialTestFunctions.test_f5_3_cons_symmetry  sH    +||! ! 	5 !-	/ 	/ 	/ 	/ 	/ 	/r   r   c                     ddfdd}t          t          j        ddgz  |          }|j        sJ t	          |j        dd	
           t	          |j        d           dS )zOReturn a minimum on a perfectly symmetric problem, based on
            gh10429r   eqc                 2    t          j        |           z
  S r   )r=   mean)r6   avgs    r   <lambda>zFTestShgoSimplicialTestFunctions.test_f0_min_variance.<locals>.<lambda>  s    uz!}}s/B r   r&   rB   rC   r   r   r   V瞯<r   N)r   r=   varsuccessr   r(   r6   )r   r-   r   r  s      @r   test_f0_min_variancez4TestShgoSimplicialTestFunctions.test_f0_min_variance  s|     %B%B%B%BCC 59Q&\tDDD{////s#####r   c                     d }dg}t          ||          }t          ||d                   }|j        sJ t          |j        |j                   t          |j        |j        d           dS )zRReturn a minimum on a perfectly symmetric 1D problem, based on
            gh10538c                     | | dz
  z  | dz
  z  S )Nr   r   r#   r?   s    r   r(   zDTestShgoSimplicialTestFunctions.test_f0_min_variance_1D.<locals>.fun  s    C=AG,,r   rC   )r   r   ư>)r   N)r   r
   r   r   r(   r6   )r   r(   r   r   refs        r   test_f0_min_variance_1Dz7TestShgoSimplicialTestFunctions.test_f0_min_variance_1D  s    
	- 	- 	- 3v&&&c&)444{)))su4000000r   N)r    r!   r"   rM   r   r   r   r   r   r  r   r   r   r  r  r  r  r  skipr!  r&  r#   r   r   r   r     s-        = = =
= = =
= = =
/ / /= = =
= = = [= = =/ / // / // / // / / [l##
$ 
$ $#
$ [l##1 1 $#1 1 1r   r   c                      e Zd Zd Zd Zd Zd Zd Zd Zd Z	e
j        j        d             Ze
j        j        d	             Zd
 Zd Zd Zd Zd Zd Zd Zd Zd Zd Ze
j                            eo
ej        dk    d          d             Zd Ze
j                            d          d             Z d Z!d Z"d Z#d Z$d Z%d  Z&d! Z'd" Z(d# Z)e
j        *                    d$g d%          d&             Z+d' Z,d( Z-d)S )*TestShgoArgumentsc                 6    t          t          ddd           dS )z>Iterative simplicial sampling on TestFunction 1 (multivariate)Nr3   r   r   r   r   r   r   s    r   test_1_1_simpl_iterz%TestShgoArguments.test_1_1_simpl_iter  s    D<HHHHHHr   c                 @    ddi}t          t          dd|d           dS )z3Iterative simplicial on TestFunction 2 (univariate)r   FN	   r   r   r   r   s     r   test_1_2_simpl_iterz%TestShgoArguments.test_1_2_simpl_iter  s8    (%0D7!-	/ 	/ 	/ 	/ 	/ 	/r   c                 6    t          t          ddd           dS )z9Iterative Sobol sampling on TestFunction 1 (multivariate)Nr4   r   r+  r   r   s    r   test_2_1_sobol_iterz%TestShgoArguments.test_2_1_sobol_iter  s    D7CCCCCCr   c                 2   t          t          j        t          j        t          j        ddd          }t
          j                            |j        t          j	        dd           t
          j                            |j
        t          j        d           dS )z7Iterative Sobol sampling on TestFunction 2 (univariate)Nr4   r   )r   r   r   r   r   r   r   )r   r   r7   r   r-   r=   r   r   r6   r   r(   r   r   r   s     r   test_2_2_sobol_iterz%TestShgoArguments.test_2_2_sobol_iter  s    79gn',G= = = 	%%ceW-?d+/ 	& 	1 	1 	1%%cgw/C$%OOOOOr   c           
          d }t           t          fD ]D}t          |j        |j        dd|ddi           t          |j        |j        dd|ddi           EdS )	zKIterative sampling on TestFunction 1 and 2  (multi and univariate)
        c                 $    t          d           d S Nz Local minimization callback testr   r?   s    r   callback_funczATestShgoArguments.test_3_1_disp_simplicial.<locals>.callback_func      455555r   r4   r   r   Tr   r   r   r   r   r   r   r   Nr   r   r   r7   r   r   r9  r   s      r   test_3_1_disp_simplicialz*TestShgoArguments.test_3_1_disp_simplicial  s    	6 	6 	6 g& 	A 	ADA!-'&$A A A A <'&$A A A A A		A 	Ar   c           
          d }t           t          fD ]D}t          |j        |j        dd|ddi           t          |j        |j        dd|ddi           Ed	S )
zAIterative sampling on TestFunction 1 and 2 (multi and univariate)c                 $    t          d           d S r7  r8  r?   s    r   r9  z<TestShgoArguments.test_3_2_disp_sobol.<locals>.callback_func*  r:  r   r4   r   r   Tr;  r   r<  Nr=  r>  s      r   test_3_2_disp_sobolz%TestShgoArguments.test_3_2_disp_sobol'  s    	6 	6 	6 g& 	A 	ADAw'&$A A A A <'&$A A A A A		A 	Ar   c                     t          d dgd          }t          d dg          }t          |j        |j                   t          |j        |j                   dS )zMUsing `args` used to cause `shgo` to fail; see #14589, #15986,
        #16506c                     | |z  |z   S r   r#   )r6   yzs      r   r  z5TestShgoArguments.test_args_gh14589.<locals>.<lambda>7  s    A	 r   )r   re   )r4   r3   )funcr   r   c                     d| z  dz   S )Nr3   r4   r#   r?   s    r   r  z5TestShgoArguments.test_args_gh14589.<locals>.<lambda>9  s    !a%!) r   )rG  r   N)r   r   r(   r6   )r   r   r%  s      r   test_args_gh14589z#TestShgoArguments.test_args_gh145894  sj     116(   ++VH===)))su%%%%%r   c                 X    t           j        ddd}t          t           dd|d           dS )z,Test known function minima stopping criteriar$  T)f_minf_tolr   Nr   r   r   r   r   r   r   r   r   r   s     r   test_4_1_known_f_minz&TestShgoArguments.test_4_1_known_f_min=  sJ     $0 *.0 0 	DD'!-	/ 	/ 	/ 	/ 	/ 	/r   c                 Z    t           j        dddd}t          t           dd|d           dS )	z*Test Global mode limiting local evalutionsr$  Tr4   )rK  rL  r   
local_iterNr   r   rM  rN  r   s     r   test_4_2_known_f_minz&TestShgoArguments.test_4_2_known_f_minH  sM     )#'  	DD'!-	/ 	/ 	/ 	/ 	/ 	/r   c           	          t           j        ddddd}t          t           j        t           j        t           j        dd|d          }t          j                            |j	        t           j
        d	d	
           dS )z8Test Global mode limiting local evaluations for 1D funcsr$  Tr4   FrK  rL  r   rQ  r   Nr   r   r   r   r   r   r   r   )r   r   r   r7   r   r-   r=   r   r   r6   r   r   r   r   s      r   test_4_4_known_f_minz&TestShgoArguments.test_4_4_known_f_minU  s     )#'!&( ( 79gn',w#*, , , 	%%ceW-?d+/ 	& 	1 	1 	1 	1 	1r   c                     t          t          j        t          j        t          j                  }t
          j                            |j        t          j	        dd           dS )z;Test Default simplicial sampling settings on TestFunction 1r   r   r   N
r   r   r7   r   r-   r=   r   r   r6   r   r3  s     r   test_5_1_simplicial_arglessz-TestShgoArguments.test_5_1_simplicial_arglesse  sS    79gn',GGG%%ceW-?d+/ 	& 	1 	1 	1 	1 	1r   c                     t          t          j        t          j        t          j        d          }t
          j                            |j        t          j	        dd           dS )z6Test Default sobol sampling settings on TestFunction 1r   r   r   r   r   NrZ  r3  s     r   test_5_2_sobol_arglessz(TestShgoArguments.test_5_2_sobol_arglessk  s\    79gn',#*, , ,%%ceW-?d+/ 	& 	1 	1 	1 	1 	1r   c                 8   ddi}t          t          j        t          j        t          j        |d          }t
          j                            |j        t          j	        dd           t
          j                            |j
        t          j        d           dS )	:Test that maximum iteration option works on TestFunction 3max_iterr3   r   r   r   r   r   r   r   Nr   r   r7   r   r-   r=   r   r   r6   r   r(   r   rV  s      r   test_6_1_simplicial_max_iterz.TestShgoArguments.test_6_1_simplicial_max_iterr      q/79gn',"LB B B%%ceW-?d+/ 	& 	1 	1 	1%%cgw/C$%OOOOOr   c                 8   ddi}t          t          j        t          j        t          j        |d          }t
          j                            |j        t          j	        dd           t
          j                            |j
        t          j        d           dS )	r`  min_iterr3   r   rb  r   r   r   Nrc  rV  s      r   test_6_2_simplicial_min_iterz.TestShgoArguments.test_6_2_simplicial_min_iter{  re  r   c                 `    dD ]*}|t           j        d}t          t           dd|d           +dS )z@Test the minimizer_kwargs arguments for solvers with constraints)COBYLASLSQP)methodr   r   MbP?r   r   r   r   r   N)r   r-   r   )r   solverr   s      r   test_7_1_minkwargsz$TestShgoArguments.test_7_1_minkwargs  s\     * 	. 	.F +1/6| =  =Wt&6%,. . . . .	. 	.r   c                     ddi}ddi}t          t          j        t          j        t          j        d         ||           dS )z'Test the minimizer_kwargs default initsftolr   r   Tr   )r   r   r   Nr   r   r7   r   r-   )r   r   r   s      r   test_7_2_minkwargsz$TestShgoArguments.test_7_2_minkwargs  sQ    "D>4.WYGLO.	A 	A 	A 	A 	A 	Ar   c                     dD ]R}d }d }|||d}t          j        d|            t          j        d           t          t          dd|d	
           SdS )z?Test minimizer_kwargs arguments for solvers without constraints)zNelder-MeadPowellCGBFGS	Newton-CGzL-BFGS-BTNCdoglegz	trust-ncgztrust-exactztrust-krylovc                 \    t          j        d| d         z  d| d         z  g          j        S )Nr3   r   r4   )r=   arrayTr?   s    r   jacz1TestShgoArguments.test_7_3_minkwargs.<locals>.jac  s*    {A!Ha!A$h#788::r   c                 6    t          j        ddgddgg          S Nr3   r   )r=   r}  r?   s    r   hessz2TestShgoArguments.test_7_3_minkwargs.<locals>.hess  s    {QFQF#3444r   )rl  r  r  z	Solver = zd====================================================================================================r   rm  r   rn  N)r   r   r   r   )r   ro  r  r  r   s        r   test_7_3_minkwargsz$TestShgoArguments.test_7_3_minkwargs  s    ' 	. 	.F; ; ;5 5 5 +1'*(, .  . L-V--...L###Wt&6%,. . . . .	. 	.r   c                 B    ddd}t          t          d d |d           d S )Nr4   T)minhgrdr   r   r   r   r   s     r   test_8_homology_group_diffz,TestShgoArguments.test_8_homology_group_diff  sB    *.0 0 	Dg!-	/ 	/ 	/ 	/ 	/ 	/r   c                 r    t          t          j        t          j        t          j        d                    dS )'Test single function constraint passingr   rY  Nrs  r   s    r   test_9_cons_gzTestShgoArguments.test_9_cons_g  s'    WYGLODDDDDDr   win32z2Failing and fix in PyPy not planned (see gh-18632))reasonc                 h    ddi}d }t          |t          j        d|          }|j        dk    sJ dS )r  maxtimer  c                 .    t          j        d           dS )Ng+=r~   )timesleepr?   s    r   r7   z0TestShgoArguments.test_10_finite_time.<locals>.f  s    Ju3r   rb   )r   r   r4   N)r   r   r   nit)r   r   r7   r   s       r   test_10_finite_timez%TestShgoArguments.test_10_finite_time  sJ     e$	 	 	 1gnAw???w!||||||r   c                    ddd}t          t          j        t          j        dd|d          }t          j                            d|j        d                    t          j                            d|j        d	                    dS )
*Test to cover the case where f_lowest == 0r~   T)rK  r   ra   Nr   r   r   r4   )r   r   r7   r   r=   r   assert_equalr6   rV  s      r   test_11_f_min_0z!TestShgoArguments.test_11_f_min_0  s{    ! !79gn$"G= = =""1ceAh///""1ceAh/////r   z no way of currently testing thisc                     ddd}t          t          j        t          j        dd|d          }t          j                            d|j                   dS )r  r  r~   )r  rK  r4   Nr   r   )r   r   r7   r   r=   r   r  r(   rV  s      r   test_12_sobol_inf_consz(TestShgoArguments.test_12_sobol_inf_cons  s[     $! !79gn"G= = =""300000r   c                 h    d }dgdz  }t          ||d          }|                    dd           dS )	z-Test init of high-dimensional sobol sequencesc                     dS Nr   r#   r?   s    r   r7   z/TestShgoArguments.test_13_high_sobol.<locals>.f      1r   rD   )   r   r   r3   r\   N)r   sampling_function)r   r7   r   SHGOcs       r   test_13_high_sobolz$TestShgoArguments.test_13_high_sobol  sP    	 	 	 !B&Q8882&&&&&r   c                 <    ddi}t          t          d|           dS )z6Test limited local iterations for a pseudo-global moderQ  rd   rN   r   Nr   r   s     r   test_14_local_iterz$TestShgoArguments.test_14_local_iter  s'    #B000000r   c                 @    ddi}t          t          dd|d           dS )z9Test minimize every iter options and cover function cacher   Tr4   rg   r   r   Nr   r   s     r   test_15_min_every_iterz(TestShgoArguments.test_15_min_every_iter  s8    ($/AQ!(	* 	* 	* 	* 	* 	*r   c                 F    ddi}ddi}t          t          d||           dS )z:Test disp=True with minimizers that do not support bounds r   Trl  znelder-meadr   )r   r   r   Nr   )r   r   r   s      r   test_16_disp_bounds_minimizerz/TestShgoArguments.test_16_disp_bounds_minimizer  sG    4.$m4, 3C	E 	E 	E 	E 	E 	Er   c                 :    d }t          t          d|           dS )=Test the functionality to add custom sampling methods to shgoc                 F    t           j                            | |f          S )N)size)r=   randomuniform)r   ds     r   samplez9TestShgoArguments.test_17_custom_sampling.<locals>.sample  s    <''aV'444r   rH   r   Nr   )r   r  s     r   test_17_custom_samplingz)TestShgoArguments.test_17_custom_sampling  s.    	5 	5 	5 	B777777r   c                    d }g d}g d}t          t          ||                    }t          ||          }t          ||          }t          ||          }|j        |j        k    sJ |j        |j        k    sJ |j        |j        k    sJ t          j        g d          }t          j	        
                    |j        |           t          j	        
                    |j        |j                   d S )Nc                 N    t          j        |                                           S r   )r=   squarer>   r?   s    r   r7   z1TestShgoArguments.test_18_bounds_class.<locals>.f  s    <??&&(((r   )g      r   g      )r   r   r]   )r   r   r~   )r*   zipr	   r   nfevmessager   r=   r}  r   r   r6   )	r   r7   lbub
bounds_old
bounds_newres_old_boundsres_new_boundsx_opts	            r   test_18_bounds_classz&TestShgoArguments.test_18_bounds_class  s   	) 	) 	) ^^]]#b"++&&
B^^
a,,a,,"n&99999%)?????%)?????MMM**%%n&6>>>%%n&6&4&6	8 	8 	8 	8 	8r   c                    t          d          5 }t          t          d|j                   ddd           n# 1 swxY w Y   t          t          dt                     t          d          5 }t          t          d|j                   ddd           n# 1 swxY w Y   t          t          dt                     dS )r  r3   rH   )r   r   N)r   r   r   mapr  )r   ps     r   test_19_parallelizationz)TestShgoArguments.test_19_parallelization  s    !WW 	3WAE2222	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3B,,,,!WW 	2Vr151111	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	2 	22s++++++s   9= =/BBBc                     d }d }dgdz  }d }d }d }d|d	d
d|dd|df}t          ||dd|           t          ||dd|d           dS )z0Test that constraints can be passed to argumentsc                 J   | d         dz    t          j        t          j        t          | d         dz  | d         dz   z                                 z  | d         t          j        t          j        t          | d         | d         dz   z
                                z  z
  S rv   rx   r?   s    r   	eggholderz=TestShgoArguments.test_20_constrained_args.<locals>.eggholder!  s    dTkNi
3qtczQqTD[/I+J+J K KLLMdUYuz#adadTk6J2K2K'L'LMMMN r   c                 `    d| d         z  d| d         z  z   d| d         z  z   d| d         z  z   S )	Nǧ8@r   g     :@r4   '   r3   g     @D@re   r#   r?   s    r   r7   z5TestShgoArguments.test_20_constrained_args.<locals>.f'  s7    1Q4<%!A$,.ad:UQqT\IIr   r   r   rd   c                 r    |dz  | d         z  |dz  | d         z  z   d| d         z  z   d| d         z  z   d	z
  S )
Ngffffff@r   gffffff@r4   g333333&@r3   g?re   rb   r#   )r6   r   s     r   g1_modifiedz?TestShgoArguments.test_20_constrained_args.<locals>.g1_modified,  sT    s7QqT>AGadN2TAaD[@3J D  r   c                    d| d         z  d| d         z  z   d| d         z  z   d| d         z  z   d	z
  d
t          j        d| d         dz  z  d| d         dz  z  z   d| d         dz  z  z   d| d         dz  z  z             z  z
  S )Nrc   r   g'@r4   gfffffD@r3   gJ@re      gRQ?gQ?gRQ?g     4@gףp=
?)r=   ry   r?   s    r   rW   z6TestShgoArguments.test_20_constrained_args.<locals>.g20  s    1Iqt+dQqTk9D1Q4KG"Lej!	)9D1Q419<L)L+/!A$!)+;*<>BQqTQY>N*O P P PP r   c                 N    | d         | d         z   | d         z   | d         z   dz
  S Nr   r4   r3   re   r#   r?   s    r   h1z6TestShgoArguments.test_20_constrained_args.<locals>.h16  s+    Q4!A$;1%!,q00r   r%   )r   )r'   r(   r   r&   r  r	  r4   )r   r   r   r   )r   r   r   r   N)r   )r   r  r7   r   r  rW   r  r-   s           r   test_20_constrained_argsz*TestShgoArguments.test_20_constrained_args  s    	 	 		J 	J 	J !	 	 		 	 		1 	1 	1  TBB++b))+ 	Q#QD9999Q#QD$	& 	& 	& 	& 	& 	&r   c           
      D   d }t          |ddgddggddddd	d
           d }ddgddgddgddgddgg}t          ||ddd	d
          }t          |g d|d	          }|j        sJ t          |j        |j                   t          |j        |j        d           dS )z|Test that shgo can handle objective functions that return the
        gradient alongside the objective value. Fixes gh-13547c                 Z    t          j        t          j        | d                    d| z  fS Nr3   )r=   r>   powerr?   s    r   rG  z2TestShgoArguments.test_21_1_jac_true.<locals>.funcF  s'    9U[A..//Q66r   rA   r4   r3   r   rb   r   rk  T)rl  r  )r   r   r   r   r   c                 :    t          j        | dz            d| z  fS r  r<   r?   s    r   rG  z2TestShgoArguments.test_21_1_jac_true.<locals>.funcR  s    9Q!V$$a!e++r   r   re   )r   r   r   )r4   r4   r4   r4   r4   )x0r   r  r  r   N)r   r   r   r   r(   r6   )r   rG  r   r   r%  s        r   test_21_1_jac_truez$TestShgoArguments.test_21_1_jac_trueB  s   	7 	7 	7 	GaV$#(/==	
 	
 	
 	
	, 	, 	, q'Aq6B7QFQF;4/6t$D$DF F Ft! ! !{)))su5111111r   
derivativer  r  hesspc                 R   d }d }d }d }|||d}|||         i}ddi}dg}	t          ||	||	          }
t          |fd
g|	d||}|
j        sJ t          j                            |
j        |j                   t          j                            |
j        |j                   dS )zzshgo used to raise an error when passing `options` with 'jac'
        # see gh-12963. check that this is resolved
        c                 H    d| d         z  | d         z  d| d         z  z   dz   S )Nre   r   r3   rb   r#   r?   s    r   	objectivezATestShgoArguments.test_21_2_derivative_options.<locals>.objectivee  s*    qt8ad?Q1X-11r   c                     d| d         z  dz   S )NrB   r   r3   r#   r?   s    r   gradientz@TestShgoArguments.test_21_2_derivative_options.<locals>.gradienth  s    qt8a<r   c                     dS NrB   r#   r?   s    r   r  z<TestShgoArguments.test_21_2_derivative_options.<locals>.hessk  r  r   c                     d|z  S r  r#   )r6   r  s     r   r  z=TestShgoArguments.test_21_2_derivative_options.<locals>.hesspn  s    q5Lr   r  rl  trust-constr)ir   r   r   r   )r  r   N)r   r   r   r=   r   r   r(   r6   )r   r  r  r  r  r  derivative_funcsr   r   r   r   r%  s               r   test_21_2_derivative_optionsz.TestShgoArguments.test_21_2_derivative_options_  s   	2 	2 	2	  	  	 	 	 		 	 	 $,TEJJ/
;<$n59f7G"$ $ $y "aS " ";K " " " {%%cgsw777%%ceSU33333r   c                 0   g d}t           t          d}ddi}t          t          |||          }t	          t          t          j        d          fddi|}|j        sJ t          |j	        |j	                   t          |j
        |j
        d           d	S )
zsEnsure the Hessian gets passed correctly to the local minimizer
        routine. Previous report gh-14533.
        )r   g?r  r   gffffff?r  r  )r  r  rl  ry  r  rb   r  r   N)r   r   r   r   r   r=   zerosr   r   r(   r6   )r   r   r   r   r   r%  s         r   test_21_3_hess_options_rosenz.TestShgoArguments.test_21_3_hess_options_rosen  s     DCC#Z88$k25&3C"$ $ $uek!nn " "[ " " "{)))su5111111r   c                    d }dd d}dg}t          ||d|d          }t          |t          j        d	          |d|
          }|j        sJ t          |j        |j                   t          |j        |j                   dS )zwshgo used to raise an error when passing `args` with Sobol sampling
        # see gh-12114. check that this is resolvedc                     | d         |z  S r  r#   )r6   r   s     r   r(   z6TestShgoArguments.test_21_arg_tuple_sobol.<locals>.fun  s    Q419r   r%   c                     | d         dz
  S r   r#   r?   s    r   r  z;TestShgoArguments.test_21_arg_tuple_sobol.<locals>.<lambda>  s    1 r   r&   r   ra   )r4   r   )r   r   r   r4   )r   r   r   N)r   r   r=   r  r   r   r(   r6   )r   r(   r   r   r   r%  s         r   test_21_arg_tuple_sobolz)TestShgoArguments.test_21_arg_tuple_sobol  s    	 	 	 !'/A/ABB3T{#*, , ,sEKNN6#.0 0 0{)))su%%%%%r   N).r    r!   r"   r,  r/  r1  r4  r?  rB  rI  r   r   r   rO  rR  rW  r[  r^  rd  rh  rp  rt  r  r  r  xfailr   sysplatformr  r  r'  r  r  r  r  r  r  r  r  r  r  parametrizer  r  r  r#   r   r   r)  r)    s       I I I/ / /D D DP P PA A AA A A& & & [/ / / [
/ 
/ 
/1 1 1 1 1 11 1 1P P PP P P
. 
. 
.A A A. . .(/ / /E E E [w:3<7#:G  I I
 
I I
0 0 0 [?@@	1 	1 A@	1	' 	' 	'1 1 1
* * *E E E8 8 88 8 8*, , ,"& "& "&H2 2 2: [\+C+C+CDD4 4 ED4>2 2 2& & & & &r   r)  c                   P    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zd ZdS )TestShgoFailuresc                     ddi}t          t          j        t          j        dd|d          }t          j                            d|j                   t          j                            d|j                   dS )z'Test failure on insufficient iterationsmaxiterr3   Nr   r   Frd   )	r   r   r7   r   r=   r   r  r   tnevrV  s      r   test_1_maxiterzTestShgoFailures.test_1_maxiter  sl    a.79gn"G= = = 	""5#+666""1ch/////r   c                 j    t          t          t          t          j        t          j        d           dS )z$Rejection of unknown sampling method	not_Sobolr  N)assert_raises
ValueErrorr   r   r7   r   r   s    r   test_2_samplingz TestShgoFailures.test_2_sampling  s2    j$	7>&1	3 	3 	3 	3 	3 	3r   c                     ddd}t          t          j        t          j        d|d          }t          j                            d|j                   t          j                            d|j                   d	S )
zoCheck that the routine stops when no minimiser is found
           after maximum specified function evaluationsra   T)maxfevr   re   r   r   r   r   Frc   N)	r   
test_tabler7   r   r=   r   r  r   r  rV  s      r   test_3_1_no_min_pool_sobolz+TestShgoFailures.test_3_1_no_min_pool_sobol  sv      ! ! :<!2a#*, , ,""5#+666""2sx00000r   c                     ddd}t          t          j        t          j        d|d          }t          j                            d|j                   dS )	zoCheck that the routine stops when no minimiser is found
           after maximum specified sampling evaluationsra   Tmaxevr   re   r   r  FN)r   r  r7   r   r=   r   r  r   rV  s      r   test_3_2_no_min_pool_simplicialz0TestShgoFailures.test_3_2_no_min_pool_simplicial  sZ     ! !:<!2a#/1 1 1""5#+66666r   c                 Z    ddg}t          t          t          t          j        |           dS )zSpecified bounds ub > lb)rB   re   re   rb   Nr  r  r   r   r7   r   r   s     r   test_4_1_bound_errz#TestShgoFailures.test_4_1_bound_err  s(    &!j$	6:::::r   c                 Z    ddg}t          t          t          t          j        |           dS )z)Specified bounds are of the form (lb, ub))re   rb   rb   r
  Nr  r  s     r   test_4_2_bound_errz#TestShgoFailures.test_4_2_bound_err  s(    V$j$	6:::::r   c                     ddd}t          t          j        t          j        t          j        d|d          }t
          j                            d|j                   dS )zxEnsures the algorithm terminates on infeasible problems
           after maxev is exceeded. Use infty constraints optionr   Tr  r   r   r   r   r   FN	r   test_infeasibler7   r   r-   r=   r   r  r   rV  s      r   test_5_1_1_infeasible_sobolz,TestShgoFailures.test_5_1_1_infeasible_sobol  se      ! ! ?$o&<.3sG#*, , , 	""5#+66666r   c                     dddd}t          t          j        t          j        t          j        d|d          }t
          j                            d|j                   dS )zEnsures the algorithm terminates on infeasible problems
           after maxev is exceeded. Do not use infty constraints optionr   TF)r  r   r   r   r  Nr  rV  s      r   test_5_1_2_infeasible_sobolz,TestShgoFailures.test_5_1_2_infeasible_sobol  sh      (-/ / ?$o&<.3sG#*, , , 	""5#+66666r   c                     ddd}t          t          j        t          j        t          j        d|d          }t
          j                            d|j                   dS )z[Ensures the algorithm terminates on infeasible problems
           after maxev is exceeded.i  Fr  r   r   r  Nr  rV  s      r   test_5_2_infeasible_simplicialz/TestShgoFailures.test_5_2_infeasible_simplicial  se     ! " " ?$o&<.3sG#/1 1 1 	""5#+66666r   c                     t           j        dz   ddddd}t           j        t           j        f}t           j        dd|dd	}t          t          t          g|R i | dS )
z<Test Global mode limiting local evaluations with f* too highrw   r$  Tr4   FrT  Nr   rU  )r   r   r7   r   r-   r   UserWarningr   )r   r   r   kwargss       r   test_6_1_lower_known_f_minz+TestShgoFailures.test_6_1_lower_known_f_min  s|     )C/#'!&( ( 	7>*!($%,	  	k41$111&11111r   c                     ddl mm} g d}fdd_         ||          }t	          |j        |j        j                   d S )Nr   )r   r   r   r3   r  r  r  r  c                 :    xj         dz  c_          |           S Nr4   )r  )r6   r(   r   s    r   r(   z"TestShgoFailures.test.<locals>.fun  s    HHMHH588Or   )scipy.optimizer   r   r  r   r6   r(   )r   r   r   resultr(   r   s       @@r   r   zTestShgoFailures.test  s}    ........999	 	 	 	 	 	 c6""fh
CH-----r   N)r    r!   r"   r  r  r  r  r  r  r  r  r  r  r   r#   r   r   r  r    s        0 0 03 3 3

1 
1 
17 7 7; ; ;
; ; ;

7 
7 
77 7 7
7 
7 
72 2 2$. . . . .r   r  c                       e Zd Zd Zd ZdS )TestShgoReturnsc                     g d}fdd_         t          |          }t          j                            j         |j                    d S )Nr  c                 B    xj         dz  c_         t          |           S r!  r  r   r6   r(   s    r   r(   z3TestShgoReturns.test_1_nfev_simplicial.<locals>.fun      HHMHH88Or   r   r  r   r=   r   r  r   r   r#  r(   s      @r   test_1_nfev_simplicialz&TestShgoReturns.test_1_nfev_simplicial  sa    999	 	 	 	 	 c6""""38V[99999r   c                     g d}fdd_         t          |d          }t          j                            j         |j                    d S )Nr  c                 B    xj         dz  c_         t          |           S r!  r(  r)  s    r   r(   z.TestShgoReturns.test_1_nfev_sobol.<locals>.fun'  r*  r   r   r   r  r+  r,  s      @r   test_1_nfev_sobolz!TestShgoReturns.test_1_nfev_sobol$  sf    999	 	 	 	 	 c67;;;""38V[99999r   N)r    r!   r"   r-  r0  r#   r   r   r%  r%    s2        
: 
: 
:
: 
: 
: 
: 
:r   r%  c                     d } t          | dgdg          }t          |t          j        ddg                    }t	          t
          ddg|d          }t          j        t          j        |j        dz            dk              sJ t          j        t          j        |j        dz            d	k              sJ |j	        sJ d S )
Nc                 Z    t          j        |           } t          j        | dz            gS r  rY   asarrayr>   r?   s    r   quadz$test_vector_constraint.<locals>.quad3  $    JqMMqAvr   g@re   r   r  r   r]  r3   r   )
r   r   rY   r}  r   r   allr>   r6   r   )r5  nlcoldcr   s       r   test_vector_constraintr:  1  s          dSEA3
/
/C bhSz&:&:;;D
uw(dG
T
T
TC6"&#%!$$+,,,,,6"&#%A&&#-.....;r   zignore:delta_gradc                  R   d } t          | dgdg          }ddi}t          t          ddg|d|          }t          j        t          j        |j        d	z            dk              sJ t          j        t          j        |j        d	z            d
k              sJ |j        sJ d S )Nc                 Z    t          j        |           } t          j        | dz            gS r  r3  r?   s    r   r5  ztest_trust_constr.<locals>.quadB  r6  r   g@re   rl  r  r  r   )r   r   r   r3   r   )r   r   r   rY   r7  r>   r6   r   )r5  r8  r   r   s       r   test_trust_constrr=  @  s          dSEA3
/
/C .1 	')  C 6"&#%!$$+,,,,,6"&#%A&&#-.....;r   c                     dgdz  } d }t          |dd          }t          t          | |          }t          t	          j        |j                  d           d }d|d}t          t          | |          }t          t	          j        |j                  d           d	gd
z  } d }d|d}t          d | |d          }t          t	          j        |j                  d           d S )N)g?r   r3   c                 $    | d         | d         z   S r   r#   r?   s    r   faultyz)test_equality_constraints.<locals>.faulty[  s    tad{r   333333@r  c                 *    | d         | d         z   dz
  S )Nr   r4   rA  r#   r?   s    r   r@  z)test_equality_constraints.<locals>.faultyb  s    tad{S  r   r  r&   r  rd   c                 N    | d         | d         z   | d         z   | d         z   dz
  S r  r#   r?   s    r   r@  z)test_equality_constraints.<locals>.faultyk  s+    tad{QqT!AaD(1,,r   c                 ,    t          j        |            S r   )rY   prodr?   s    r   r  z+test_equality_constraints.<locals>.<lambda>q  s    BGAJJ, r   r   )r   r   r   r   )r   r   r   r   rY   r>   r6   )r   r@  r8  r   r   s        r   test_equality_constraintsrF  W  s   \AF   fc3
/
/C
uV
5
5
5CBF35MM3'''! ! !  //K
uV
=
=
=CBF35MM3'''Z!^F- - -  //K
	  C BF35MM3'''''r   c                      d } | dd}dddddd	}t          t          d
gdz  ||          }|j        d                                         dk    sJ |j        d         d         dk    sJ d S )Nc                 6    t          j        | dz            dz
  S r  )rY   r>   r?   s    r   r-   ztest_gh16971.<locals>.consy  s    vad||ar   r%   )r(   r'   rj  rb   r   g?)rhobegtolcatol)rl  r   r  r3   )r   r   rl  cobylar   rK  )r   r   r   lower)r-   cr   r   s       r   test_gh16971rO  x  s          f%%At<< 
 	y{<L	 	 	A h'--//8;;;;i(1T999999r   )	r#   r   r   NNNNr   r4   )Er   r  r=   rY   r  multiprocessingr   numpy.testingr   r   r   r   r  r   r"  r   r	   r
   r   r   r   r   r   scipy.optimize._constraintsr   scipy.optimize._shgor   r   r.   r0   r   r   r   rF   r}  r   r   rQ   r   r_   r   rt   r   r|   r   r*   r  boundsLJr  r   onesr  r   r  r   r  r   r'  r   r   r   r)  r  r%  r:  filterwarningsr=  rF  rO  r#   r   r   <module>rW     s
    



                   2 2 2 2 2 2 2 2  1 1 1 1 1 1 1 1H H H H H H H H H H H H H H H H H H H H = = = = = = % % % % % %+ + + + + + + +       $    +gw/"#Q) ) )
+ff-"#Q) ) )
+lL9"#Q) ) )    $    +gY",#/ #.%+
|0;}0:|0;}0;}0:|0;}0;}0;}0;}	/> 	#? 	#? %0EK 1K 1K 1K %L %L!  . +hZ",$0>"-%+
|n"="=$/EK$?$?	  # # # # #$ # # #J +gw/",j!9#&      $   D +i]Q."L "L "L#.      $    +k;7$7#8"&!57 7 7
    %   < 4TFQJ	**++	X$(6!%	
 	
 	
    $    
XL1,"% *
1 3
 
 
    (    _Y	$:+-$),c
4 4 4

    -   0 '&w.@4826) ) ) ,9=;?./    D3) 3) 3) 3) 3) 3) 3) 3)ti1 i1 i1 i1 i1 i1 i1 i1X]& ]& ]& ]& ]& ]& ]& ]&Bq. q. q. q. q. q. q. q.j: : : : : : : :4   /00  10,( ( (B: : : : :r   