
    ^Mh|                        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         _/var/www/html/test/jupyter/venv/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 }| Nt          | t          t          f          s| f} n	 | D ]} |                    d| d           t          |          }nd }|S )Nineqtypefun)
isinstancetuplelistappend)gconss     r   wrap_constraintsr/      s~    D}!eT]++ 	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      @npsumr7   s    r   r-   zStructTest1.g.   s    """S())r   Nr    r!   r"   r8   r-   r/   r.   r#   r   r   r1   r1   *   s>        % % %* * * ADDDr   r1   )   )r   r   r   r5   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>   sinr6   s     r   r8   zStructTest2.fA   s    B"&))##r   c                 4    dt          j        | d          z
  S )N:   r   r;   r=   r@   s    r   r-   zStructTest2.gD   s    BF11%%%%%r   Nr    r!   r"   __doc__r8   r-   r/   r.   r#   r   r   rG   rG   <   sH         $ $ $& & & ADDDr   rG   )r   <   g⡽4$?0_r<g GK@g
]<@gI{_H@gi<&,@g>E@gqŤCQ4@g8$B@go.n:@g^5	>@)
rP   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   r4   r5   r#   r6   s     r   r8   zStructTest3.f{   s"    qtk!QqTaK//r   c                 *    | d         | d         z  dz
  S )Nr   r5         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   r4   r5   rU   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   r5   rU   r4   r#   r@   s    r   r-   zStructTest3.g   s8    tad{T!1Q419qtqy#84#???r   r   N)r    r!   r"   rN   r8   rV   rX   r-   r   r>   inf_StructTest3__nlcr.   r#   r   r   rR   rR   i   sv         "0 0 0" " ", , ,@ @ @  1bf--E8DDDr   rR   )r4   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   
   r4      r5               rC         r#   r6   s     r   r8   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 )Nr4   r   re   r5   rd   rb      r#   r@   s    r   rV   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   r5   ra   r4   rd   g     q@r#   r@   s    r   rX   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   r5   r4   rC   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   r4   r5   re   rb   rf   rC   r#   r@   s    r   g4zStructTest4.g4   sn    QqTQY1*Q1X!_<q1Q419}Lqt8 1Q4i( ) 	)r   Nr    r!   r"   rN   r8   rV   rX   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 Nr5   g     G@r          @r>   rJ   sqrtabsr6   s     r   r8   zStructTest5.f   s    dTkN26"'#ad3h!A$+.F*G*G"H"HIIId26"'#adadTk&:";";<<===>	
r   NrA   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           r5   r4         ?rw   )printNintrange)r   r7   argsksijabxdydzdeduds                 r   r8   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   NrM   r#   r   r   r|   r|      s>         
  ( 	AADDDr   r|   rC   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         ?r4   r5   re   r#   r6   s     r   r8   zStructTestS.f   sS    1!QqTCZA$55Q4#:!#$'(tcza&78 	9r   NrA   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         @r5   r\   d   r#   r6   s     r   r8   zStructTestTable.f
  s%    Q43;;1Q43;;23r   NrA   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 r3   r#   )r   r7   r   s      r   r8   zStructTestInfeasible.f  r9   r   c                 *    | d         | d         z   dz
  S Nr   r5   r#   r@   s    r   rV   zStructTestInfeasible.g1!  s    tad{Qr   c                 ,    | d         | d         z   dz
   S r   r#   r@   s    r   rX   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   )rB   r5   
Not a testr#   h㈵>r   sobolr5   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   r8   r   r.   r   logginginfor   r>   testingr   r7   r   r(   r   xlr   funl)testr   	test_atolr   r   r   r   r   r   r   ress              r   run_testr   7  sG    tvt{49%( 0'.A A AC
 
.3..L#   "

""35$/(1(1 	# 	3 	3 	3
 $

""37#'#4(1 	# 	3 	3 	3 #

""36#'#3(1 	# 	3 	3 	3 %

""38#'#5(1 	# 	3 	3 	3 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_sobol`       	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_sobole  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_sobolj  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_sobolp  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_sobolu  r   r   c                 .    t          t                     dS )%NLP: Hock and Schittkowski problem 18Nr   test3_1r   s    r   test_f3_sobolz(TestShgoSobolTestFunctions.test_f3_sobolz  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, multimodalrO   )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   rO   rb   )r   r   Nr   r   s    r   test_f5_2_sobolz*TestShgoSobolTestFunctions.test_f5_2_sobol  s     	Ba((((((r   N)r    r!   r"   rN   r   r   r   r   r   r   pytestmarkslowr   r   r   r#   r   r   r   r   Z  s         
  
  
+ + +  
  
   [6 6 6     
) ) ) ) )r   r   c                   D   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e
j                            d          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   r5   
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)]r5   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   r5   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   r5   r   r   Nr   r   s    r   test_f3_simplicialz2TestShgoSimplicialTestFunctions.test_f3_simplicial  s    A|<<<<<<r   c                 4    t          t          dd           dS )r   r5   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   rC   ,  r5   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   rC   Tr  r  r	  r5   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   r5   r   r   r   r   r   Nr   r   s     r   test_f5_2_cons_symmetryz7TestShgoSimplicialTestFunctions.test_f5_2_cons_symmetry  sG     !1v! ! 	C !-	/ 	/ 	/ 	/ 	/ 	/r   ra   c                 F    g ddd}t          t          d|dd           dS )	z(Asymmetrically constrained test function)r   r   r   re   Tr  i'  r5   r   r  N)r   test_sr   s     r   test_f5_3_cons_symmetryz7TestShgoSimplicialTestFunctions.test_f5_3_cons_symmetry  sJ      ,||! ! 	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)r7   avgs    r   <lambda>zFTestShgoSimplicialTestFunctions.test_f0_min_variance.<locals>.<lambda>  s    rwqzzC/? r   r&   rC   rD   r   r   r   V瞯<r   N)r   r>   varsuccessr   r(   r7   )r   r.   r   r  s      @r   test_f0_min_variancez4TestShgoSimplicialTestFunctions.test_f0_min_variance  s|     %?%?%?%?@@ 26!vh,DAAA{////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   rD   )r   r   ư>)r   N)r   r
   r   r   r(   r7   )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"   rN   r   r   r   r   r   r  r   r   r   r  r  r  r  	fail_slowr  skipr!  r&  r#   r   r   r   r     sJ        = = =
= = =
= = =
/ / /= = =
= = = [= = =/ / // / // / / [2/ / / [l##
$ 
$ $#
$ [l##1 1 $#1 1 1r   r   c                   D   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&e
j        '                    d!          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)Nr4   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)Nr5   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)Nr5   r   )r   r   r   r   r   r   r   )r   r   r8   r   r.   r>   r   r   r7   r   r(   r   r   r   s     r   test_2_2_sobol_iterz%TestShgoArguments.test_2_2_sobol_iter  sw    79gn',G= = = 	
""35'*<4d"SSS

""37G,@t"LLLLLr   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   r5   r   r   Tr   r   r   r   r   r   r   r   Nr   r   r   r8   r   r   r:  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 r8  r9  r@   s    r   r:  z<TestShgoArguments.test_3_2_disp_sobol.<locals>.callback_func)  r;  r   r5   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#   )r7   yzs      r   r  z5TestShgoArguments.test_args_gh14589.<locals>.<lambda>6  s    A	 r   )r   re   )r5   r4   )funcr   r   c                     d| z  dz   S )Nr4   r5   r#   r@   s    r   r  z5TestShgoArguments.test_args_gh14589.<locals>.<lambda>8  s    !a%!) r   )rH  r   N)r   r   r(   r7   )r   r   r%  s      r   test_args_gh14589z#TestShgoArguments.test_args_gh145893  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 evaluationsr$  Tr5   )rL  rM  r   
local_iterNr   r   rN  rO  r   s     r   test_4_2_known_f_minz&TestShgoArguments.test_4_2_known_f_minG  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$  Tr5   FrL  rM  r   rR  r   Nr   r   r   r   r   r   r   r   )r   r   r   r8   r   r.   r>   r   r   r7   r   r   r   r   s      r   test_4_4_known_f_minz&TestShgoArguments.test_4_4_known_f_minT  sz     )#'!&( ( 79gn',w#*, , , 	
""35'*<4d"SSSSSr   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   r8   r   r.   r>   r   r   r7   r   r4  s     r   test_5_1_simplicial_arglessz-TestShgoArguments.test_5_1_simplicial_arglessc  sF    79gn',GGG

""35'*<4d"SSSSSr   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   Nr[  r4  s     r   test_5_2_sobol_arglessz(TestShgoArguments.test_5_2_sobol_arglessh  sO    79gn',#*, , ,

""35'*<4d"SSSSSr   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_iterr4   r   r   r   r   r   r   r   Nr   r   r8   r   r.   r>   r   r   r7   r   r(   r   rW  s      r   test_6_1_simplicial_max_iterz.TestShgoArguments.test_6_1_simplicial_max_itern      q/79gn',"LB B B

""35'*<4d"SSS

""37G,@t"LLLLLr   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 )	ra  min_iterr4   r   rc  r   r   r   Nrd  rW  s      r   test_6_2_simplicial_min_iterz.TestShgoArguments.test_6_2_simplicial_min_iterv  rf  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COBYQA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\     4 	. 	.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   r8   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 )Nr4   r   r5   )r>   arrayTr@   s    r   jacz1TestShgoArguments.test_7_3_minkwargs.<locals>.jac  s*    xQqT1qt8 45577r   c                 6    t          j        ddgddgg          S Nr4   r   )r>   r  r@   s    r   hessz2TestShgoArguments.test_7_3_minkwargs.<locals>.hess  s    x!Q!Q 0111r   )rn  r  r  z	Solver = zd====================================================================================================r   ro  r   rp  N)r   r   r   r   )r   rq  r  r  r   s        r   test_7_3_minkwargsz$TestShgoArguments.test_7_3_minkwargs  s    ' 	. 	.F8 8 82 2 2 +1'*(, .  . L-V--...L###Wt&6%,. . . . .	. 	.r   c                 B    ddd}t          t          d d |d           d S )Nr5   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   rZ  Nru  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   r8   z0TestShgoArguments.test_10_finite_time.<locals>.f  s    Ju3r   rb   )r   r   r5   N)r   r   r   nit)r   r   r8   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)rL  r   ra   Nr   r   r   r5   )r   r   r8   r   r>   r   assert_equalr7   rW  s      r   test_11_f_min_0z!TestShgoArguments.test_11_f_min_0  s{    ! !79gn$"G= = =

358,,,

358,,,,,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  rL  r5   Nr   r   )r   r   r8   r   r>   r   r  r(   rW  s      r   test_12_sobol_inf_consz(TestShgoArguments.test_12_sobol_inf_cons  s[     $! !79gn"G= = =

SW-----r   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   r8   z/TestShgoArguments.test_13_high_sobol.<locals>.f      1r   rE   )   r   r   r4   r\   N)r   sampling_function)r   r8   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 moderR  rd   rO   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   Tr5   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   Trn  znelder-meadr   )r   r   r   Nr   )r   capsysr   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    9$$1a&$111r   rI   r   Nr   )r   r  s     r   test_17_custom_samplingz)TestShgoArguments.test_17_custom_sampling  s.    	2 	2 	2 	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   r8   z1TestShgoArguments.test_18_bounds_class.<locals>.f  s    9Q<<##%%%r   )g      r   g      )r   r   r]   )r   r   r~   )r+   zipr	   r   nfevmessager   r>   r  r   r   r7   )	r   r8   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%)?????%)?????''

"">#3U;;;

"">#3^5EFFFFFr   ra   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  r4   rI   )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    A$+rvbgc!A$*!t2L.M.M&N&NOOOA$rvbgc!A$!A$+*>&?&?@@AAABr   c                 `    d| d         z  d| d         z  z   d| d         z  z   d| d         z  z   S )	Nǧ8@r   g     :@r5   '   r4   g     @D@re   r#   r@   s    r   r8   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@r5   g333333&@r4   g?re   rb   r#   )r7   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'@r5   gfffffD@r4   gJ@re      gRQ?gQ?gRQ?g     4@gףp=
?)r>   ry   r@   s    r   rX   z6TestShgoArguments.test_20_constrained_args.<locals>.g2*  s    1Q4$qt)#d1Q4i/$qt);b@1qL4!a</$qtQw,>adAgM  r   c                 N    | d         | d         z   | d         z   | d         z   dz
  S Nr   r5   r4   re   r#   r@   s    r   h1z6TestShgoArguments.test_20_constrained_args.<locals>.h12  s+    Q4!A$;1%!,q00r   r%   )r   )r'   r(   r   r&   r  r	  r5   )r   r   r   r   )r   r   r   r   N)r   )r   r  r8   r   r  rX   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 Nr4   )r>   r?   powerr@   s    r   rH  z2TestShgoArguments.test_21_1_jac_true.<locals>.funcB  s%    6"(1a..))1q500r   rB   r5   r4   r   rb   r   rm  T)rn  r  )r   r   r   r   r   c                 :    t          j        | dz            d| z  fS r  r=   r@   s    r   rH  z2TestShgoArguments.test_21_1_jac_true.<locals>.funcN  s    6!q&>>1q5((r   r   re   )r   r   r   )r5   r5   r5   r5   r5   )x0r   r  r  r   N)r   r   r   r   r(   r7   )r   rH  r   r   r%  s        r   test_21_1_jac_truez$TestShgoArguments.test_21_1_jac_true>  s   	1 	1 	1 	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   r4   rb   r#   r@   s    r   	objectivezATestShgoArguments.test_21_2_derivative_options.<locals>.objectivea  s*    qt8ad?Q1X-11r   c                     d| d         z  dz   S )NrC   r   r4   r#   r@   s    r   gradientz@TestShgoArguments.test_21_2_derivative_options.<locals>.gradientd  s    qt8a<r   c                     dS NrC   r#   r@   s    r   r  z<TestShgoArguments.test_21_2_derivative_options.<locals>.hessg  r  r   c                     d|z  S r  r#   )r7   r  s     r   r  z=TestShgoArguments.test_21_2_derivative_options.<locals>.hesspj  s    q5Lr   r  rn  trust-constr)ir   r   r   r   )r  r   N)r   r   r   r>   r   r   r(   r7   )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 " " " {

""37CG444

""35#%00000r   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  rn  r{  r  rb   r  r   N)r   r   r   r   r   r>   zerosr   r   r(   r7   )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"$ $ $ubhqkk " "+ " " "{)))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#   )r7   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   )r5   r   )r   r   r   r5   )r   r   r   N)r   r   r>   r  r   r   r(   r7   )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{#*, , ,sBHQKKT#.0 0 0{)))su%%%%%r   N)/r    r!   r"   r-  r0  r2  r5  r@  rC  rJ  r   r   r   rP  rS  rX  r\  r_  re  ri  rr  rv  r  r  r  xfailr   sysplatformr  r  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M M MA A AA A A& & & [/ / / [
/ 
/ 
/T T TT T T
T T TM M MM M M
. 
. 
.A A A. . .(/ / /E E E [w:3<7#:G  I I
 
I I
- - - [?@@	. 	. A@	.	' 	' 	'1 1 1
* * *E E E8 8 8G G G( [2, , ,$& $& $&L2 2 2: [\+C+C+CDD1 1 ED1>2 2 2& & & & &r   r*  c                   t    e Zd Zd 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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maxiterr4   Nr   r   Frd   )	r   r   r8   r   r>   r   r  r   tnevrW  s      r   test_1_maxiterzTestShgoFailures.test_1_maxiter  sl    a.79gn"G= = = 	
s{333

38,,,,,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   r8   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_tabler8   r   r>   r   r  r   r  rW  s      r   test_3_1_no_min_pool_sobolz+TestShgoFailures.test_3_1_no_min_pool_sobol  sv      ! ! :<!2a#*, , ,

s{333

CH-----r   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  r8   r   r>   r   r  r   rW  s      r   test_3_2_no_min_pool_simplicialz0TestShgoFailures.test_3_2_no_min_pool_simplicial  sZ     ! !:<!2a#/1 1 1

s{33333r   c                 Z    ddg}t          t          t          t          j        |           dS )zSpecified bounds ub > lb)rC   re   re   rb   Nr   r  r   r   r8   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_infeasibler8   r   r.   r>   r   r  r   rW  s      r   test_5_1_1_infeasible_sobolz,TestShgoFailures.test_5_1_1_infeasible_sobol  se      ! ! ?$o&<.3sG#*, , , 	
s{33333r   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  rW  s      r   test_5_1_2_infeasible_sobolz,TestShgoFailures.test_5_1_2_infeasible_sobol  sh      (-/ / ?$o&<.3sG#*, , , 	
s{33333r   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  rW  s      r   test_5_2_infeasible_simplicialz/TestShgoFailures.test_5_2_infeasible_simplicial  se     ! " " ?$o&<.3sG#/1 1 1 	
s{33333r   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$  Tr5   FrU  Nr   rV  )r   r   r8   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   r4   r"  r"  r"  r"  c                 :    xj         dz  c_          |           S Nr5   )r  )r7   r(   r   s    r   r(   z"TestShgoFailures.test.<locals>.fun	  s    HHMHH588Or   )scipy.optimizer   r   r  r   r7   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   thread_unsafer  r   r#   r   r   r  r    s        - - -3 3 3

. 
. 
.4 4 4; ; ;
; ; ;

4 
4 
44 4 4
4 
4 
4 [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   r7   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""

&+66666r   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/  r0  s      @r   test_1_nfev_sobolz!TestShgoReturns.test_1_nfev_sobol!  sf    999	 	 	 	 	 c67;;;

&+66666r   N)r    r!   r"   r1  r4  r#   r   r   r)  r)    s2        
7 
7 
7
7 
7 
7 
7 
7r   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  r>   asarrayr?   r@   s    r   quadz$test_vector_constraint.<locals>.quad0  $    JqMMqAvr   g@re   r   r  r   r^  r4   r   )
r   r   r>   r  r   r   allr?   r7   r   )r9  nlcoldcr   s       r   test_vector_constraintr>  .  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  r7  r@   s    r   r9  ztest_trust_constr.<locals>.quad?  r:  r   g@re   rn  r  r  r   )r   r   r   r4   r   )r   r   r   r>   r;  r?   r7   r   )r9  r<  r   r   s       r   test_trust_constrrA  =  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   r4   c                 $    | d         | d         z   S r   r#   r@   s    r   faultyz)test_equality_constraints.<locals>.faultyX  s    tad{r   333333@r  c                 *    | d         | d         z   dz
  S )Nr   r5   rE  r#   r@   s    r   rD  z)test_equality_constraints.<locals>.faulty_  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   rD  z)test_equality_constraints.<locals>.faultyh  s+    tad{QqT!AaD(1,,r   c                 ,    t          j        |            S r   )r>   prodr@   s    r   r  z+test_equality_constraints.<locals>.<lambda>n  s    BGAJJ, r   r   )r   r   r   r   )r   r   r   r   r>   r?   r7   )r   rD  r<  r   r   s        r   test_equality_constraintsrJ  T  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  r=   r@   s    r   r.   ztest_gh16971.<locals>.consv  s    vad||ar   r%   )r(   r'   rk  rb   r   g?)rhobegtolcatol)rn  r   r  r4   )r   r   rn  cobylar   rO  )r   r   r   lower)r.   cr   r   s       r   test_gh16971rS  u  s          f%%At<< 
 	y{<L	 	 	A h'--//8;;;;i(1T999999r   )	r#   r   r   NNNNr   r5   )Er   r  numpyr>   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/   r1   r   r   r   rG   r  r   r   rR   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>  filterwarningsrA  rJ  rS  r#   r   r   <module>r\     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",#/ #+"(ZL-8M-7L-8M-8M-7L-8M-8M-8M-8M	,; 	#< 	#< %-BH .H .H .H %I %I!  . +hZ",$0>"*"(ZL>":":$,BHl^$<$<	  # # # # #$ # # #J +gw/",j!9#&      $   D +i]Q."L "L "L#.      $    +k;7$7#8"&!57 7 7
    %   < 4TFQJ	**++	X$(6!%	
 	
 	
    $    
XL1,"% '

S 0
 
 
    (    _Y	$:+-$),c
4 4 4

    -   0 '&w.@4826) ) ) ,9=;?./    D3) 3) 3) 3) 3) 3) 3) 3)tj1 j1 j1 j1 j1 j1 j1 j1ZY& Y& Y& Y& Y& Y& Y& Y&zr. r. r. r. r. r. r. r.l7 7 7 7 7 7 7 74   /00  10,( ( (B: : : : :r   