a
    Rejz                     @   s   d dl Z d dlmZ d dlZd dlmZmZmZ d dl	m
Z d dlmZmZmZ d dlmZmZmZmZmZmZ dd Zd	d
 ZG dd dZG dd dZG dd dZG dd dZdd Zdd ZdS )    N)product)assert_allcloseassert_equalassert_)raises)
csr_matrix
csc_matrix
lil_matrix)_adjust_scheme_to_boundsapprox_derivativecheck_derivativegroup_columns_eps_for_method_compute_absolute_stepc                  C   s   g dg dg dg dg dg dg dg} t jtttfD ]b}|| }t d}t g d	}t||}t|| g d
}t g d}t||}t|| q<t|}t|}t|| d S )N)   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   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   r   r   r   )	npasarrayr   r   r	   arangearrayr   r   )	structureZ	transformAorderZgroups_truegroupsgroups_1groups_2 r    V/var/www/sunrise/env/lib/python3.9/site-packages/scipy/optimize/tests/test__numdiff.pytest_group_columns   s*    	



r"   c                  C   s   t t jj} | d | d | d d}dD ]8}ttt jt j|||  ttt jt j|||  q*t t jj} | d | d | d d}dD ]R}ttt jt j|||  ttt jt j|||  ttt jt j|||  qd S )N      ?UUUUUU?2-pointz3-pointcs)r   finfofloat64epsr   r   Z
complex128float32)EPSrelative_stepmethodr    r    r!   test_correct_fp_eps+   s@    r/   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )TestAdjustSchemeToBoundsc                 C   s  t d}t dd}t |}t |}|t j  |t j t||dd||\}}t|| tt 	| t||dd||\}}t|| tt 	| t||dd||\}}t|| tt 	|  t||dd||\}}t|| tt 	|  d S )Nr   {Gz?r   1-sidedr   2-sided)
r   zerosfullZ
empty_likefillinfr
   r   r   all)selfx0hZ	inf_lowerZ	inf_upper
h_adjusted	one_sidedr    r    r!   test_no_boundsP   s4    






z'TestAdjustSchemeToBounds.test_no_boundsc                 C   s   t g d}t d }t d}t g dd }t||dd||\}}t|| t||dd||\}}t|t g dd  t||dd	||\}}t|t | tt |  t||dd	||\}}t|t g dd  t|t g d
 d S )N)        g333333?g333333r   )r   r   皙?r   r2   r   )r   r@   r   r3   )FTT)	r   r   onesr
   r   absr   r8   r   )r9   r:   lbubr;   r<   _r=   r    r    r!   test_with_boundl   s$    

z(TestAdjustSchemeToBounds.test_with_boundc                 C   s  t ddg}t ddg}t ddg}t ddg}t||dd||\}}t|t ddg t||d	d||\}}t|t d
dg t||dd||\}}t|t ddg t|t ddg t||d	d||\}}t|t ddg t|t ddg d S )NgQg?r?   gQ?皙r   r2   gQr   g?r3   FTgQ?gQ)r   r   r
   r   r   )r9   rD   rE   r:   r;   r<   rF   r=   r    r    r!   test_tight_bounds   s$    z*TestAdjustSchemeToBounds.test_tight_boundsN)__name__
__module____qualname__r>   rG   rI   r    r    r    r!   r0   O   s   r0   c                   @   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dDddZdEd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d5d6 Zd7d8 Zd9d: Zd;d< Zd=d> Z d?d@ Z!dAdB Z"dCS )FTestApproxDerivativesDensec                 C   s
   t |S Nr   sinhr9   xr    r    r!   fun_scalar_scalar   s    z,TestApproxDerivativesDense.fun_scalar_scalarc                 C   s
   t |S rN   r   coshrQ   r    r    r!   jac_scalar_scalar   s    z,TestApproxDerivativesDense.jac_scalar_scalarc                 C   s,   t |d d t |d t |d gS Nr   r   r   r   tanexprQ   r    r    r!   fun_scalar_vector   s    z,TestApproxDerivativesDense.fun_scalar_vectorc                 C   s8   t d|d  t |d d t |d gddS Nr   r   r@   r   r   r   cosrZ   ZreshaperQ   r    r    r!   jac_scalar_vector   s
    (z,TestApproxDerivativesDense.jac_scalar_vectorc                 C   s$   t |d |d  t |d  S Nr   r   r   sinlogrQ   r    r    r!   fun_vector_scalar   s    z,TestApproxDerivativesDense.fun_vector_scalarc                 C   s    t |d t |t |gS Nr   rX   rQ   r    r    r!   wrong_dimensions_fun   s    z/TestApproxDerivativesDense.wrong_dimensions_func              	   C   s|   t |d t |d |d   t |d  t |d |d  |d   |d t |d |d   t |d  gS Nr   r   r   r   r_   rd   rc   rQ   r    r    r!   jac_vector_scalar   s    **z,TestApproxDerivativesDense.jac_vector_scalarc                 C   sH   t |d t |d  |d t |d  |d d |d d  gS Nr   r   r         ࿩r   r   rc   r_   rQ   r    r    r!   fun_vector_vector   s
    z,TestApproxDerivativesDense.fun_vector_vectorc                 C   s   t t |d |d t |d  g|d  t |d  t |d gd|d d  |d d  d|d d  |d d  ggS Nr   r   r   r   rl   g      rm   rQ   r    r    r!   jac_vector_vector   s
    "$6z,TestApproxDerivativesDense.jac_vector_vector      ?c                 C   s*   t t ||d  t ||d  gS ra   r   r   rZ   r9   rR   c0c1r    r    r!   fun_parametrized   s    z+TestApproxDerivativesDense.fun_parametrizedrA   c              
   C   s:   t |t ||d   dgd|t ||d   ggS ra   rr   rs   r    r    r!   jac_parametrized   s    z+TestApproxDerivativesDense.jac_parametrizedc                 C   s   t |dkr|S t jS )N:0yE>r   rC   nanrQ   r    r    r!   fun_with_nan   s    z'TestApproxDerivativesDense.fun_with_nanc                 C   s   t |dkrdS t jS )Nrx   rq   ry   rQ   r    r    r!   jac_with_nan   s    z'TestApproxDerivativesDense.jac_with_nanc                 C   s,   t |d |d  t |d |d  gS ra   )r   r   r_   rQ   r    r    r!   fun_zero_jacobian   s    z,TestApproxDerivativesDense.fun_zero_jacobianc              
   C   sV   t |d |d g|d  t |d |d   |d  t |d |d   ggS rh   )r   r   rc   rQ   r    r    r!   jac_zero_jacobian   s    >z,TestApproxDerivativesDense.jac_zero_jacobianc                 C   s   t | }t|S rN   )r   r   itemmathrZ   )r9   rR   Zxpr    r    r!   jac_non_numpy   s    z(TestApproxDerivativesDense.jac_non_numpyc                 C   sh   d}t | j|dd}t | j|}t | j|dd}| |}t||dd t||dd t||dd d S )	Nrq   r&   r.   r'   ư>rtol&.>-q=r   rS   rV   r   r9   r:   
jac_diff_2
jac_diff_3
jac_diff_4jac_truer    r    r!   test_scalar_scalar   s    
z-TestApproxDerivativesDense.test_scalar_scalarc                 C   sp   d}t | j|ddd}t | j|dd}t | j|ddd}| |}t||dd t||d	d t||d
d d S )Nrq   r&   "\O>r.   abs_step)r   r'   r   r   r   r   r   r   r    r    r!   test_scalar_scalar_abs_step   s    
z6TestApproxDerivativesDense.test_scalar_scalar_abs_stepc                 C   sn   d}t | j|dd}t | j|}t | j|dd}| t|}t||dd t||dd t||dd d S )	Nr#   r&   r   r'   r   r   r   r   )r   r[   r`   r   
atleast_1dr   r   r    r    r!   test_scalar_vector   s    z-TestApproxDerivativesDense.test_scalar_vectorc                 C   sr   t ddg}t| j|dd}t| j|}t| j|dd}| |}t||dd t||dd t||d	d d S )
N      Y@rl   r&   r   r'   r   r   Hz>r   )r   r   r   re   rj   r   r   r    r    r!   test_vector_scalar  s    
z-TestApproxDerivativesDense.test_vector_scalarc                 C   s~   t ddg}t| j|ddd}t| j|dt jd}t| j|ddd}| |}t||dd	 t||d
d	 t||dd	 d S )Nr   rl   r&   r   r   )r   rel_stepr'   r   r   A:)>r   )r   r   r   re   r7   rj   r   r   r    r    r!   test_vector_scalar_abs_step  s    
z6TestApproxDerivativesDense.test_vector_scalar_abs_stepc                 C   sr   t ddg}t| j|dd}t| j|}t| j|dd}| |}t||dd t||dd t||d	d d S )
N      Y皙?r&   r   r'   h㈵>r   r   r   r   r   r   rn   rp   r   r   r    r    r!   test_vector_vector   s    
z-TestApproxDerivativesDense.test_vector_vectorc                 C   s<   d}t tt| j| | t|}t tt| j||d d S )Nrq   )f0)assert_raisesRuntimeErrorr   rg   r   r   
ValueError)r9   r:   r   r    r    r!   test_wrong_dimensions,  s    z0TestApproxDerivativesDense.test_wrong_dimensionsc                 C   sZ   t ddg}t| j|ddd}t| j|dd}| |}t||dd t||dd d S )	NrH   rA   r&   g-C6?)r.   r   )r   r1   r   r   )r9   r:   r   r   r   r    r    r!   test_custom_rel_step4  s    
z/TestApproxDerivativesDense.test_custom_rel_stepc              
   C   s   t ddg}d}d}d}d}| j|||d}t ddg}| |||}t| j|d|||ft|d||fd	}	t| j||||ft|d||fd
}
t|	|dd t|
|dd d S )Nrq         r?          @)ru   gưr   r&   )r.   r   r   argskwargsbounds)r   r   r   r   r   r   r   r   )r   r   rv   rw   r   dictr   )r9   r:   rt   ru   rD   rE   r   r   r   r   r   r    r    r!   test_options>  s$    
z'TestApproxDerivativesDense.test_optionsc                 C   s|   t d }t d}t ddg}ttt| j|||fd t ddg}t| j|d||fd}| |}t||d	d
 d S )Nr   g       r   r   r   rq   r&   r.   r   r   r   )	r   rB   r   r   r   r   rn   rp   r   )r9   rD   rE   r:   jac_diffr   r    r    r!   test_with_bounds_2_pointP  s    

z3TestApproxDerivativesDense.test_with_bounds_2_pointc                 C   s   t ddg}t ddg}t ddg}| |}t| j|}t||dd t| j||t jfd}t||dd t| j|t j |fd}t||dd t| j|||fd}t||dd d S )Nrq   r   r   r   r   )r   r   rp   r   rn   r   r7   )r9   rD   rE   r:   r   r   r    r    r!   test_with_bounds_3_point^  s$    

z3TestApproxDerivativesDense.test_with_bounds_3_pointc                 C   s   t ddg}|d }|d }| |}t| j|d||fd}t||dd t| j|dd||fd}t||dd t| j|||fd	}t||dd t| j|d||fd
}t||dd d S )Ng      $@r   g&.!>r&   r   r   r   r.   r   r   r   r   r   )r   r   rp   r   rn   r   )r9   r:   rD   rE   r   r   r    r    r!   rI   t  s*    
z,TestApproxDerivativesDense.test_tight_boundsc                 C   s   d}d}d}|  |}t| j|dd||fd}t| j|d||fd}t||dd t||d	d d}|  |}t| j|dd||fd}t| j|d||fd}t||dd t||d	d d S )
N:0yErx   r?   r&   r   r   r   r   r   )r|   r   r{   r   )r9   rD   rE   r:   r   r   r   r    r    r!   test_bound_switches  s0    



z.TestApproxDerivativesDense.test_bound_switchesc                 C   sh   d}|  |}t| j |dd}t| j |}t||dd t||dd ttt| j |fi tdd d S )Nrq   r&   r   r   r   rx   r'   )r   r   r   r   	TypeErrorr   )r9   r:   r   r   r   r    r    r!   test_non_numpy  s    
z)TestApproxDerivativesDense.test_non_numpyc                    s  t jd dd fdd t jdddt jdt jdt jt d	d	g}t |d
fd}t |t jd
fd} |j	t jksJ t
||dd  fdd}t||d
d}||j	t jksJ t
||dd dd }dd }dd }dd }t jt jt jfD ]n}	t |	j}
t d|	}td |||d
}||||||
}t||d
|d||  t |k sJ qd S )Nr   c                 S   s   | d | d |  S ra   r    )prR   r    r    r!   func  s    z0TestApproxDerivativesDense.test_fp.<locals>.funcc                    s    | || S rN   r    )r   rR   y)r   r    r!   err  s    z/TestApproxDerivativesDense.test_fp.<locals>.errr   d   dtyper   r&   )r.   r   gMbP?)atolc                    s    |  tjS rN   )astyper   r+   )r   )r   rR   r   r    r!   err_fp32  s    z4TestApproxDerivativesDense.test_fp.<locals>.err_fp32r   c                 S   s
   t | S rN   r   rc   rR   r    r    r!   f  s    z-TestApproxDerivativesDense.test_fp.<locals>.fc                 S   s
   t | S rN   )r   r_   r   r    r    r!   g  s    z-TestApproxDerivativesDense.test_fp.<locals>.gc                 S   s   t |  S rN   r   r   r    r    r!   hess  s    z0TestApproxDerivativesDense.test_fp.<locals>.hessc                 S   s\   | d t t||t|||   }||  t t||t|||   }|| S rf   )maxr   rC   )r;   r:   r   r   r,   t0t1r    r    r!   	calc_atol  s    **z5TestApproxDerivativesDense.test_fp.<locals>.calc_atolrq   r   )r   randomseedZlinspacer)   r   r   r   r+   r   r   float16r(   r*   r   rC   )r9   Zp0Zjac_fp64Zjac_fpr   r   r   r   r   r   r,   r:   r;   r   r    )r   r   rR   r   r!   test_fp  sB    	z"TestApproxDerivativesDense.test_fpc                 C   s   t ddg}t| j| j|}t|dk  t| j| j|}t|dk  t ddg}t| j| j|}t|dk t| j| j|}t|dk d S )Ng      $
   r   r   r?   r   )r   r   r   rn   rp   r   r}   r~   )r9   r:   accuracyr    r    r!   test_check_derivative  s$    z0TestApproxDerivativesDense.test_check_derivativeN)rq   )rA   )#rJ   rK   rL   rS   rV   r[   r`   re   rg   rj   rn   rp   rv   rw   r{   r|   r}   r~   r   r   r   r   r   r   r   r   r   r   r   r   rI   r   r   r   r   r    r    r    r!   rM      s@   


9rM   c                   @   sL   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S )TestApproxDerivativeSparsec                 C   s   t jd d| _ddt | j  | _ddt | j  | _t | j| _d| jd d d  | jd d d< d| jdd d  | jdd d< | 	| j| _
d S )Nr   2   rH   r   rA   gP?r   )r   r   r   nr   rD   rE   emptyr:   jacJ_truer9   r    r    r!   setup_method  s      z'TestApproxDerivativeSparse.setup_methodc                 C   sD   |dd  d |d d d  }t dd| ft d| df S )Nr   r   r@   r   r   )r   Zhstack)r9   rR   er    r    r!   fun  s     zTestApproxDerivativeSparse.func                 C   s   |j }t||f}d|d  |d< d|d d  |d< td|d D ]d}d||d   |||d f< d	|| d  d
||   |||f< d||d  d  |||d f< qFd	|d d  |d< d|d  |d< |S )Nr   r   r   r   r   r   r   r   i	   r   r@   r@   r@   r]   r@   r]   )sizer   r4   range)r9   rR   r   Jir    r    r!   r     s    $"zTestApproxDerivativeSparse.jacc                 C   sb   t j||ftd}d|d< d|d< td|d D ]}d|||d |d f< q0d|d< d|d< |S )Nr   r   r   r   r   r   r   )r   r4   intr   )r9   r   r   r   r    r    r!   r     s    z$TestApproxDerivativeSparse.structurec                 C   s  |  | j}t| j}t||}tj| t||}tg d||gtj | j	gtj| j
gD ]\}}}}t| j| j|||f||fd}	tt|	t t|	 | jdd t| jd}
|
d d d  d9  < t| j| j||
||fd}	t|	 | jd	d q`d S )
Nr%   )r.   r   sparsityr   r   rx   r   r@   )r.   r   r   r   )r   r   r   r   r   r   shuffler   r7   rD   rE   r   r   r:   r   
isinstancer   r   toarrayr   Z	full_like)r9   r   r   r   r   r.   r   lur   r   r    r    r!   test_all!  s(    

z#TestApproxDerivativeSparse.test_allc                 C   s6   |  | j}t| j| j|d}t| | jdd d S )N)r   r   r   )r   r   r   r   r:   r   r   r   )r9   r   r   r    r    r!   test_no_precomputed_groups6  s    z5TestApproxDerivativeSparse.test_no_precomputed_groupsc                 C   sn   t j| j| jftd}t | j}dD ]B}t| j| j|d}t| j| j||f|d}t||	 ddd q&d S )Nr   r%   r   )r   r.   gV瞯<gVs<)r   r   )
r   rB   r   r   r   r   r   r:   r   r   )r9   r   r   r.   ZJ_denseZJ_sparser    r    r!   test_equivalence;  s    
z+TestApproxDerivativeSparse.test_equivalencec                    s`    fdd}t  j| j j jfd}t|dk  t  j| j j jfd}t|dk  d S )Nc                    s   t  | S rN   )r   r   r   r   r    r!   r   F  s    z=TestApproxDerivativeSparse.test_check_derivative.<locals>.jacr   r   )r   r   r:   rD   rE   r   )r9   r   r   r    r   r!   r   E  s    

z0TestApproxDerivativeSparse.test_check_derivativeN)rJ   rK   rL   r   r   r   r   r   r   r   r   r    r    r    r!   r     s   
r   c                   @   st   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S )"TestApproxDerivativeLinearOperatorc                 C   s
   t |S rN   rO   rQ   r    r    r!   rS   T  s    z4TestApproxDerivativeLinearOperator.fun_scalar_scalarc                 C   s
   t |S rN   rT   rQ   r    r    r!   rV   W  s    z4TestApproxDerivativeLinearOperator.jac_scalar_scalarc                 C   s,   t |d d t |d t |d gS rW   rX   rQ   r    r    r!   r[   Z  s    z4TestApproxDerivativeLinearOperator.fun_scalar_vectorc                 C   s8   t d|d  t |d d t |d gddS r\   r^   rQ   r    r    r!   r`   ]  s
    (z4TestApproxDerivativeLinearOperator.jac_scalar_vectorc                 C   s$   t |d |d  t |d  S ra   rb   rQ   r    r    r!   re   a  s    z4TestApproxDerivativeLinearOperator.fun_vector_scalarc              	   C   s|   t |d t |d |d   t |d  t |d |d  |d   |d t |d |d   t |d  gS rh   ri   rQ   r    r    r!   rj   d  s    **z4TestApproxDerivativeLinearOperator.jac_vector_scalarc                 C   sH   t |d t |d  |d t |d  |d d |d d  gS rk   rm   rQ   r    r    r!   rn   k  s
    z4TestApproxDerivativeLinearOperator.fun_vector_vectorc                 C   s   t t |d |d t |d  g|d  t |d  t |d gd|d d  |d d  d|d d  |d d  ggS ro   rm   rQ   r    r    r!   rp   r  s
    "$6z4TestApproxDerivativeLinearOperator.jac_vector_vectorc                 C   s   d}t | j|ddd}t | j|dd}t | j|ddd}| |}tjd tdD ]^}tjjd	dd
d}t|	||| dd t|	||| dd t|	||| dd qVd S )Nrq   r&   Tr.   as_linear_operatorr   r'   r   r   r   r   r   r   h㈵>)
r   rS   rV   r   r   r   r   uniformr   dotr9   r:   r   r   r   r   r   r   r    r    r!   r   y  s2    
z5TestApproxDerivativeLinearOperator.test_scalar_scalarc                 C   s   d}t | j|ddd}t | j|dd}t | j|ddd}| t|}tjd tdD ]d}tjjd	dd
d}t	|
||
|dd t	|
||
|dd t	|
||
|dd q\d S )Nr#   r&   Tr   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     s2    z5TestApproxDerivativeLinearOperator.test_scalar_vectorc                 C   s   t ddg}t| j|ddd}t| j|dd}t| j|ddd}| |}t jd td	D ]x}t jjd
d	|j	d}t
||t ||dd t
||t ||dd t
||t ||dd q`d S )Nr   rl   r&   Tr   r   r'   r   r   r   r   r   r   r   r   )r   r   r   re   rj   r   r   r   r   shaper   r   r   r   r    r    r!   r     s2    
z5TestApproxDerivativeLinearOperator.test_vector_scalarc                 C   s   t ddg}t| j|ddd}t| j|dd}t| j|ddd}| |}t jd td	D ]f}t jjd
d	|j	d}t
||||dd t
||||dd t
||||dd q`d S )Nr   r   r&   Tr   r   r'   r   r   r   r   r   r   r   r   )r   r   r   rn   rp   r   r   r   r   r   r   r   r   r    r    r!   r     s&    
z5TestApproxDerivativeLinearOperator.test_vector_vectorc                 C   s.   t ddg}ttt| j|ddt jfd d S )Nr   r   r&   r   r   )r   r   r   r   r   rn   r7   )r9   r:   r    r    r!   test_exception  s
    
z1TestApproxDerivativeLinearOperator.test_exceptionN)rJ   rK   rL   rS   rV   r[   r`   re   rj   rn   rp   r   r   r   r   r   r    r    r    r!   r   R  s   r   c                  C   s  dd } t | ddgddd}t|ddg t | ddgdd	d}t|ddg t | ddgdddgd}t|ddg t | ddgddd	gd}t|ddg t | ddgdd	dgd}t|ddg t | ddgddtj dfd
}t|ddg t | ddgdd	dtjfd
}t|ddg d S )Nc                 S   s&   t | d d  t | d d  S ra   )r   rC   r   r    r    r!   r     s    z"test_absolute_step_sign.<locals>.fr@   r&   rx   r   r   rq   r   )r.   r   r   )r   r   r   r7   )r   Zgradr    r    r!   test_absolute_step_sign  s4    	
r   c            
      C   sj  g d} t g d}t t jj}|d |d |d d}t d}| D ]}|| }t ||d |d |t |d  g}td |||}t|| | dkt	d d	 }	td | ||}t||	|  qJt g d
}t |d |d  |d |d d |d t |d  g}t|||d}t|| | dkt	d d	 }	t|| |d}t||	|  d S )Nr%   )r   r   r   g     j@r#   r$   rq   r   r   r   r   )rA   r   r   r   r&   )
r   r   r(   r)   r*   rC   r   r   r   float)
methodsr:   r,   r-   r   r.   r   Zcorrect_stepr   Zsign_x0r    r    r!   test__compute_absolute_step  s>    



r   )r   	itertoolsr   numpyr   Znumpy.testingr   r   r   Zpytestr   r   Zscipy.sparser   r   r	   Zscipy.optimize._numdiffr
   r   r   r   r   r   r"   r/   r0   rM   r   r   r   r   r    r    r    r!   <module>   s     $J  `[5