o
    tBh                     @   s   d Z ddlZddlZddlmZmZ ddlmZ ddlm	Z	m
Z
mZ ddlmZ ddlmZ ddlmZ ddlZdd	lmZmZmZmZmZmZmZ dd
lmZmZ ddlZG dd dZdS )z@
Unit tests for the differential global minimization algorithm.
    N)DifferentialEvolutionSolver_ConstraintWrapperdifferential_evolution)BoundsNonlinearConstraintLinearConstraint)rosen)
csr_matrix)Version)assert_equalassert_allcloseassert_almost_equalassert_array_equalassert_string_equalassert_suppress_warnings)raiseswarnsc                   @   s
  e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zejje e!j"e d5k d6d7d8d9 Z#d:d; Z$d<d= Z%d>d? Z&d@dA Z'dBdC Z(dDdE Z)dFdG Z*dHdI Z+dJdK Z,dLdM Z-dNdO Z.dPdQ Z/dRdS Z0dTdU Z1dVdW Z2dXdY Z3dZd[ Z4d\d] Z5d^d_ Z6d`da Z7dbdc Z8ddde Z9dfdg Z:dhdi Z;djdk Z<dldm Z=dndo Z>ejj?ejj@eAB dpkdqd7drds ZCdtdu ZDdvS )wTestDifferentialEvolutionSolverc                 C   sx   t jdd| _t ddgddgg| _ddg| _t| jdg| _t| jdgdd	d
| _	t 
t dddj}|| j	_d S )Nraise)invalid               @r   r   r   d   r               ?)popsizemutation皙?皙?)npseterr
old_seterrarraylimitsboundsr   	quadraticdummy_solverdummy_solver2
atleast_2darangeT
population)selfr1    r3   /var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/scipy/optimize/tests/test__differential_evolution.pysetup_method   s   

z,TestDifferentialEvolutionSolver.setup_methodc                 C   s   t jdi | j d S )Nr3   )r%   r&   r'   r2   r3   r3   r4   teardown_method-   s   z/TestDifferentialEvolutionSolver.teardown_methodc                 C   s   |d d S )Nr      r3   )r2   xr3   r3   r4   r+   0   s   z)TestDifferentialEvolutionSolver.quadraticc                 C   s  t t| jdd}t|jd t|jjd t t| jdd}t|jd t|jjd t t| jdd}t|jd t|jjd t t| jdd}t|jd t|jjd t t| jdd}t|jd t|jjd	 t t| jd
d}t|jd
 t|jjd t t| jdd}t|jd t|jjd	 t t| jdd}t|jd t|jjd	 t t| jdd}t|jd t|jjd t t| jdd}t|jd t|jjd t t| jdd}t|jd t|jjd t t| jdd}t|jd t|jjd d S )Nbest1expstrategy_best1best1binrand1bin_rand1rand1exprand2exp_rand2best2bin_best2rand2binrandtobest1bin_randtobest1randtobest1expcurrenttobest1bin_currenttobest1currenttobest1exp)r   r	   r*   r   r<   mutation_func__name__r2   solverr3   r3   r4   test__strategy_resolves3   s   z7TestDifferentialEvolutionSolver.test__strategy_resolvesc                 C   H   t dg}| jd}t|| t dg}| jd}t|| d S )Ng?r8                     ?)r%   r(   r-   r=   r   r@   r2   resulttrialr3   r3   r4   test__mutate1~   s   
z-TestDifferentialEvolutionSolver.test__mutate1c                 C   rR   )NgrS   r#   )r%   r(   r-   rE   r   rC   rY   r3   r3   r4   test__mutate2   s   
z-TestDifferentialEvolutionSolver.test__mutate2c                 C   s&   t dg}| jd}t|| d S )Ng333333?rS   )r%   r(   r-   rH   r   rY   r3   r3   r4   test__randtobest1   s   z1TestDifferentialEvolutionSolver.test__randtobest1c                 C   s(   t dg}| jdd}t|| d S )Nr#   r   rS   )r%   r(   r-   rK   r   rY   r3   r3   r4   test__currenttobest1   s   z4TestDifferentialEvolutionSolver.test__currenttobest1c                 C   s*   d}t | j| j|d}t|jt| d S )N)r    r   r"   )r   r+   r*   r   ditherlist)r2   r"   rP   r3   r3   r4   test_can_init_with_dithering   s   z<TestDifferentialEvolutionSolver.test_can_init_with_ditheringc                 C   s   t }d}ttt|| j|d d}ttt|| j|d dtjf}ttt|| j|d d}t|| j|d}td|j td |j	 d S )N)r    rT   r`   r   r#   r    )
r	   assert_raises
ValueErrorr   r*   r%   nanr   scalera   )r2   funcr"   rP   r3   r3   r4   +test_invalid_mutation_values_arent_accepted   s:   
zKTestDifferentialEvolutionSolver.test_invalid_mutation_values_arent_acceptedc                 C   sH   dd }t tdd t|ddg W d    d S 1 sw   Y  d S )Nc                 S   s   t t | d t | gS )Nr8   )r%   r(   sumr9   r3   r3   r4   rj      s   zETestDifferentialEvolutionSolver.test_invalid_functional.<locals>.funcz,func\(x, \*args\) must return a scalar value)matchr8   )rf   RuntimeErrorr   )r2   rj   r3   r3   r4   test_invalid_functional   s   "z7TestDifferentialEvolutionSolver.test_invalid_functionalc                 C   J   t dg}td| j| t dgdgg| j_td| j| d S )N333333?   r   r   )r%   r(   r   r,   _scale_parametersr)   r2   r[   r3   r3   r4   test__scale_parameters      z6TestDifferentialEvolutionSolver.test__scale_parametersc                 C   rs   )Nru   rt   r   r   )r%   r(   r   r,   _unscale_parametersr)   rw   r3   r3   r4   test__unscale_parameters   ry   z8TestDifferentialEvolutionSolver.test__unscale_parametersc                 C   sH   t g d}| j| t|d d tt |dk|dk  d S )N)g??r   g     r@gh㈵r8   r}   r   r   )r%   r(   r,   _ensure_constraintr   r   logical_andallrw   r3   r3   r4   test__ensure_constraint   s    z7TestDifferentialEvolutionSolver.test__ensure_constraintc                 C   s.   t | jdg}| }t|j| |j d S Nro   )r   r+   solver   funr9   r2   rP   rZ   r3   r3   r4   test_differential_evolution   s   z;TestDifferentialEvolutionSolver.test_differential_evolutionc                 C   s(   t | jdg}| }t|j|j d S r   )r   r+   r   r   r9   r   r3   r3   r4   test_best_solution_retrieval   s   z<TestDifferentialEvolutionSolver.test_best_solution_retrievalc                 C   sz   ddg}d}ddd}t t||d}t|j| ddd}t t||d}t|j| dd	d
}t t||d}|js;J d S )Nr   r8   z8callback function requested stop early by returning Truer   c                 S   s   dS )NTr3   paramconvergencer3   r3   r4   callback_python_true      zVTestDifferentialEvolutionSolver.test_callback_terminates.<locals>.callback_python_true)callbackc                 S   s   dgS )N
   r3   r   r3   r3   r4   callback_evaluates_true   s   zYTestDifferentialEvolutionSolver.test_callback_terminates.<locals>.callback_evaluates_truec                 S   s   g S Nr3   r   r3   r3   r4   callback_evaluates_false  r   zZTestDifferentialEvolutionSolver.test_callback_terminates.<locals>.callback_evaluates_false)r   )r   r	   r   messagesuccess)r2   r*   expected_msgr   rZ   r   r   r3   r3   r4   test_callback_terminates   s   


z8TestDifferentialEvolutionSolver.test_callback_terminatesc                 C   s2   dg}d}dd }t |||dd}t|jd d S )Nr   )      ?r   g      @c                 W   s8   t |tkr
td|d |d |   |d | d   S )Nzargs should be a tupler   r   r8   r   )typetuplerg   )r9   argsr3   r3   r4   r+     s   $zLTestDifferentialEvolutionSolver.test_args_tuple_is_passed.<locals>.quadraticT)r   polishgUUUUUU?)r   r   r   )r2   r*   r   r+   rZ   r3   r3   r4   test_args_tuple_is_passed  s   z9TestDifferentialEvolutionSolver.test_args_tuple_is_passedc                 C   s    t }dg}ttt||dd d S )NrT   abcr;   )r	   rf   rg   r   )r2   rj   r*   r3   r3   r4   test_init_with_invalid_strategy  s   
z?TestDifferentialEvolutionSolver.test_init_with_invalid_strategyc                 C   sV   t }dg}ttt|| ddg}ttt|| tt tddgddg}t|jd d S )Nr   r   )rT   rU   rV   r   r8   )r   r   )r	   rf   rg   r   r   r   r9   )r2   rj   r*   rZ   r3   r3   r4   test_bounds_checking%  s   z4TestDifferentialEvolutionSolver.test_bounds_checkingc           
      C   s   t jddddd}tt|dd d f |dd d f }td |dd}d}||d	\}}}}}	ttt 	t 
||||||	gd d S )
Ng      (@float64)dtyper8   rW   r   r   r!   rV   )r%   r/   reshaperb   zipr   _select_samplesr   lenuniquer(   )
r2   r)   r*   rP   	candidater1r2r3r4r5r3   r3   r4   test_select_samples7  s   & z3TestDifferentialEvolutionSolver.test_select_samplesc                 C   s4   t t| jdd}| }t|jd t|jd d S )Nr   )maxiterFz/Maximum number of iterations has been exceeded.)r   r	   r*   r   r   r   r   r   r3   r3   r4   test_maxiter_stops_solveA  s   z8TestDifferentialEvolutionSolver.test_maxiter_stops_solvec                 C   s   t t| jddd}| }t|jd t|jd t|jd t t| jdddd}| }t|jd	 t|jd t|jd t t| jddd
dd}| }t|jd
 t|jd t|jd d S )Nr   F)maxfunr   r8   z9Maximum number of function evaluations has been exceeded.rV   (   )r!   r   r   )   /   deferred)r!   r   r   updatingz8Maximum number of function evaluations has been reached.)r   r	   r*   r   r   nfevr   r   r   r3   r3   r4   test_maxfun_stops_solveJ  sD   
	z7TestDifferentialEvolutionSolver.test_maxfun_stops_solvec                 C   s0   t | jdgdd}|  tt|jd d S )Nr|   r   {Gz?tolr   )r   r+   r   r   r%   argminpopulation_energiesrO   r3   r3   r4   test_quadraticz  s   z.TestDifferentialEvolutionSolver.test_quadraticc                 C   s   t | jdgdd d S )Nr   r   r   )r   r+   r6   r3   r3   r4   test_quadratic_from_diff_ev  s   
z;TestDifferentialEvolutionSolver.test_quadratic_from_diff_evc                 C   sL   t | jdgdddd}t | jdgdddd}t|j|j t|j|j d S )Nr   Fr   r    )r   seedr   )r   r+   r   r9   r   )r2   rZ   result2r3   r3   r4   test_seed_gives_repeatability  s   z=TestDifferentialEvolutionSolver.test_seed_gives_repeatabilityz1.17z)Generator not available for numpy, < 1.17)reasonc              	   C   s8   t j }g d}|D ]}t| jdgd|d|d qd S )N)randomlatinhypercubesobolhaltonr   Fr    )r   r   r   init)r%   r   default_rngr   r+   )r2   rnginitsr   r3   r3   r4   test_random_generator  s   
z5TestDifferentialEvolutionSolver.test_random_generatorc                 C   s   t t| jddd}|  d S )Nr:   r   )r<   r   r   r	   r*   r   rO   r3   r3   r4   test_exp_runs  s   z-TestDifferentialEvolutionSolver.test_exp_runsc                 C   s   ddg}t t|ddd d S )N)rV   1   r   )r!   r   )r   r	   )r2   r*   r3   r3   r4   test_gh_4511_regression  s   	z7TestDifferentialEvolutionSolver.test_gh_4511_regressionc                 C   sF   t t| jdd}||j |  tt|j	d t|j
d d S )NrT   r   r   rW   )r   r	   r*   _calculate_population_energiesr1   _promote_lowest_energyr   r%   r   r   _nfevrO   r3   r3   r4   "test_calculate_population_energies  s
   zBTestDifferentialEvolutionSolver.test_calculate_population_energiesc           
      C   s   t t| jddd}t|\}}tt|dd t|jd tt	t| t t| j}t|\}}t
|D ]\}}|\}}	||	ksBJ ||	}}|dkrN d S q4d S )NrT      )r!   r   r   r8   2   )r   r	   r*   nextr   r%   sizer   rf   StopIteration	enumerate)
r2   rP   r9   r   _fun_previsoln	x_currentfun_currentr3   r3   r4   test_iteration  s"   

z.TestDifferentialEvolutionSolver.test_iterationc                 C   s,   t t| jddd}|  t|jdk  d S )N皙?F)r   r   )r   r	   r*   r   r   r   rO   r3   r3   r4   test_convergence  s
   
z0TestDifferentialEvolutionSolver.test_convergencec                 C   s   t t| jd d d}|  d S )N)r   r   r   rO   r3   r3   r4   test_maxiter_none_GH5731  s   
z8TestDifferentialEvolutionSolver.test_maxiter_none_GH5731c              	   C   s(  t ttgt| jfR i ddi tt| j}|  t|jd tt	
t	|j |  t|jd tt	
t	|j |jdd t|jd tt	
t	|j tt| jdd}|jdd t|jd tt	
t	|j t	dd	d
dd}tt| j|ddddd}t|jd tt	
t	|j t|jdk t|jdk t	||dd}t|jd d | tt	|jd d d tt	|jd d d t	dd	ddd	}t ttgt| jfR i d|i t	jjdddd}tt| j|d}t|jd |d  d S )Nr   rubbishr   r   )
qmc_enginer   )r   re   rT   r   rV   r8   rD   {Gz?r   )r   r<   atolr   r!   )rV   r8      r   r   )lowhighr   x0)rf   rg   r   r	   r*   init_population_randomr   r   r   r%   r   isinfr   init_population_lhsinit_population_qmclinspacer   num_population_memberspopulation_shapecliprz   r   r1   minmaxr   uniformr   )r2   rP   r1   unscaled_populationr   r3   r3   r4   test_population_initiation  sd   z:TestDifferentialEvolutionSolver.test_population_initiationc                 C   s`   t t| jddgd}|jsJ tt t t| jddgd W d    d S 1 s)w   Y  d S )Nr   r$   r    @)r   r	   r*   r   rf   rg   )r2   resr3   r3   r4   test_x08  s
   

"z'TestDifferentialEvolutionSolver.test_x0c                 C   s"   dd }ddg}t ||dd d S )Nc                 S   s   | d dk r	t jS | d S )Nr   r    r   )r%   infrm   r3   r3   r4   sometimes_infD  s   zWTestDifferentialEvolutionSolver.test_infinite_objective_function.<locals>.sometimes_infr   F)r*   dispr   )r2   r  r*   r3   r3   r4    test_infinite_objective_functionA  s   z@TestDifferentialEvolutionSolver.test_infinite_objective_functionc                 C   s@   ddg}t t|dd}t|jdk t|jjtu  |  d S )Nr   r   )r   )r   r	   r   	_updating_mapwrapper_mapfuncmapr   r2   r*   rP   r3   r3   r4   test_deferred_updatingK  s
   z6TestDifferentialEvolutionSolver.test_deferred_updatingc              	   C   s   ddg}t t|}t|jdk tt t t|dd}W d    n1 s'w   Y  W d    n1 s6w   Y  t|jdk d S )Nr   	immediater8   )workersr   )r   r	   r   r  r   UserWarningr  r3   r3   r4   test_immediate_updatingS  s   

z7TestDifferentialEvolutionSolver.test_immediate_updatingc              	   C   s   ddg}t d4}tt|d|jd}t|jjd u t|jdk |	  W d    n1 s1w   Y  W d    n1 s@w   Y  tt|ddd}t|jjd u t|jdk |	  W d    d S 1 slw   Y  d S )Nr   r8   r   )r   r  )
multiprocessingPoolr   r	   r  r   r	  poolr  r   )r2   r*   prP   r3   r3   r4   test_parallel`  s&   

 
"z-TestDifferentialEvolutionSolver.test_parallelc                 C   s&   t tddg}|  t|  d S )Nr   )r   r	   r   r   	convergedrO   r3   r3   r4   test_convergedo  s   z.TestDifferentialEvolutionSolver.test_convergedc                 C   s   dd }dd }t |tj d}ttddg|d}|ddg}t|d	 t |tj d
}ttddg||fd}|ddg}t|g d |ddg}t|g d |ddg}t|g d d S )Nc                 S      | d | d  gS Nr   r   r3   rm   r3   r3   r4   constr_fu     zNTestDifferentialEvolutionSolver.test_constraint_violation_fn.<locals>.constr_fc                 S   $   | d d | d  | d | d  gS Nr   r8   r   r3   rm   r3   r3   r4   	constr_f2x     $zOTestDifferentialEvolutionSolver.test_constraint_violation_fn.<locals>.constr_f2ffffff?r   constraintsr   r#   ?333333?rt   {Gz?r   r   r  g@r   r    r   r   r   )r   r%   r  r   r	   _constraint_violation_fnr   )r2   r  r  nlcrP   cvnlc2r3   r3   r4   test_constraint_violation_fnt  s$   


z<TestDifferentialEvolutionSolver.test_constraint_violation_fnc                 C   s  dd }dd }t |tj d}ttddg|d}|tddgd	d	gg\}}t|d
d
g t|tdgdgg |j	dksCJ t |tj d}ttddg||fd}|tddgddgg\}}t|d
d
g t|tg dg dg |tddgd	d	gg\}}t|d
d
g t|tg dg dg |j	dksJ |tddgd	d	gg\}}t|dd
g t|tg dg dg |j	dksJ d S )Nc                 S   r  r  r3   rm   r3   r3   r4   r    r  zZTestDifferentialEvolutionSolver.test_constraint_population_feasibilities.<locals>.constr_fc                 S   r  r  r3   rm   r3   r3   r4   r    r   z[TestDifferentialEvolutionSolver.test_constraint_population_feasibilities.<locals>.constr_f2r!  r   r"  r    r   Fr#   r  )r8   r   r$  g333333?)r#   r   r   r&  r(  )r8   rT   rX   Tr)  )
r   r%   r  r   r	   #_calculate_population_feasibilitiesr(   r   r   shape)r2   r  r  r+  rP   feasr,  r-  r3   r3   r4   (test_constraint_population_feasibilities  sB   

zHTestDifferentialEvolutionSolver.test_constraint_population_feasibilitiesc                 C   sz   dd }t |tj d}ttddg|d}tt | }W d    n1 s(w   Y  ||jdks6J |j	s;J d S )Nc                 S      t | d | d  gS r  r%   r(   rm   r3   r3   r4   r       zGTestDifferentialEvolutionSolver.test_constraint_solve.<locals>.constr_fr!  r   r"  )
r   r%   r  r   r	   r   r  r   r9   r   )r2   r  r+  rP   r  r3   r3   r4   test_constraint_solve  s   


z5TestDifferentialEvolutionSolver.test_constraint_solvec                 C   sF  dd }t |tj d}ttddg|ddd}tt | }W d    n1 s*w   Y  |jdks6J |j	r;J ttddg|d	d
}t
| |j rPJ t|j rZJ d}|jd }|j|dg |jd|g< |j|dgd d f |jd|gd d f< |j|dgd d f |jd|gd d f< |  t|jd | d S )Nc                 S   r3  r  r4  rm   r3   r3   r4   r    r5  zLTestDifferentialEvolutionSolver.test_impossible_constraint.<locals>.constr_fre   r   rT   r   )r#  r!   r   r   F)r#  r      )r   r%   r  r   r	   r   r  r   maxcvr   r   feasibler   isfiniter   constraint_violationr1   r   r   )r2   r  r+  rP   r  lr,  r3   r3   r4   test_impossible_constraint  s2   





(z:TestDifferentialEvolutionSolver.test_impossible_constraintc              
   C   sR  dd }t |tj d}ttddg|d}|j}|ddtdgd	dtdgs+J |d	dtdgddtdgd
ks@J |ddtdgddtdgsSJ |ddtdgd	d
td	gsfJ |dd
tddgd	d
td	d	gs{J |dd
tddgd	d
td	dgsJ |d	d
tddgd	d
td	dgd
ksJ d S )Nc                 S   r  r  r3   rm   r3   r3   r4   r    r  zCTestDifferentialEvolutionSolver.test_accept_trial.<locals>.constr_fr!  r   r"  r#   Tr   r   Fg#@r    g?)r   r%   r  r   r	   _accept_trialr(   )r2   r  r+  rP   fnr3   r3   r4   test_accept_trial  s.   
& &&z1TestDifferentialEvolutionSolver.test_accept_trialc                    s  t g d}t dt jdg}t g d}tt|||}||dk s*J |g ddk s7J t g d}t g dg dg d	g tt t j d|}||dk sbJ |g d
dk soJ ttt	 t j d|}||dk sJ |g d
dk sJ  fdd}t
|t j d}t|g d
}||dk sJ |g d
dk sJ d S )N)r   r7  ru   r    F   )r   r8   rT   r   )rX         )r   r8   rT   rU   )rV   r   r   rW   )r   r      r   )r   r8   r   rU   c                    s
     | S r   )dotrm   Ar3   r4   r   !  s   
zDTestDifferentialEvolutionSolver.test_constraint_wrapper.<locals>.fun)r%   r(   r  r   r   	violationanyr   r   r
   r   )r2   lbubr   pcr   	nonlinearr3   rF  r4   test_constraint_wrapper  s*   z7TestDifferentialEvolutionSolver.test_constraint_wrapperc                 C   s   dd }t |ddgddg}t|ddg}t|jd dks!J t|ddgd	d	g t|dd
gd	dg t|d
d
gddg t|ddgddg t|ddgddg d S )Nc                 S   s.   t | d d | d  | d d | d  gS r  r4  rm   r3   r3   r4   cons_f*  s   .zQTestDifferentialEvolutionSolver.test_constraint_wrapper_violation.<locals>.cons_fre   g333333r8   r    r   r   r   r%  r#   r'  g333333gRQ?r   g=
ףp=?)r   r   r%   r   r*   r   rH  r   )r2   rO  r+  rL  r3   r3   r4   !test_constraint_wrapper_violation)  s   zATestDifferentialEvolutionSolver.test_constraint_wrapper_violationc              	   C   s  dd }t d}d|dg df< d|ddd	gf< d
|dg df< d|dg df< d|dddgf< d
|dg df< d|dg df< d|dddgf< d
|dg df< |dd dd f }t g d}t|t j |}dgd dgd  dg }t||dd|dd}d}d }t||| |jsJ t|j|d!d" t|j	|d#d" t
t ||j |k t
t |jt |d d d$f k t
t |jt |d d df k tt|t j |}t||dd|dd}t||| |jsJ t|j|d!d" t|j	|d#d" t
t ||j |k t
t |jt |d d d$f k t
t |jt |d d df k d%d& }	d'd( }
t|d dd d f t j |d d }t|ddd d f t j |dd }t|	t j |dd }t|
t j |dd }||||f}t }|t t||d)d|dd}W d    n	1 sw   Y  t|j|d!d" t|j	|d#d" t
t ||j |k t
t |jt |d d d$f k t
t |jt |d d df k d S )*Nc                 S   sX   t dg| f} t d| dd  d| dd  | dd   t | dd   }|S )Nr   rV   r   )r%   hstackrl   r9   r   r3   r3   r4   f:  s   Dz2TestDifferentialEvolutionSolver.test_L1.<locals>.f)r      )r8   r8   r   r   r   )r   r8   r      )r   r8   r   )rp   re   r   rT   )rU   rV   r   rU   )r   rT   r   rU  rV   rU  rW   )rW   r   rU  r   )r8   rT   rU  r   rD  r   	   )rD  rW  r   )	r   r   r   r   r   r   r   r   r   r   r   r>     r<   r   r#  r!   )r   r   r   r   r   r   r   r   r   rT   rT   rT   r   igMb@?r   g{Gzt?r   c                 S   sL   t dg| f} d| d  d| d   | d  | d  d| d  | d  gS )Nr   r8   rT   rU  r   rV  r%   rQ  rm   r3   r3   r4   c1u  s   &z3TestDifferentialEvolutionSolver.test_L1.<locals>.c1c                 S   s,   t dg| f} d| d  | d  | d  S )Nr   rp   rD  rW  r   r[  rm   r3   r3   r4   c2z     z3TestDifferentialEvolutionSolver.test_L1.<locals>.c2r?   )r%   zerosr(   r   r  r   r   r   r9   r   r   r   r
   r   r   filterr  )r2   rS  rG  bLr*   r  x_optf_optr\  r]  L2NN2r#  supr3   r3   r4   test_L17  sr   


&&

&&((
&*z'TestDifferentialEvolutionSolver.test_L1c           
      C   s  dd }dd }t |dtj}dgd }|}t }|t t||dd	|d
}W d    n1 s2w   Y  d}d}	t||	| t|j| t|j	|	dd |j
sUJ ttt||j	dk tt|j	t|d d df k tt|j	t|d d df k d S )Nc                 S   s   t dg| f} | d d d d| d d d   | d d  d| d d	 d   d| d d
   d| d
 d   | d d  d| d
  | d   d| d
   d| d   }|S )Nr   r   r   r8   rV   r   rT   rU   rU  rW   r   rD  r[  rR  r3   r3   r4   rS    s   B


z2TestDifferentialEvolutionSolver.test_L2.<locals>.fc                 S   s   t dg| f} dd| d d   d| d d   | d  d| d d   d| d   dd	| d   | d d  d
| d
 d   d| d   dd| d   d| d   d| d d   | d  | d  d| d d  | d d  d| d  | d   d| d d   d| d
   d| d   gS )Nr      r8   r   rT   rU   rV         rW   rD  r   i  r   rU  r[  rm   r3   r3   r4   r\    s   F6:>

z3TestDifferentialEvolutionSolver.test_L2.<locals>.c1r   r   r   r?   rX  r<   r   r#  gy\
E@)gܤ@g$H8?g	޿g0ev@gn8)gk_@/?g0g+?h㈵>rZ  r   )r   r%   r  r   r`  r  r   r   r   r9   r   r   r   r(   )
r2   rS  r\  rf  r*   r#  rh  r  rd  rc  r3   r3   r4   test_L2  s(   


&*z'TestDifferentialEvolutionSolver.test_L2c                 C   s  dd }t d}d|dg df< d|dg df< d	|d
g df< |dd dd f }t g d}dd }t||t j}t|dt j}dgd }||f}t }	|	t t	||d|d
d}
W d    n1 siw   Y  d}d}t
|||dd t
|
j|dd t
|
j|dd |
jsJ tt ||
j |k tt t ||
jdk tt |
jt |d d df k tt |
jt |d d df k d S )Nc                 S   s   t dg| f} | d d | d d  | d | d   d| d   d| d   | d d d  d| d d	 d   | d	 d d  d| d
 d d   d	| d d   d| d d d   d| d d d   | d d d  d }|S )Nr   r   r8   rT     rT   r   rU   rV   rW   r   rD  rU  rW  -   r[  rR  r3   r3   r4   rS    s*   >z2TestDifferentialEvolutionSolver.test_L3.<locals>.f)rU   rU  )rm  r   rT   ir   )r   r8   r   rD  )r   rD     rp   r8   )rD  rp   r   r8   rT   )r   r8   rW  r   )ir   ic                 S   sX  t dg| f} d| d  d| d   d| d d d   d	| d
   d| d d d  d| d d d   d| d d   d	| d   d | d d  d| d d d   d| d  | d   d| d   d| d   d| d d  d| d   | d d d  d| d   d d| d d d  d| d d d   d| d d   | d  d gS )Nr   rT   r   rW   r8   r   rW  rD  r   r   r   rU   x   rT  rV   r   r   g      ru   r[  rm   r3   r3   r4   r\    s   6FL:Bz3TestDifferentialEvolutionSolver.test_L3.<locals>.c1r   r   r   rX  )r   r#  r!   )
gWXp?`@g@gVW@!@g3Ib@grq?g` ?g$F-t%?g6N#@g7h @g]y @g<6cN8@ro  rZ  ư>)r%   r_  r(   r   r  r   r   r`  r  r   r   r9   r   r   r   r   r2   rS  rG  ra  r\  rb  rf  r*   r#  rh  r  rc  rd  r3   r3   r4   test_L3  s8   
	


&*z'TestDifferentialEvolutionSolver.test_L3c              	   C   s  dd }t d}d|dddgf< d|d	g d
f< d|dddgf< |dd dd f }t g d}dd }t|t j d}t|dt j}dgdgd	  dgd  }||f}t }	|	t t	||dd|dd}
W d    n1 ssw   Y  d}g d}t
|||dd t
|
j|dd t dkrt t jjdk rt
|
j|ddd  n	t
|
j|d!d"d  |
jsJ tt ||
j |k tt t ||
jdk tt |
jt |d d df k tt |
jt |d d df k d S )#Nc                 S   s   t | d d S )NrT   )r%   rl   rm   r3   r3   r4   rS    r  z2TestDifferentialEvolutionSolver.test_L4.<locals>.f)rU   rW  ){Gzd?rx  r   rU   rW   )rx  rx  g{Gzdr8   )rV   r   rU   )r   g{GzrT   rD  rV   )r   r   r   c                 S   s   t dg| f} | d | d  d| d   d| d   d | d | d	  d
| d   | d | d   d
| d   | d | d  d | d | d   d| d   gS )Nr   r   rW   g!> 
@rU   r   gSUX@r8   r   i  rV   rT   rD  i i	  r[  rm   r3   r3   r4   r\    s
   *6.z3TestDifferentialEvolutionSolver.test_L4.<locals>.c1r   )r   '  )  ry  )r   rz  r?   rX  rY  gh|?@)got@g?@g:@gt$f@gVfyr@gh o?k@gCq@gVfx@MbP?rZ  Windowsgv!>gy&1l?)rtolr   gh㈵>ga2U0*c?)r%   r_  r(   r   r  r   r   r`  r  r   r   r   platformsystemr   intpitemsizer9   r   r   r   )r2   rS  rG  ra  r\  rb  rf  r*   r#  rh  r  rd  rc  r3   r3   r4   test_L4  s>   


&*z'TestDifferentialEvolutionSolver.test_L4c           	      C   s   dd }dd }t |tj d}dgd }|}t||dd	|d
}d}d}t|||dd t|j|dd |js:J ttt	||j
dk tt|j
t	|d d df k tt|j
t	|d d df k d S )Nc                 S   sf   t dg| f} t dt j | d  d t dt j | d   | d d | d | d    }| S )Nr   r8   r   rT   )r%   rQ  sinpirR  r3   r3   r4   rS    s
   2z2TestDifferentialEvolutionSolver.test_L5.<locals>.fc                 S   sD   t dg| f} | d d | d  d d| d  | d d d  gS )Nr   r   r8   rU   r[  rm   r3   r3   r4   r\  %  s   z3TestDifferentialEvolutionSolver.test_L5.<locals>.c1r   )r   r   r8   r?   rX  rn  )gJIť?gU(C@g#gh㈵>rZ  -C6?r   )r   r%   r  r   r   r   r   r   r   r(   r9   	r2   rS  r\  rf  r*   r#  r  rc  rd  r3   r3   r4   test_L5  s    


&*z'TestDifferentialEvolutionSolver.test_L5c           	      C   s   dd }dd }t |dtj}ddg}|}t||dd	|d
d}d}d}t|||dd t|j|dd t|j|dd |jsAJ tt	t
||jdk tt	|jt
|d d df k tt	|jt
|d d df k d S )Nc                 S   s4   t dg| f} | d d d | d d d  }|S )Nr   r   r   rT   r8   r7  r[  rR  r3   r3   r4   rS  <  s    z2TestDifferentialEvolutionSolver.test_L6.<locals>.fc                 S   sZ   t dg| f} | d d d | d d d  d | d d d  | d d d  d gS )Nr   r   rV   r8   r   rW   gp=
׳T@r[  rm   r3   r3   r4   r\  A  s   "$z3TestDifferentialEvolutionSolver.test_L6.<locals>.c1r   )   r   r   r?   rX  gHz>)r<   r   r#  r   )gq=
ף0,@g??g1ru  rZ  r{  r  r   )r   r%   r  r   r   r   r9   r   r   r   r(   r  r3   r3   r4   test_L6:  s"   

&*z'TestDifferentialEvolutionSolver.test_L6c           	      C   s0  dd }dd }t |g dg d}ddgd	gd
  }|}t||dd|d}g d}d}t||| t|j|dd t|j|dd |jsHJ ttt	||jt	g dk ttt	||jt	g dk tt|jt	|d d df k tt|jt	|d d df k d S )Nc                 S   sH   t dg| f} d| d d  d| d  | d   d| d   d	 }|S )
Nr   g+dvqn@rT   r8   g"C?r   rV   gtۈB@gn@r[  rR  r3   r3   r4   rS  X  s   "
z2TestDifferentialEvolutionSolver.test_L7.<locals>.fc                 S   s   t dg| f} dd| d  | d   d| d  | d   d	| d
  | d   dd| d  | d   d| d  | d   d| d
 d   dd| d
  | d   d| d  | d
   d| d
  | d   gS )Nr   gl#fUU@g+WPIw?r8   rV   g=BD?r   rU   gOlb?rT   gaۢ T@g+<$!6}?g h?g^5a?g/h"@gxLCs?gݗT?g^gED_?r[  rm   r3   r3   r4   r\  ^  s   ***z3TestDifferentialEvolutionSolver.test_L7.<locals>.c1)r   Z   r7  )\   n      )N   f   )!   rr  )   rr  rT   r?   rX  rn  )gq S@g] @@g˻f=@g:F@g%ShMcB@gCgbr{  rZ  r   r   )
r   r   r   r9   r   r   r   r%   r   r(   r  r3   r3   r4   test_L7V  s$   

((&*z'TestDifferentialEvolutionSolver.test_L7ppc64lezfails on ppc64lec              	   C   s  dd }t d}d|dddgf< d|dddgf< |dd dd f }t d	d	g}d
d }t||t j}t|t ddt dd}dgd dgd  }||f}t }	|	t	 t
||dd|dd}
W d    n1 snw   Y  d}d}t|||dd t|
jd d |d d dd t|
jdd  |dd  dd t|
j|dd |
jsJ tt ||
j |k tt t ||
jdk tt t ||
jdk tt |
jt |d d df k tt |
jt |d d df k d S )Nc                 S   sL   t dg| f} d| d  d| d d   d| d   d| d d   }|S )Nr   rT   r   ru  r8   g^>r[  rR  r3   r3   r4   rS    s   8z2TestDifferentialEvolutionSolver.test_L8.<locals>.f)rT   rV   )r   re   r   rU   rT   r8   皙c              	   S   s   t dg| f} dt | d  d  dt | d  d   d | d  dt | d d  dt | d | d  d   d | d  dt | d d  dt | d | d  d   d	 gS )
Nr   rz  rT   rX   rU   gfffff@r   r8   g33333;@)r%   rQ  r  rm   r3   r3   r4   r\    s   .22z3TestDifferentialEvolutionSolver.test_L8.<locals>.c1MbPr{  )r   i  )r  g?r>   rX  i  )r<   r   r#  r   )gEGr?@gT㥛D@g&D
n?gd,[ٿg#J{@rZ  gMb`?r   r   )r%   r_  r(   r   r  r   fullr   r`  r  r   r   r9   r   r   r   r   rv  r3   r3   r4   test_L8  s<   


	  
&*z'TestDifferentialEvolutionSolver.test_L8c           	      C   s   dd }dd }t |dgdg}dgd }|}t||d	d
|d}tdd dg}d}t||| tt|j|dd t|j|dd |jsJJ t	t
t||jdk t	t
t||jdk t	t
|jt|d d df k t	t
|jt|d d df k d S )Nc                 S   s,   t dg| f} | d d | d d d  S )Nr   r   r8   r[  rm   r3   r3   r4   rS    r^  z2TestDifferentialEvolutionSolver.test_L9.<locals>.fc                 S   s&   t dg| f} | d | d d  gS r  r[  rm   r3   r3   r4   r\    s   z3TestDifferentialEvolutionSolver.test_L9.<locals>.c1r  r{  rd   r8   r?   rX  rn  r    g      ?rZ  r   r   )r   r   r%   sqrtr   absr9   r   r   r   r   r(   r  r3   r3   r4   test_L9  s$   


&*z'TestDifferentialEvolutionSolver.test_L9N)ErN   
__module____qualname__r5   r7   r+   rQ   r\   r]   r^   r_   rc   rk   rr   rx   r{   r   r   r   r   r   r   r   r   r   r   r   r   r   pytestmarkskipifr   r%   __version__r   r   r   r   r   r   r   r   r  r  r  r  r  r  r.  r2  r6  r=  r@  rN  rP  ri  rp  rw  r  r  r  r  slowxfailr~  machiner  r  r3   r3   r3   r4   r      s    K
	

	0
	
	F	
)'Y'23.3r   ) __doc__r  r~  %scipy.optimize._differentialevolutionr   r   scipy.optimizer   scipy.optimize._constraintsr   r   r   r	   scipy.sparser
   scipy._lib._pep440r   numpyr%   numpy.testingr   r   r   r   r   r   r   r  r   rf   r   r   r3   r3   r3   r4   <module>   s    $