o
    h                     @   s4   d dl Zd dlmZmZ dd Zdd Zdd ZdS )	    N)_lbfgsbminimizec                 C   s   g d}g d}g d}g d}d}d}d}d}t g d	}	t g d
}
t g d}t g d}t | |r?|}|	}n%t | |rJ|}|
}nt | |rU|}|}nt | |r`|}|}ntdt |t |fS )z8simplified objective func to test lbfgsb bound violationg     ?g     ?gleffff?g%I$I?gbsE]?)      ?        r   r   r   )r   r   gDW?r   gB|m?)r   r   gҥ?r   g0Ԕ?gܥ+@g5s@g2a@g`_@)gFg ?gwQgb@gf漍B@)g8ۿg	MpB#?g8B92@g柇P?@g 3)g8F޿gƌJ?g%Gg;S9H@gMG@)g4</޿g dZ?gc4y?g(,7H@g @z<Simplified objective function not defined at requested point)nparrayallclose
ValueErrorcopy)xx0x1x2x3f0f1f2f3g0g1g2g3fg r   [/var/www/vscode/kcb/lib/python3.10/site-packages/scipy/optimize/tests/test_lbfgsb_setulb.pyobjfun   s8   r   c                  C   s  d} d}d}d}d}t j| fdt jd}t j| t jd}t j| t jd}t g d	}t |}	t jd
t jd}
t j| t jd}t jd| |  d|   d| |  d|  t jd}t jd|  t jd}t jdt jd}t jdt jd}t jdt jd}t jdt jd}t jdt jd}tdD ]-}t	|	\}
}t
||	||||
||||||||||| |	|k r|	|k sJ dqdS )zdtest if setulb() violates bounds

    checks for violation due to floating point rounding error
       
   g    cAgh㈵>      )shape
fill_valuedtype)r$   r   r               ,         z9_lbfgsb.setulb() stepped to a point outside of the boundsN)r   fullint32zerosfloat64onesr   r   ranger   r   setulball)nmfactrpgtolmaxlsnbdlow_bnd	upper_bndr   r   r   r   waiwataskln_tasklsaveisavedsaven_iterr   r   r   test_setulb_floatroundE   s<   
4rD   c                  C   s:   dd } t | tddgddd}tjj|jddd	 d S )
Nc                 S   s"   |  tj} t| d d|  fS )Nr!   )astyper   float32sum)r   r   r   r   fun_single_precisiont   s   z0test_gh_issue18730.<locals>.fun_single_precisionr   Tzl-bfgs-b)r   jacmethodr   gV瞯<)atol)r   r   r   testingassert_allclosefun)rH   resr   r   r   test_gh_issue18730q   s
   rP   )numpyr   scipy.optimizer   r   r   rD   rP   r   r   r   r   <module>   s
    @,