o
    i                    @   s  d Z ddlZddlZddlZddlmZmZmZm	Z	m
Z
mZ ddlmZ ddlmZmZ ddlmZ ddlmZ ddlmZ dd	lmZ ddlZddlZd
ZzddlmZ W n eya   dZY nw d
Z zddl!Z!ddl"m#Z$ W n ey{   dZ Y nw dd Z%dd Z&dd Z'dd Z(		diddZ)dd Z*dd Z+djddZ,dd  Z-dkd$d%Z.d&d' Z/ej0j1d(d) Z2d*d+ Z3ej0j1d,d- Z4d.d/ Z5d0d1 Z6dZ7dZ8dZ9dZ:dZ;G d2d3 d3Z<ej0=d4G d5d6 d6e<Z>ej0=d4G d7d8 d8e<Z?ej0=d4G d9d: d:e<Z@G d;d< d<e<ZAG d=d> d>e>ZBG d?d@ d@e>ZCG dAdB dBe>ZDG dCdD dDe?ZEe r7G dEdF dFe?ZFerBG dGdH dHe?ZGG dIdJ dJe?ZHG dKdL dLe?ZIej0=d4G dMdN dNZJG dOdP dPe@ZKG dQdR dRe@ZLG dSdT dTeAZMG dUdV dVeAZNG dWdX dXZOej0=d4G dYdZ dZZPG d[d\ d\ePZQG d]d^ d^ePZRG d_d` d`ePZSej0=d4G dadb dbZTG dcdd ddeTZUG dedf dfeTZVG dgdh dheTZWdS )lz"
Unit test for Linear Programming
    N)assert_assert_allcloseassert_equalassert_array_lessassert_warnssuppress_warnings)raises)linprogOptimizeWarning)approx_derivative)MatrixRankWarning)LinAlgWarning)VisibleDeprecationWarningT)UmfpackWarningF)choleskyc                 C   s0   t | j d t | j|k d t| jdd d S )NIncorrectly reported successz)Incorrectly reported number of iterations   z(Failed to report iteration limit reachedr   successr   status)resmaxiter r   [/home/kim/smarthome/.venv/lib/python3.10/site-packages/scipy/optimize/tests/test_linprog.py_assert_iteration_limit_reached!   s   r   c                 C       t | j d t| jdd d S )Nincorrectly reported success   z"failed to report infeasible statusr   r   r   r   r   _assert_infeasible'      r   c                 C   r   )Nr      z!failed to report unbounded statusr   r   r   r   r   _assert_unbounded-   r    r"   c                 C   s"   t | j d t | jdv d d S )Nr   r      z%failed to report optimization failure)r   r   r   r   r   r   r   )_assert_unable_to_find_basic_feasible_sol3   s   r%   :0yE>c                 C   sn   | j sd| j d| j }t|t| jd |d ur%t| j|d||d |d ur5t| j|d||d d S d S )Nzlinprog status z, message: r   z*converged to an unexpected objective valueerr_msgrtolatolz#converged to an unexpected solution)r   r   messageAssertionErrorr   r   funx)r   desired_fun	desired_xr)   r*   msgr   r   r   _assert_success=   s   
r2   c                 C   s  t jd}| | d d  d }t | d | d  d }|| d | | }t | d | | f}g }g }t| d D ]}| }d||ddddf< ||	  |d q:t| D ]"}t| D ]}	| }d|dd||	f< ||	  |d qbq\t| D ]&}| }|dd|ddf |dd|ddf< ||	  || qt| D ]&}| }|dddd|f |dddd|f< ||	  || q| }|ddt| t| f |ddt| t| f< ||	  || | }|ddt| td|  d df |ddt| td|  d df< ||	  || t j
t |td}
t j
|td}||
jd }|
||||fS )z
    Generates a linear program for which integer solutions represent an
    n x n magic square; binary decision variables represent the presence
    (or absence) of an integer 1 to n^2 in each position of the square.
    r   r   r   r$   NZdtype)nprandomRandomStatearangereshapezerosrangecopyappendflattenarrayvstackfloatrandshape)nrngMnumbersr:   ZA_listZb_listiZA_rowjAbcr   r   r   magic_squareQ   sT   ((0
""
rM   c                 C   s   t jd}|j| |fd }t | | | f}t | }t| D ]}|d }d|||| || f< ||  ||< q!t || | f}t |}	t|D ]}|d }d|||d|f< d|	|< qLt ||f}
t ||	f}|
|| fS )z -> A b c LP test: m*n vars, m+n constraints
        row sums == n/m, col sums == 1
        https://gist.github.com/denis-bz/8647461
    r   sizer   N)	r5   r6   r7   Zexponentialr:   r;   r@   ZhstackZravel)mrD   rE   rL   ZArowZbrowrI   Zj1ZAcolZbcolrJ   rK   r   r   r   lpgen_2d   s"   


rQ   c                 C   s   t j| }d\}}}||d }|||d }||d }|||d }||d }	|| }
||}t j |
|
|  k < t j||| k< t |
|fj}|||||	|fS )N)
      2         ?)r5   r6   r7   rB   infr@   T)seedrE   Zm_eqZm_ubrD   rL   A_ubb_ubA_eqb_eqlbubboundsr   r   r   very_random_gen   s   

r`   c                  C   sZ   g d} g dg dg dg dg}g d}g dg}dg}g d	}d
}| ||||||fS )N)r3      r$   )rc      	   )r   r3   r   )rR   rc      )rd   r3   r!   r$   )rf   rd   rb   rd   )rg   r   r   )gk椀?gs?r   gkL?gqY:^@r   )rL   rY   rZ   r[   r\   x_starf_starr   r   r   nontrivial_problem   s   
rm   ra   re   d   c                 C   sf  t j| }|jdd||fd}||}|jdd||fd}|||  |jdd|d }t || }	d|	d|< tjd| || f}
d}t	|D ]<}||dd|f  |
|d|f< d|
||| f< d| |dd|f  |
|d d|f< d|
|d || f< |d7 }qL|

 }
t d| }||ddd< | |ddd< dg| d	g|  }|	|
||fS )
z
    Training data is {(x0, y0), (x1, y2), ..., (xn-1, yn-1)}
        x in R^d
        y in R
    n: number of training samples
    d: dimension of x, i.e. x in R^d
    phi: feature map R^d -> R^m
    m: dimension of feature space
    r   r   rN   h㈵>Nr   r3   NNr   N)r5   r6   r7   normalrandnonesscipysparseZ
lil_matrixr;   Ztocscr:   )rX   rP   drD   rE   phiZw_truer.   yrL   rY   idxiirZ   bndsr   r   r   l1_regression_prob   s*   

 (
r}   c                    s   i   fdd}t ddg}ddgddgddgg}g d}t||||| jd	}t|d
ddgd t d |d  t d |d  t d |d  t d |d  d S )Nc                    s   |  d}|  d}t|  ddv  t|  dtdv  tt|  dt tt|t tt|t | d  d< | d	  d	< | d
  d
< | d  d< d S )Nr+   completeZphaser   r   r   r$   nitr.   r-   slackcon)popr   r;   
isinstanceintboolstr)r   r+   r~   Zlast_cbr   r   cb   s   

z!generic_callback_test.<locals>.cbrf   r   r   r   rR   ra   r$   rY   rZ   callbackmethod      2rd   r/   r0   r-   r.   r   r   )r5   r?   r	   r   r2   r   selfr   rL   rY   rZ   r   r   r   r   generic_callback_test   s   r   c                  C   s   t ddg} ddgddgddgg}g d}ttt| ||dd ttt| ||d	d d
}tjt|d t| ||ddid W d    d S 1 sIw   Y  d S )Nrf   r   r   r   r   r   ekki-ekki-ekkirY   rZ   r   z
highs-ekkiz>Unrecognized options detected: {'rr_method': 'ekki-ekki-ekki'}match	rr_methodrY   rZ   options)r5   r?   assert_raises
ValueErrorr	   pytestwarnsr
   )rL   rY   rZ   r+   r   r   r    test_unknown_solvers_and_options  s   "r   c                  C   sR   t ddg} ddgddgddgg}g d}t| ||dd}t|d	dd
gd d S )Nrf   r   r   r   r   r   highsr   r   rd   r   )r5   r?   r	   r2   )rL   rY   rZ   r   r   r   r   test_choose_solver  s
   r   c                   C   s   t t tddd W d    n1 sw   Y  t t tddd W d    n1 s1w   Y  t t tddd W d    d S 1 sMw   Y  d S )Nr   interior-pointr   revised simplexsimplex)r   r   DeprecationWarningr	   r   r   r   r   test_deprecation(  s   "r   c               	   C   s  t ddd} d}| jdksJ | j|sJ td\}}}}}dgt| }dgt| }dd	i}	t ||||d|	|d
} d}| jdksFJ | j|sNJ ddi}	t ||||d|	d} d}| jdkseJ | j|smJ t dddd} d}| jdks}J | j|sJ t ddd} d}| jdksJ | j|sJ ddlm}
 |
dd\}}d}|dksJ ||sJ |
d d \}}d}|dksJ ||sJ d S )Nr   r   r   z6Optimization terminated successfully. (HiGHS Status 7:r   rd   r   r   
time_limit皙?)rL   r[   r\   r_   r   r   integralityz%Time limit reached. (HiGHS Status 13:r   rR   highs-dsrL   r[   r\   r_   r   r   z*Iteration limit reached. (HiGHS Status 14:)r   r3   )r_   r   z+The problem is infeasible. (HiGHS Status 8:r   r3   z+The problem is unbounded. (HiGHS Status 10:r!   )_highs_to_scipy_status_message:   zHello!z;The HiGHS status code was not recognized. (HiGHS Status 58:r$   z>HiGHS did not provide a status code. (HiGHS Status None: None))r	   r   r+   
startswithrM   lenZscipy.optimize._linprog_highsr   )r   r1   rJ   rK   rL   rG   rF   r_   r   r   r   r   r+   r   r   r   test_highs_status_message2  sL   r   c                   C   s$   t ddgddggdgddgd d S )Nr   r3   r         rY   rZ   r   )r	   r   r   r   r   test_bug_17380_  s   $r   c                   @   s  e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
ejjdd Zejjdd Zdd Zdd Zdd Zdd Zdd Zejj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d0d1 Zd2d3 Zd4d5 Z d6d7 Z!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 Z/dTdU Z0dVdW Z1dXdY Z2dZd[ Z3d\d] Z4d^d_ Z5d`da Z6dbdc Z7ddde Z8dfdg Z9dhdi Z:djdk Z;dldm Z<dndo Z=dpdq Z>drds Z?dtdu Z@dvdw ZAdxdy ZBdzd{ ZCd|d} ZDd~d ZEdd ZFdd ZGdd ZHdd ZIdd ZJdd ZKdd ZLdd ZMdd ZNdd ZOdd ZPdd ZQdd ZRejjSeTjUdkdddd ZVdS )LinprogCommonTestsa  
    Base class for `linprog` tests. Generally, each test will be performed
    once for every derived class of LinprogCommonTests, each of which will
    typically change self.options and/or self.method. Effectively, these tests
    are run for many combination of method (simplex, revised simplex, and
    interior point) and options (such as pivoting rule or sparse treatment).
    c                 C   s   t |  d S N)r   r   r   r   r   test_callback{  s   z LinprogCommonTests.test_callbackc                 C   8   t dd\}}}t|||| jddid}t|dd d S )NrS   dispTrY   rZ   r   r   T*Pr/   rQ   r	   r   r2   r   rJ   rK   rL   r   r   r   r   	test_disp~  
   zLinprogCommonTests.test_dispc                 C   sT   ddg}ddgddgg}ddg}d}d}t |||||f| j| jd	}t|d
d d S )Nr3   r$   rf   r   r   rd   rp   rf   N)rY   rZ   r_   r   r   r   )r	   r   r   r2   )r   rL   rJ   rK   	x0_bounds	x1_boundsr   r   r   r   test_docstring_example  s   z)LinprogCommonTests.test_docstring_examplec              	   C   s0   dg}dgg}d}t tt|||| j| jd d S )Nr   Zhellor[   r\   r   r   )r   	TypeErrorr	   r   r   )r   rL   r[   r\   r   r   r   test_type_error  s   
z"LinprogCommonTests.test_type_errorc              
   C   sr   t dg}t dgg}t dg}| }dt jf}t|||tt|| j| jd}t	|ddgd t
|| d S )N      ?      @      r   r   rj   r   )r5   r?   r<   rV   r	   r[   r\   r   r   r2   r   )r   rL   rY   Z	b_ub_origrZ   r_   r   r   r   r   test_aliasing_b_ub     
z%LinprogCommonTests.test_aliasing_b_ubc              
   C   sr   t dg}t dgg}t dg}| }dt jf}t|tt|||| j| jd}t	|ddgd t
|| d S )Nr   r   r   r   r!   r   )r5   r?   r<   rV   r	   rY   rZ   r   r   r2   r   )r   rL   r[   Z	b_eq_origr\   r_   r   r   r   r   test_aliasing_b_eq  r   z%LinprogCommonTests.test_aliasing_b_eqc              
   C   sV   dg}dgg}dg}dgg}dg}d}t ||||||| j| jd}t|ddgd d S )Nr   r   g       @)            $@r   r   r   )r	   r   r   r2   r   rL   rY   rZ   r[   r\   r_   r   r   r   r   test_non_ndarray_args  s   z(LinprogCommonTests.test_non_ndarray_argsc                    sr   t ddg}ddgddgddgg}g d}	 	 d fdd	} fd	d
 jD }d|d< tt|||||d d S )Nrf   r   r   r   r   r   c              
      s   t | ||||| j|d d S Nr   r	   r   )rL   rY   rZ   r[   r\   r_   r   r   r   r   f  s   
z2LinprogCommonTests.test_unknown_options.<locals>.fc                       i | ]}| j | qS r   r   .0keyr   r   r   
<dictcomp>      z;LinprogCommonTests.test_unknown_options.<locals>.<dictcomp>*   Zspamr   )NNNNNN)r5   r?   r   r   r
   )r   rL   rY   rZ   r   or   r   r   test_unknown_options  s   
z'LinprogCommonTests.test_unknown_optionsc              	   C   s   t ddgddgddgg}t g d}t ddg }dt jfgt| }dgt| }t jt t||||| j|d}W d    n1 sJw   Y  t j	|j
dd	g t j	|jd
 d S )Nr3   r   r!   r   r      r   r   rL   rY   rZ   r_   r   r   ?ffffff@ffffff)r5   r?   rV   r   testingr   r
   r	   r   r   r.   r-   r   rY   rZ   rL   r_   r   r   r   r   r   test_integrality_without_highs  s   
z1LinprogCommonTests.test_integrality_without_highsc                    sT  d fdd	}t t|g dddgd tj }|td t t|g dg dd W d    n1 s5w   Y  t t|g dd	dgd t t|d
dgd
dggd
dgd t t|d
dgd
ggd
gd t t|d
dgd
dggd
dgd t t|d
dgd
ggd
gd t t|d
dgd
gd
d d jv r jd rd S t t|d
dgtdd
d d S )Nc              
          t | ||||| j jd d S r   r	   r   r   rL   rY   rZ   r[   r\   r_   r   r   r   r        
z1LinprogCommonTests.test_invalid_inputs.<locals>.fr   r   r!   r   r!   r$   r_   zCreating an ndarray from ragged)r   r   )r!   r$      r   r   r   r   )rY   rZ   )r[   r\   _sparse_presolve)r   r   r!   )rY   r\   NNNNN)	r   r   r5   r   r   filterr   r   r:   )r   r   supr   r   r   test_invalid_inputs  s"     
 z&LinprogCommonTests.test_invalid_inputsc              	      s  d fdd	}t jd}d}d}tj||d}||}||}||| }||| }	t |	|f}
|| } jdv rkt	t
d j d	d
 t||||
 j jd W d    d S 1 sdw   Y  d S i  j} jdv ryd|d< t||||
 j|d}|jsJ d S )Nc              
      r   r   r   r   r   r   r   r     r   z5LinprogCommonTests.test_sparse_constraints.<locals>.fr   rn      rU   >   r   r   zMethod 'z.' does not support sparse constraint matrices.r   r   >   r   Trv   r   )r5   r6   r7   ru   rv   rB   rs   Zcolumn_stackr   r   r   r	   r   r   )r   r   rE   rP   rD   r[   Zx_validrL   r^   r]   r_   r\   r   r   r   r   r   test_sparse_constraints  s0   



"


z*LinprogCommonTests.test_sparse_constraintsc                 C   sh   g d}g dg dg dg}g d}t jd d}t|||| jd|id	}t|| t|j| d S )
Nr$   ra   r!   r   r   r   r   r   r!   r3   r   r   r!         @ra   r   r3   r   ra   rR   r$   r   r   r3         iX  r   r!   r   r   )r5   r6   rX   r	   r   r   r   r   )r   rL   rJ   rK   r   r   r   r   r   test_maxiter"  s   
zLinprogCommonTests.test_maxiterc                 C   s  | j dd}tdgd| j| j d}t|dd |r t|jd tg dg d| j| j d}t|d	g d
 |r>t|jd tddgddg| j| j d}t|dddg |r\t|jd tg dg dg dgddgg d| j| j d}t|dg d |rt|jd d S d S )NpresolveTr   )r   r   r_   r   r   r   r   ))r   r   )r3   r3   )r!   r!   r   )r   r3   r!   )r   r!   r   )r   r   r   r   r   r   r   r   r   rh   ))r   )r   rR   )      @r
  r[   r\   r_   r   r      )r   rh   r
  )r   getr	   r   r2   r   r   r   do_presolver   r   r   r   test_bounds_fixed1  s6   z$LinprogCommonTests.test_bounds_fixedc                 C   s   | j dd}tdgd| j| j d}t| |rt|jd tdgdg| j| j d}t| |r6t|jd tg dg d| j| j d}t| |rRt|jd d S d S )	Nr  Tr   r   r  r   r   ))r   r   r   r   )r   r  r	   r   r   r   r   r  r   r   r   test_bounds_infeasibleQ  s    z)LinprogCommonTests.test_bounds_infeasiblec              	      s   j dd}| o jdk}g d}dtjtjfdg}dtj tj fdg}|r fdd}tt# tt	 |||d	 W d    n1 sKw   Y  W d    n1 sZw   Y  tt, tt	 |||d	 W d    n1 s{w   Y  W d    d S W d    d S 1 sw   Y  d S t
|| j j d
}t| |rt|jd t
|| j j d
}t| |rt|jd d S d S )Nr  Tr   r   r   r   c                    s   t | | j jd}|S )Nr  r   )rL   r_   r   r   r   r   gz  s   z6LinprogCommonTests.test_bounds_infeasible_2.<locals>.gr   rL   r_   r   r   r   )r   r  r   r5   rV   r   r   RuntimeWarningr   
IndexErrorr	   r   r   r   )r   r  Zsimplex_without_presolverL   Zbounds_1Zbounds_2r  r   r   r   r   test_bounds_infeasible_2h  sB   
"z+LinprogCommonTests.test_bounds_infeasible_2c                 C   s&   ddg}t || j| jd}t| d S )Nr3   r   r   )r	   r   r   r"   )r   rL   r   r   r   r   test_empty_constraint_1  s   z*LinprogCommonTests.test_empty_constraint_1c                 C   s`   g d}dt jft j dfddg}t||| j| jd}t| | jddr.t|jd d S d S )N)r3   r   r3   r   r   r3   r   r  r  T)	r5   rV   r	   r   r   r"   r  r   r   r   rL   r_   r   r   r   r   test_empty_constraint_2  s   z*LinprogCommonTests.test_empty_constraint_2c                 C   sL   g d}dt jft j dfddg}t||| j| jd}t|g ddd d S )N)r   r3   r   r3   r   r  r  )r   r   r3   r   r   r0   r/   )r5   rV   r	   r   r   r2   r  r   r   r   test_empty_constraint_3  s   z*LinprogCommonTests.test_empty_constraint_3c              
   C   sb   t ddgd }ddgddgddgg}g d}t|||ttt| j| jd}t|ddd	gd
 d S )Nr!   r   r3   r   r   r   r   ird   r   	r5   r?   r	   r[   r\   r_   r   r   r2   r   rL   rY   rZ   r   r   r   r   test_inequality_constraints  s   z.LinprogCommonTests.test_inequality_constraintsc              
   C   sX   ddg}ddgddgddgg}g d}t |||ttt| j| jd}t|d	d
dgd d S )Nrd   r!   r   r3   r   r   )r   r3   r3   r   r   gUUUUUU?gUUUUUU?r   r	   r[   r\   r_   r   r   r2   r  r   r   r   test_inequality_constraints2  s   z/LinprogCommonTests.test_inequality_constraints2c              
   C   sp   ddg}d}t |tttt|| j| jd}t|ddgd ddg}t |tttt|| j| jd}t|ddgd d S )Nr   r   r   r   r0   )r	   rY   rZ   r[   r\   r   r   r2   r  r   r   r   test_bounds_simple  s   z%LinprogCommonTests.test_bounds_simplec              
   C   Z   t dg}t dgg}t dg}d}t|tt|||| j| jd}t|ddgd d S )Nr   r   )r   Nr   r!   r   r5   r?   r	   rY   rZ   r   r   r2   r   rL   r[   r\   r_   r   r   r   r   test_bounded_below_only_1     z,LinprogCommonTests.test_bounded_below_only_1c              
   C   s`   t d}t d}t g d}dt jf}t|tt|||| j| j	d}t
||t |d d S )Nr!   r   rU   r   r  r5   rt   eyer?   rV   r	   rY   rZ   r   r   r2   sumr&  r   r   r   test_bounded_below_only_2  s   


z,LinprogCommonTests.test_bounded_below_only_2c              
   C   r$  )Nr   r   )Nr   r   r!   r   r%  r&  r   r   r   test_bounded_above_only_1  r(  z,LinprogCommonTests.test_bounded_above_only_1c              
   C   sb   t d}t d}t g d}t j df}t|tt|||| j| j	d}t
||t |d d S )Nr!   r   r$   r   r  r)  r&  r   r   r   test_bounded_above_only_2  s   

z,LinprogCommonTests.test_bounded_above_only_2c              
   C   sd   t d}t d}t g d}t j t jf}t|tt|||| j| j	d}t
||t |d d S )Nr!   r   r   r  r)  r&  r   r   r   test_bounds_infinity  s   

z'LinprogCommonTests.test_bounds_infinityc              
   C   s   t ddgd }t jddgddggt jd}ddg}t j t jf}dt jf}||f}t|||tt|| j| jd}t	|d	d
dgd d S )Nr3   r$   rf   r   r   r4   rd   r   gm۶m&g$I$Ig%I$I@r   )
r5   r?   float64rV   r	   r[   r\   r   r   r2   )r   rL   rY   rZ   r   r   r_   r   r   r   r   test_bounds_mixed  s   

z$LinprogCommonTests.test_bounds_mixedc              
   C   sV   ddg}ddgddgddgg}g d}dd	g}t |||tt|| j| jd
}t| d S Nrj   r   rh   r   r   r   )   r   r!   r   r   rq   r   )r	   r[   r\   r   r   r   r   rL   rY   rZ   r_   r   r   r   r    test_bounds_equal_but_infeasible     z3LinprogCommonTests.test_bounds_equal_but_infeasiblec              
   C   sV   ddg}ddgddgddgg}g d}dd	g}t |tt|||| j| jd
}t| d S r2  )r	   rY   rZ   r   r   r   r&  r   r   r   !test_bounds_equal_but_infeasible2  r7  z4LinprogCommonTests.test_bounds_equal_but_infeasible2c              
      sj   ddg}ddgddgg}ddg}ddg} fd	d
 j D }d|d< t|||tt| j|d}t| d S )Nr   r   g?g@r$   ra   r   r4  c                    r   r   r   r   r   r   r   r   $  r   zDLinprogCommonTests.test_bounds_equal_no_presolve.<locals>.<dictcomp>Fr  r   )r   r	   r[   r\   r   r   r   rL   rY   rZ   r_   r   r   r   r   r   test_bounds_equal_no_presolve  s   z0LinprogCommonTests.test_bounds_equal_no_presolvec              
   C   s   d\}}t jd}||}d|d< |||}d|d d df< ||}g dg}d}g d}	t||||||	| j| jd}
t|
dd	 d S )
Nr   r   r   )r   r   r   r   r!   )rg   rR   r;  )rg   Nrp   r   gyMj#r   )r5   r6   r7   rB   r	   r   r   r2   )r   rP   rD   rE   rL   r[   r\   rY   rZ   r_   r   r   r   r   test_zero_column_1+  s   


z%LinprogCommonTests.test_zero_column_1c              
   C   s   | j dv r	t  tjd}d\}}||}d|d< |||}d|d d df< ||}|||}d|d d df< ||}d}	t||||||	| j | jd}
t	|
 | j
dd	rjd
| j vrlt|
jd d S d S d S )N>   	highs-ipmr   r   r#   r3   r   rp   r   r  Tr   )r   r   xfailr5   r6   r7   rB   r	   r   r"   r  r   r   )r   rE   rP   rD   rL   r[   r\   rY   rZ   r_   r   r   r   r   test_zero_column_2:  s(   



z%LinprogCommonTests.test_zero_column_2c              
   C   sR   g d}g dg dg dg}g d}t |tt||t| j| jd}t|dd d S )Nr   r   r   r   r   r   r   r   r!   r   r   r!   r   r	   rY   rZ   r_   r   r   r2   r   rL   r[   r\   r   r   r   r   test_zero_row_1U  s   z"LinprogCommonTests.test_zero_row_1c              
   C   sR   g dg dg dg}g d}g d}t |||ttt| j| jd}t|dd d S )Nr@  rA  rB  r   r   r   r   r   )r   rY   rZ   rL   r   r   r   r   test_zero_row_2]  s   z"LinprogCommonTests.test_zero_row_2c              
   C   s   d\}}t jd}||}|||}d|dd d f< ||}t|tt||t| j| j	d}t
| | j	ddrCt|jd d S d S Nr#   i  r   r   r  T)r5   r6   r7   rB   r	   rY   rZ   r_   r   r   r   r  r   r   )r   rP   rD   rE   rL   r[   r\   r   r   r   r   test_zero_row_3e  s   

z"LinprogCommonTests.test_zero_row_3c              
   C   s   d\}}t jd}||}|||}d|dd d f< || }t|||ttt| j| j	d}t
| | j	ddrDt|jd d S d S rG  )r5   r6   r7   rB   r	   r[   r\   r_   r   r   r   r  r   r   )r   rP   rD   rE   rL   rY   rZ   r   r   r   r   test_zero_row_4t  s   
z"LinprogCommonTests.test_zero_row_4c              
   C   sr   g d}g dg dg dg dg}g d}t |tt||t| j| jd}t| | jddr7t|j	d	 d S 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  Tr   )
r	   rY   rZ   r_   r   r   r   r  r   r   rD  r   r   r   test_singleton_row_eq_1  s   z*LinprogCommonTests.test_singleton_row_eq_1c              
   C   sX   g d}g dg dg dg dg}g d}t |tt||t| j| jd}t|dd d S )	NrJ  rK  rL  rM  )r   r   r   r$   r   r$   r   rC  rD  r   r   r   test_singleton_row_eq_2  s   z*LinprogCommonTests.test_singleton_row_eq_2c              
   C   sz   g d}g dg dg dg dg}g d}g d}t |||tt|| j| jd}t| | jd	d
r;t|jd d S d S )NrJ  rK  rL  r3   r   r   r   rM  )r   r   r   r$   rp   rq   rq   rq   r   r  Tr   	r	   r[   r\   r   r   r   r  r   r   r5  r   r   r   test_singleton_row_ub_1  s   z*LinprogCommonTests.test_singleton_row_ub_1c              
   C   s`   g d}g dg dg dg dg}g d}g d}t |||tt|| j| jd}t|d	d
 d S )NrJ  rK  rL  rP  rM  )r   r         r$   rQ  r   rU   r   )r	   r[   r\   r   r   r2   r5  r   r   r   test_singleton_row_ub_2  s   z*LinprogCommonTests.test_singleton_row_ub_2c              
   C   sN   ddg}ddgddgddgg}g d}t |||ttt| j| jd}t| d S )Nr3   r   r   )r   r   r	  r   )r	   r[   r\   r_   r   r   r   r  r   r   r   test_infeasible  s   z"LinprogCommonTests.test_infeasiblec              
   C   s\   dg}dgg}d}d}t |||tt|| j| jd}t| | jddr,t|jd d S d S )	Nr   r   r$   )r   rd   r   r  Tr   rR  r5  r   r   r   !test_infeasible_inequality_bounds  s   z4LinprogCommonTests.test_infeasible_inequality_boundsc              
   C   sR   t ddgd }ddgddgg}ddg}t|||ttt| j| jd}t| d S )Nr   r3   r   r   )	r5   r?   r	   r[   r\   r_   r   r   r"   r  r   r   r   test_unbounded  s   z!LinprogCommonTests.test_unboundedc                    s`   dg}dg} fdd j D }d|d< t|| j|d} jdkr*t|jd	 d S t| d S )
Nr   Nr   c                    r   r   r   r   r   r   r   r     r   zQLinprogCommonTests.test_unbounded_below_no_presolve_corrected.<locals>.<dictcomp>Fr  r  r   r   )r   r	   r   r   r   r"   )r   rL   r_   r   r   r   r   r   *test_unbounded_below_no_presolve_corrected  s   
z=LinprogCommonTests.test_unbounded_below_no_presolve_correctedc              
   C   s   t g d}t g dg dg dg}t g d}g d}t|||tt|| j| jd}t| | j 	dsOt
|jd	 t j t
|jd
d d d
S d
S )|
        Test whether presolve pathway for detecting unboundedness after
        constraint elimination is working.
        )r   r   r   r   r3   r3   r   r   r   r   r   r   r   r   r   r   r   r   )r   r   r   r   r   r3   r   r   r   )rp   rp   rp   r  r  rq   r   r   r3   N$   $The problem is (trivially) unboundedr5   r?   r	   r[   r\   r   r   r"   lowerr   r   r.   rV   r+   r5  r   r   r   *test_unbounded_no_nontrivial_constraints_1  s"   
z=LinprogCommonTests.test_unbounded_no_nontrivial_constraints_1c              
   C   s   t g d}t g dg dg dg}t g d}g d}t|||tt|| j| jd}t| | j 	dsPt
|jd	 t j  t
|jd
d d d
S d
S )r[  )r   r   r   r   r3   r   r\  r]  )r   r   r   r   r   r   r^  )rp   rp   rp   r  r  )Nr   r   r   r3   Nr_  r`  ra  r5  r   r   r   *test_unbounded_no_nontrivial_constraints_2  s"   
z=LinprogCommonTests.test_unbounded_no_nontrivial_constraints_2c              
   C   sd   t g dd }g dg dg dg}g d}t|||ttt| j| jd}t|g dd	d
d d S )N)rn   rR   r   r3   r  )rS   r   r   )   rS   r   )r   rn   '  r   )r   r   rf  gh㈵>Hz>)r0   r*   r)   r  r  r   r   r   test_cyclic_recovery  s   z'LinprogCommonTests.test_cyclic_recoveryc              
   C   s   t g d}t g dg dg dg}g d}d}dd | j D }||d	< t|||ttt| j|d
}| jdkrJ| j	dsJt
||d	  d S t|g dd d S )N)rg   9   re   g      8@)rU   g      r   re   )rU   g      rT  r   rK  )r   r   r   rn   c                 S   s   i | ]\}}||qS r   r   )r   r   valr   r   r   r     s    z8LinprogCommonTests.test_cyclic_bland.<locals>.<dictcomp>r   r   r   bland)r   r   r   r   r"  )r5   r?   r   itemsr	   r[   r\   r_   r   r  r   r2   )r   rL   rY   rZ   r   r   r   r   r   r   test_cyclic_bland  s   
z$LinprogCommonTests.test_cyclic_blandc           	      C   s   d\}}t jd}||}|||}||}d|dd d f  |dd d f< |d  d9  < t }|td t|tt	||t
| j| jd}W d    n1 sTw   Y  t| d S )N)rR   rR   r   r   r   r3   A_eq does not appear...r   )r5   r6   r7   rB   r   r   r
   r	   rY   rZ   r_   r   r   r   )	r   rP   rD   rE   rL   r[   r\   r   r   r   r   r   $test_remove_redundancy_infeasibility,  s   

 z7LinprogCommonTests.test_remove_redundancy_infeasibilityc           	   
   C   sB   t  \}}}}}}}t|||||t| j| jd}t|||d d S )Nr   r   )rm   r	   r_   r   r   r2   	r   rL   rY   rZ   r[   r\   rk   rl   r   r   r   r   test_nontrivial_problem@  s
   z*LinprogCommonTests.test_nontrivial_problemc                 C   s   t dd\}}}t '}|td |td |t t|||ttt	| j
| jd}W d    n1 s5w   Y  t|dd d S )NrS   $Solving system with option 'sym_pos'invalid value encounteredr   r   r   )rQ   r   r   r
   r  r   r	   r[   r\   r_   r   r   r2   )r   rY   rZ   rL   r   r   r   r   r   test_lpgen_problemH  s   
z%LinprogCommonTests.test_lpgen_problemc                 C   s2  g d}d\}}|||d|dddd|ddg|dd|d|ddddddgdd||ddddddddgdddddd||dd|dgdddd|||d|dddgddddddd||dd|gddddddddd|||gg}g d}t  }|t t|tt||t| j| jd}W d    n1 sw   Y  t	|dddd	 d S )
N)r   r$   re      r$   r!   ra   rh   r   r        r  r   )r      i!   r   r   ir   i  ư>rg  )r/   r*   r)   )
r   r   r   r	   rY   rZ   r_   r   r   r2   )r   rL   rD   pr[   r\   r   r   r   r   r   test_network_flowU  s$   
z$LinprogCommonTests.test_network_flowc           	      C   s   g d}ddgddgddgddgddgg}d\}}||dddg|d||dgd||d|gddd||gg}g d}t  4}trB|t |td	 |td
 |td |t t|tt	|||| j
| jd}W d    n1 sqw   Y  t|dd d S )N)r   r   r   r!   r   r   r$   r   r!   r   r  )rj   r   r   r$   scipy.linalg.solve
Ill...rn  Solving system with option...r   r3  r   )r   has_umfpackr   r   r  r
   r   r	   rY   rZ   r   r   r2   )	r   rL   r_   rD   r{  r[   r\   r   r   r   r   r   "test_network_flow_limited_capacityj  s4   


z5LinprogCommonTests.test_network_flow_limited_capacityc              
   C   sL   g d}g dg dg}ddg}t |||ttt| j| jd}t|dd d S )	N)r   rf   rj   )r!   r   r   )r   r   r!   rR   r  r   ir   r   r  r   r   r   (test_simplex_algorithm_wikipedia_example  s   z;LinprogCommonTests.test_simplex_algorithm_wikipedia_examplec              
   C   s\   g d}g dg dg dg}g d}t |tt||t| j| jd}t|dg dd	d
d d S )Nr   r   r   r   r  r   g     s@)g     P@r   g     1@r   g     f@r   gTqs*>rg  )r/   r0   r*   r)   rC  rD  r   r   r   test_enzo_example  s   
z$LinprogCommonTests.test_enzo_examplec                 C   s   g d}g dg dg dg dg dg}g d}t  }|td t|tt||t| j| jd	}W d    n1 s;w   Y  t	|d
g dd d S )N)r   g333333@g%@r   g333333g%)r3   r3   r3   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   )rT  皙?333333?r  r  rn  r   gRQ)r  g?        r  r   r  r   )
r   r   r
   r	   rY   rZ   r_   r   r   r2   )r   rL   r[   r\   r   r   r   r   r   test_enzo_example_b  s"   
z&LinprogCommonTests.test_enzo_example_bc              
   C   s   d}t | }dt j t d|d  |d  }t t |d t |f}ddg}t|tt	||t
| j| jd}t|dt |d d S )NrS   r   r   r   r   r   )r5   rt   pir8   r@   cossinr	   rY   rZ   r_   r   r   r2   r:   )r   rP   rL   tmpr[   r\   r   r   r   r   #test_enzo_example_c_with_degeneracy  s   "z6LinprogCommonTests.test_enzo_example_c_with_degeneracyc           	   
   C   s   d}t | }dt j t | |d  }t |d }d|d< t |}d|d< t ||f}ddg}t|tt	||t
| j| jd}t| d S )NrT   r   r   r  r   r   )r5   rt   r  r8   r  r  r@   r	   rY   rZ   r_   r   r   r"   )	r   rP   rL   r  Zrow0Zrow1r[   r\   r   r   r   r   &test_enzo_example_c_with_unboundedness  s   
z9LinprogCommonTests.test_enzo_example_c_with_unboundednessc              
      s   d}t | }dt j t | |d  }t t |d t |f}ddg} fdd jD }d|d< t|t	t
||t j|d}t| d S )	NrT   r   r   c                    r   r   r   r   r   r   r   r     r   zMLinprogCommonTests.test_enzo_example_c_with_infeasibility.<locals>.<dictcomp>Fr  r   )r5   rt   r  r8   r@   r  r  r   r	   rY   rZ   r_   r   r   )r   rP   rL   r  r[   r\   r   r   r   r   r   &test_enzo_example_c_with_infeasibility  s   z9LinprogCommonTests.test_enzo_example_c_with_infeasibilityc              
   C   s   t g d}t g dg dg dg dg dg}t g d}t g dg d	g}t d
d
g}t|||||t| j| jd}t|d
t |dd d S )N)ggQ뱿Mbp?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   g      4@)r   r   r3   r   r3   r   )r   r   r3   r   r3   r   r   r   g>)r/   r0   r*   )r5   r?   r	   r_   r   r   r2   Z
zeros_like)r   rL   rY   rZ   r[   r\   r   r   r   r   test_basic_artificial_vars  s   
z-LinprogCommonTests.test_basic_artificial_varsc           	   
   C   s   t d\}}}}}}t||||||| j| jd}t|j t|j t|j  d| jvr4t|jdk t	||j
 |j t	|||j
  |jdd t	|||j
  |jdd dD ] }|| v ryt|| d tjsmJ t|| d	 tjsyJ qYd S )
Nr   rY   rZ   r[   r\   r_   r   r   r   z%Optimization terminated successfully.gdy=r*   )eqlinineqlinrb  upper	marginalsresidual)r`   r	   r   r   r   r   r   r   r+   r   r.   r-   r   r   keysr   r5   Zndarray)	r   rL   rY   rZ   r[   r\   r_   r   r   r   r   r   test_optimize_result  s$   



z'LinprogCommonTests.test_optimize_resultc           
   /   C   sb  g d}d}d}d}t g dddd|dd	dd
ddd
dd
ddddddgddd|ddd	d
d
dd
d
dddddddgg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dddd
d
dddd
||ddd|dddddgdddddd
d
dddd
||d|ddddgdd
dd ddd|d
d
ddddd|dddgddddd
dd
ddd|d
|ddd|ddgdd
d
d ddd||d
ddddddd|dgddddd
d
dddd||d
ddddd|gg}t g d!}t g d"}t '}|td# |td$ |t t|||tt	|| j
| jd%}	W d    n	1 s$w   Y  t|	d&d' d S )(N)rq   r   rn   r  r  r  r  r  r     r  r  r  r  r  rq   rq   rq   rq   rq   rq   gqq?     g)r   gQr   r   rf   r   r   r   r3   r3   r   r3   r3   r   r   r   r   r   r   r   r   g333333rf   r3   )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   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   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   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   r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   r   r   r   r   r   r3   r   r   r   r   r   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r3   r   r   r   r   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r3   r   r   r   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r3   r   r   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r3   r   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r3   gףp=
?r   gffffff?g @)r  r   r   rn   rn   rn   rn   rn   rn   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   rr  rs  r   g^Zr   )r5   r?   r   r   r
   r  r   r	   r[   r\   r   r   r2   )
r   r_   r   r  hrY   rZ   rL   r   r   r   r   r   test_bug_5400  s\   ((((((((
z LinprogCommonTests.test_bug_5400c              
   C   s   t g d}t g dg dg}t ddg}t g dg }t dg }d}t||||||| j| jd	}t|d
t g dd d S )NrA  )r   r  r  )     @r  r  g      @r  )r  g    .Ag    .Ag    cArp   r   gfffff-@)r   g@r   r   r5   r?   r	   r   r   r2   )r   rL   r[   r\   rY   rZ   r_   r   r   r   r   test_bug_6139C  s   	
z LinprogCommonTests.test_bug_6139c                 C   sV  t g dg}t dg}t g dg dg dg dg dg}t g d}t g d	g d
gj}t g d}t 4}trG|t |td |td |td |t	 t
||||||| j| jd}W d    n1 svw   Y  d}	t g d}
t||	|
d d}t|d d df | |j t|j|d d df |  d S )N)
r   r   r   g(\?r   ?r   r   g(\?r   g%䃞?)
r   r   r   zG?r   r   r   gɿr   g)\(̿)
r   r   r   r   r   r   r   r   r   r   )
r   r   r   gQ?r   r   r   r   r   r   )
r   gQ      пr   r   r   g{Gz r   r   gQ?)
r   r   r   r   r   r   r   r  r   r   )gGz?r   gjt?g+gI+)
gzG
ףp=
(\?r  gQտgGzGz?
ףp=
?g333333\(\)
Gz?g{Gz?gzG@gQ?r  rU   g)\(?g(\?{Gz?g333333ÿ)
g=
ףp=gffffff?r   g(\g(\gp=
ף?g
ףp=
@g{Gz?gQ?gQ?z%Solving system with option 'cholesky'rr  rs  r   gV)
r  r  r  r  r  rU   r  r  r  r  r   rz  r   r   )r5   r?   rW   r   r  r   r   r
   r  r   r	   r   r   r2   r   r.   )r   r[   r\   rY   rZ   r_   rL   r   r   r/   r0   r*   r   r   r   test_bug_6690Y  sH   

 z LinprogCommonTests.test_bug_6690c                 C   s   t d\}}}}}t '}|td |td |t t|tt||t	| j
| jd}W d    n1 s6w   Y  d}t||d t||j| tt|jjd |j d S )Nr!   rn  rs  r   -U?r   ro   )rM   r   r   r
   r  r   r	   rY   rZ   r_   r   r   r2   r   dotr.   r   r5   r:   rO   )r   r[   r\   rL   _r   r   r/   r   r   r   test_bug_7044  s   
z LinprogCommonTests.test_bug_7044c                 C   s   t g d}t g dg dg dg dg dg dg dg d	g d
g dg dg dg dg dg dg dg dg dg dg dg dg dg}t g d}t g dg}t dgg}dgd }t||||||| j| jd}t|ddd  d S )!N)	r3   r   r   r   r   r   r   r   r   )	r   g     g     x@g     8g     Xg      g     }@g      Tg     Pr)	r   g     @g      E@g     H@g      m@g     @g     x@g     hg     L@)	r   g      jg     g     ؀@g     t@g     @g     g     p@g     K@)	r   g     Ȋ@g     ؊g     T@g     |@g     pg     Xg     Љ@g      ~@)	r   g     s@     zg     n@r  g     @h@g     {g      Zg     z)	r   g     @g     8@g     v@g     b@g     ؉g     `@g     ȃ@g     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  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  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   r   )	r  r   r   r   r   r   r   r   r   r   rp   re   r   ghb$[@rz  r/   r*   r  r   r   r   r   test_bug_7237  sB   
z LinprogCommonTests.test_bug_7237c                 C   s   t g dg dg dg dg}t |jd }t |jd  }dg|jd  }t !}|td |t t	|||t
t|| j| jd	}W d    n1 sQw   Y  | jd
ddk rj| jdkrjt| d S t|ddd d S )N)X  i  iD4       3i)ii    @i  )r  r   r   r  r   )r  r   r   r  r   r   r   r   rs  r   tolg&.>绽|=r   g rz  r  )r5   r?   r:   rC   rt   r   r   r  r   r	   r[   r\   r   r   r  r%   r2   )r   rY   rZ   rL   r_   r   r   r   r   r   test_bug_8174  s&   
z LinprogCommonTests.test_bug_8174c                 C   s   t g d}t d }t dgdgdgdgdgdgdgg}t g dg dg dg dg dg}t d	gd
gd
gd
gd
gg}t #}trN|t |td t|||||t	| j
| jd}W d    n1 slw   Y  t|dd d S )N)r   r   r   r   r   r   r   rh   r   )r   r   r   r   r   r   r   )r  g?g?r   r   r   r3   )r  r   r   r   r   r   gUUUUUU)r   r  r   r   r   r   )r   r   r  r   r   r   r  rn   r   rn  r   g?E@r   )r5   r?   identityr   r  r   r   r
   r	   r_   r   r   r2   r   rL   rY   rZ   r[   r\   r   r   r   r   r   test_bug_8174_2  s(   &
z"LinprogCommonTests.test_bug_8174_2c                 C   sn   t g d}t g dg dg dg dg dg}t g d}t|||| j| jd}t|g d	d
 d S )N)rh   r   rj         ?r  )r$         @r  r         )r   r   r   r   rU   )r!   rT  r$   g      )rc   )r3   g      @r   r  r   )r  r   g      r3   g      #@)r   r   r   r   r   )rY   rZ   r   r   )r   r   rx  gUUUUUU@gUUUUUU#@r"  )r5   r?   r	   r   r   r2   r  r   r   r   test_bug_8561  s   z LinprogCommonTests.test_bug_8561c           
      C   sH  g d}g dg dg dg dg}g d}g d}d}t  (}tr'|t |td	 |t t|||tt|| j	| j
d
}W d    n1 sJw   Y  |g d |d d|d< t  (}trj|t |td	 |t t|||tt|| j	| j
d
}W d    n1 sw   Y  d}	t|||	d t|||	d d S )N)rg   rR   rd   r!   )ra   ri   rj   rd   )ri   ra   r$   rb   )rj   r$   ra   rj   )r!   rf   rf   rg   )re   r  rj   )rq   rq   rq   rq   g      B@rs  r   )r   r   r3   r   r   rp   r   ro   )r/   r)   )r   r  r   r   r  r   r	   r[   r\   r   r   r=   r2   )
r   rL   rY   rZ   r_   r/   r   res1res2r)   r   r   r   test_bug_8662   sB   

	


z LinprogCommonTests.test_bug_8662c              
   C   sR   ddg}ddgg}dg}ddg}t |tt|||| j| jd}t|dd	gd
d d S )Nr   r   r   rc   rb   rq   rp   r   g۶m۶m?gI$I$@r  )r	   rY   rZ   r   r   r2   r&  r   r   r   test_bug_8663$  s   
z LinprogCommonTests.test_bug_8663c                 C   s   dg}dgdgg}ddg}dgdgdgg}g d}t  !}|t |td t|||||t| j| jd	}W d    n1 s@w   Y  t| d S )
Nr$   r   r   r   ri   re   r!   r   rR   r~  r   )	r   r   r  r
   r	   r_   r   r   r   r  r   r   r   test_bug_8664/  s   
z LinprogCommonTests.test_bug_8664c              
   C   sx   t g d}t g dg dg}t ddg}g d}t|||tt|| j| jd}t|dd t||j	 |j
 d	S )
zv
        Test whether bug described at:
        https://github.com/scipy/scipy/issues/8973
        was fixed.
        )r   r   r   r   r3   )r   r   r   r   r   )r   r   r   r   r   r   r   )rp   rp   rp   r  r  r   r   N)r5   r?   r	   r[   r\   r   r   r2   r   r.   r-   r5  r   r   r   test_bug_8973?  s   z LinprogCommonTests.test_bug_8973c              
   C   sX   t d}t dgg}t dg}d}t|||tt|| j| jd}t|dgdd dS )z
        Additional test for:
        https://github.com/scipy/scipy/issues/8973
        suggested in
        https://github.com/scipy/scipy/pull/8985
        review by @antonior92
        r   r   rp   r   r   r  N)	r5   r:   r?   r	   r[   r\   r   r   r2   r5  r   r   r   test_bug_8973_2P  s   
z"LinprogCommonTests.test_bug_8973_2c              
   C   s   t d}t dgg}t dg}d}ddg}ddgddgg}ddg}dd	g}d
di}|| j t|||tt|| j|d}t	|ddgdd dS )zg
        Test for linprog docstring problem
        'disp'=True caused revised simplex failure
        r   r   rp   r3   r$   rf   r   rd   r   r   Tr   rR   r   r  N)
r5   r:   r?   updater   r	   r[   r\   r   r2   r9  r   r   r   test_bug_10124`  s   
z!LinprogCommonTests.test_bug_10124c                 C   s   t g dg dg dg dg dg dg}t g d}t jdd	t d
fdd}t }|td t|tt	||t
| j| jd}W d   n1 sMw   Y  t|g ddd dS )zq
        Test for redundancy removal tolerance issue
        https://github.com/scipy/scipy/issues/10349
        )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   )      rR         f   r   r   r$   NZaxisrn  r   )   \   r   r  r   rR   r  r  )r5   r?   Zconcatenater:   r   r   r
   r	   rY   rZ   r_   r   r   r2   )r   r[   r\   rL   r   r   r   r   r   test_bug_10349s  s    
z!LinprogCommonTests.test_bug_10349darwinz0Failing on some local macOS builds, see gh-13846reasonc                 C   s  g d}g dg dg dg dg dg dg dg d	g d
g dg
}g d}i }| j ds4ddi}|| j t @}|td trK|t |t	d |t	d |t	d |t	d |t
d t|tt||t| j |d}W d   n1 sw   Y  t|jd dS )zA
        Test that autoscale fixes poorly-scaled problem
        )       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  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  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   )
g     Ag      Ag     @Ag      A   Ar  r  r  r  r  r   	autoscaleTr~  r}  zdivide by zero encountered...zoverflow encountered...zinvalid value encountered...Ill-conditioned matrix...r   Nl )r   r   r  r   r   r   r
   r  r   r  r   r	   rY   rZ   r_   r   r-   )r   rL   r[   r\   r   r   r   r   r   r   test_bug_10466  s>   
z!LinprogCommonTests.test_bug_10466N)W__name__
__module____qualname____doc__r   r   r   r   r   r   r   r   markthread_unsafer   r   r   r   r  r  r  r  r  r  r  r  r!  r#  r'  r,  r-  r.  r/  r1  r6  r8  r:  r<  r?  rE  rF  rH  rI  rN  rO  rS  rU  rV  rW  rX  rZ  rc  rd  rh  rm  ro  rq  rt  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  skipifsysplatformr  r   r   r   r   r   n  s    


! 
)
								
7,)$r   zignore::DeprecationWarningc                   @      e Zd ZdZdS )LinprogSimplexTestsr   Nr  r  r  r   r   r   r   r   r    s    r  c                   @      e Zd ZdZdd ZdS )LinprogIPTestsr   c                 C      t d d S )Nz*Test is failing, but solver is deprecated.r   skipr   r   r   r   r       zLinprogIPTests.test_bug_10466N)r  r  r  r   r  r   r   r   r   r    s    r  c                   @   s(   e Zd ZdZdd Zdd Zdd ZdS )	LinprogRSTestsr   c                 C   r  Nz Intermittent failure acceptable.r  r   r   r   r   r    r  zLinprogRSTests.test_bug_5400c                 C   r  r  r  r   r   r   r   r    r  zLinprogRSTests.test_bug_8662c                 C   r  r  r  r   r   r   r   r|    r  z LinprogRSTests.test_network_flowN)r  r  r  r   r  r  r|  r   r   r   r   r    s
    r  c                   @   s   e Zd Zdd Zejjejdddiddiddiddid	did
diddiddigdd Zdd Z	ej
ddd Zdd Zdd Zejjdddd ZdS )LinprogHiGHSTestsc              	   C   st   dd }t ddg}ddgddgddgg}g d}ttt||||| jd	 t|||| jd
}t|dddgd d S )Nc                 S   s   d S r   r   r   r   r   r   r     s   z+LinprogHiGHSTests.test_callback.<locals>.cbrf   r   r   r   r   r   r   r   r   rd   r   )r5   r?   r   NotImplementedErrorr	   r   r2   r   r   r   r   r     s   zLinprogHiGHSTests.test_callbackr   r   r3   r   r  r   Zdual_feasibility_toleranceZprimal_feasibility_toleranceZipm_optimality_toleranceZ!simplex_dual_edge_weight_strategyZekkic                    s*    fdd}|  j tt||d d S )Nc                    s   t d j| d d S )Nr   r   r   r   r   r   r   r     s   z7LinprogHiGHSTests.test_invalid_option_values.<locals>.fr   )r  r   r   r
   )r   r   r   r   r   r   test_invalid_option_values  s   z,LinprogHiGHSTests.test_invalid_option_valuesc                 C   sH   t d\}}}}}d}t||||| j| jd}t|jdk| jdk d S )Nr$   r   r  r   r=  )rM   r	   r   r   r   Zcrossover_nit)r   r[   r\   rL   r  r_   r   r   r   r   test_crossover   s   
z LinprogHiGHSTests.test_crossoverrR   c           
   
      s  t dd\ t jjd}j\ fdd}t|d|jd}t|jj	|  fdd	}t|d|jd}t|j
j	|  fd
d}tjdd t|d|jd}d|t < W d    n1 sw   Y  t|jj	|  fdd}tjdd t|d|jd}	d|	t < W d    n1 sw   Y  t|jj	|	 d S )Nr   rX   r  c              	      s   t |  jdjS Nr   r	   r   r-   r.   )r[   rY   r\   r_   rL   r   r   r   f_bub  
   z/LinprogHiGHSTests.test_marginals.<locals>.f_bubz3-point)r   Zf0c              	      s   t  | jdjS r  r  r  )r[   rY   rZ   r_   rL   r   r   r   f_beq  r  z/LinprogHiGHSTests.test_marginals.<locals>.f_beqc              	      s*   t | gj}t |jdjS r  r5   r?   rW   r	   r   r-   r.   r_   )r[   rY   r\   rZ   rL   r   r^   r   r   f_lb"     z.LinprogHiGHSTests.test_marginals.<locals>.f_lbignore)invalidc              	      s*   t | gj}t |jdjS r  r  r   )r[   rY   r\   rZ   rL   r]   r   r   r   f_ub/  r  z.LinprogHiGHSTests.test_marginals.<locals>.f_ub)r`   r	   r   r   rW   r   r-   r   r  r  r  r5   Zerrstateisfiniterb  r  )
r   r   r  Zdfdbubr  Zdfdbeqr  Zdfdlbr  Zdfdubr   )	r[   rY   r\   rZ   r_   rL   r]   r   r^   r   test_marginals  s.   

z LinprogHiGHSTests.test_marginalsc           	   
   C   sv   t dd\}}}}}}t||||||| j| jd}| |j|jj  |j|jj  |jj |j	j }t
|ddd d S )Nr   r  r  r   -q=r  )r`   r	   r   r   rW   r  r  r  r  rb  r   )	r   rL   rY   rZ   r[   r\   r_   r   Zresidr   r   r   test_dual_feasibility:  s   
z'LinprogHiGHSTests.test_dual_feasibilityc              
   C   sX   t dd\}}}}}}t||||||| j| jd}t|jj|||j   ds*J d S )Nr   r  r  r   )	r`   r	   r   r   r5   allcloser  r  r.   r   r   r   r   test_complementary_slacknessH  s
   
&z.LinprogHiGHSTests.test_complementary_slacknessz&Upstream / Wrapper issue, see gh-20589r  c                 C   s   g d}g d}g d}g d}g d}g d}t j|t jd}t j|t jd}t j|t jd}	tjj|||	ffdd	}
t j|t jd}t j|t jd}tjj|d
d
|
||d}|j	dks_J |j
}t j|
| | }t j|jj}t|| d
S )zH
        Test that `linprog` now solves a poorly-scaled problem
        )!)     @g    j/aA)r  Nr  N)r  gS8(U@r  r  r  r  r  r  r  r  r  rp   rp   rp   rp   rp   rp   rp   rp   rp   rp   rp   rp   rp   rp   rp   rp   rp   rp   rp   rp   )Sr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   MbP?MbP<:=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  r   r   )Sr   r   r   r   r   r   r!   r!   r$   r$   r   r   rd   rd   rh   rh   ra   ra   re   re   rR   ru  ru  r   r   r   r      r  r3  r3  r3  r3  r  r  rv  rv  rv  rv     r  rw  rw  rw  rw  rx  rx  rS   rS   rS   rS      r     r  r  r     r     r  r  r     r     r  r  r     r     r  r  r     r     r  r  r     r  )Sr  r3  r  rv  rx  rw  r  rS   r  r  r  r  r  r  r  r  r  r  r  r   r       r!   r3  r  r$   r   r$   r   r  r  r   r   r   r   rv  r  r   r$   r   r$   rw  r  rd   r   rd   r   rS   rx  rh   rd   rh   rd   r  r  ra   rh   ra   rh   r  r  re   ra   re   ra   r  r  rR   re   rR   re   r  r  ru  rR   ru  rR   r  r  r   ru  r   ru  ) r  r  r  r  r  r  r  r  r  r  g    *aAr  gelaz?g?2gZPd%q?g4g̚y?gC1|g֛x?gKcgE3-w?g-.f?`gsF]Fv?g@ʥugbu?gNgжlt?g=ЂgnTs?gQ3og,'* +=r?g)[n)!r  r  r  gl'V3r  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  r4   )r  ry  )rC   N)rY   rZ   r[   r\   r_   r   )r5   asarrayr0  Zint32ru   rv   Z	csr_arrayoptimizer	   r   r.   ZlinalgZnormr  r  r   )r   Z
boundariesZ
eq_entriesZ
eq_indizesZeq_varsZ	eq_valuesZcoefficientsZnp_eq_entriesZnp_eq_indizesZ
np_eq_varsZa_eqr\   rL   resultr.   Zn_r_xZn_rr   r   r   test_bug_20336S  s,   
z LinprogHiGHSTests.test_bug_20336N)r  r  r  r   r   r  r  Zparametrizer  r  	fail_slowr  r	  r  r>  r"  r   r   r   r   r    s,    


1r  c                       s<   e Zd Zdd Zdd Zdd Zejj fddZ	  Z
S )	TestLinprogSimplexDefaultc                 C   s
   i | _ d S r   r   r   r   r   r   setup_method  s   
z&TestLinprogSimplexDefault.setup_methodc                 C   r  NzSimplex fails on this problem.r  r   r   r   r   r    r  z'TestLinprogSimplexDefault.test_bug_5400c                 C   r  r&  r  r   r   r   r   test_bug_7237_low_tol     z/TestLinprogSimplexDefault.test_bug_7237_low_tolc                    J   | j ddi tt t   W d    d S 1 sw   Y  d S Nr  r  r   r  r   r   r
   superr  r   	__class__r   r   test_bug_8174_low_tol     "z/TestLinprogSimplexDefault.test_bug_8174_low_tol)r  r  r  r%  r  r'  r   r  r  r/  __classcell__r   r   r-  r   r$    s    r$  c                       s4   e Zd Zdd Zdd Zejj fddZ  Z	S )TestLinprogSimplexBlandc                 C      ddi| _ d S )Nrk  Tr   r   r   r   r   r%    r  z$TestLinprogSimplexBland.setup_methodc                 C   r  r&  r  r   r   r   r   r    r  z%TestLinprogSimplexBland.test_bug_5400c              	      s   | j ddi tt+ tt t   W d    n1 s#w   Y  W d    d S W d    d S 1 s;w   Y  d S r*  )	r   r  r   r   r,   r   r
   r,  r  r   r-  r   r   r/    s   "z-TestLinprogSimplexBland.test_bug_8174_low_tol)
r  r  r  r%  r  r   r  r  r/  r1  r   r   r-  r   r2    s
    r2  c                       s   e Zd Zdd Zedjdk Zej	
dZejjeoedd fdd	Z fd
dZdd Zejj fddZdd Zdd Z  ZS )TestLinprogSimplexNoPresolvec                 C   r3  )Nr  Fr   r   r   r   r   r%    r  z)TestLinprogSimplexNoPresolve.setup_methodr   ra   linuxz"Fails with warning on 32-bit linux)	conditionr  c                       t    d S r   )r,  r  r   r-  r   r   r    s   z*TestLinprogSimplexNoPresolve.test_bug_5400c                    sL   | j ddi tjtdd t  W  d    S 1 sw   Y  d S )Nr  r  zlinprog status 4r   )r   r  r   r   r,   r,  r  r   r-  r   r   test_bug_6139_low_tol  s   $z2TestLinprogSimplexNoPresolve.test_bug_6139_low_tolc                 C   r  r&  r  r   r   r   r   r'    r  z2TestLinprogSimplexNoPresolve.test_bug_7237_low_tolc                    r)  r*  r+  r   r-  r   r   r/    r0  z2TestLinprogSimplexNoPresolve.test_bug_8174_low_tolc                 C   r  Nz#Tests behavior specific to presolver  r   r   r   r   rc    r  zGTestLinprogSimplexNoPresolve.test_unbounded_no_nontrivial_constraints_1c                 C   r  r9  r  r   r   r   r   rd    r  zGTestLinprogSimplexNoPresolve.test_unbounded_no_nontrivial_constraints_2)r  r  r  r%  r5   ZintpitemsizeZ	is_32_bitr  r  r   Zis_linuxr   r  r>  r  r8  r'  r  r/  rc  rd  r1  r   r   r-  r   r4    s    	r4  c                       s:   e Zd ZddiZejjejdkdd fddZ	  Z
S )TestLinprogIPDenserv   Fr  z:Fails on some macOS builds for reason not relevant to testr  c                    r7  r   r,  r  r   r-  r   r   r    s   z TestLinprogIPDense.test_bug_6139)r  r  r  r   r   r  r  r  r  r  r1  r   r   r-  r   r;    s    r;  c                   @   s   e Zd ZdddZdS )TestLinprogIPSparseCholmodTrv   r   Nr  r  r  r   r   r   r   r   r=    s    r=  c                   @   s   e Zd ZdddZdd ZdS )TestLinprogIPSparseUmfpackTFr>  c                 C   r  )Nz2Failing due to numerical issues on some platforms.r  r   r   r   r   r    r  z=TestLinprogIPSparseUmfpack.test_network_flow_limited_capacityN)r  r  r  r   r  r   r   r   r   r@    s    
r@  c                       st   e Zd ZddddZejjejdkddej	d fdd	Z
ejjd
d fddZdd Zdd Z  ZS )TestLinprogIPSparseTFrv   r   Zsym_posr  /Fails on macOS x86 Accelerate builds (gh-20510)r  rThis test is sensitive to machine epsilon level perturbations in linear system solution in _linprog_ip._sym_solve.c                    r7  r   r<  r   r-  r   r   r       z!TestLinprogIPSparse.test_bug_6139Fails with ATLAS, see gh-7877c                    r7  r   r,  r  r   r-  r   r   r    r(  z!TestLinprogIPSparse.test_bug_6690c           	         s   t d\}}}}}d}t 6}tr|t |td |td  fdd jD }d|d< t|t	t
||| j|d	}W d    n1 sGw   Y  t|d
d d S )Nr!   r   zMatrix is exactly singularr~  c                    r   r   r   r   r   r   r   r   (  r   zLTestLinprogIPSparse.test_magic_square_sparse_no_presolve.<locals>.<dictcomp>Fr  r   r  r   )rM   r   r  r   r   r   r
   r   r	   rY   rZ   r   r2   )	r   r[   r\   rL   r  r_   r   r   r   r   r   r   $test_magic_square_sparse_no_presolve  s   
z8TestLinprogIPSparse.test_magic_square_sparse_no_presolvec           
         s   t d\}}}}}t =}|td |td  fdd jD }d}|D ]}||d< t|tt||t j	|d}	t
|	d	d
 q'W d    d S 1 sLw   Y  d S )Nr!   rn  zInvalid permc_spec optionc                    r   r   r   r   r   r   r   r   5  r   zATestLinprogIPSparse.test_sparse_solve_options.<locals>.<dictcomp>)ZNATURALZMMD_ATAZMMD_AT_PLUS_AZCOLAMDr   
permc_specr   r  r   )rM   r   r   r
   r   r	   rY   rZ   r_   r   r2   )
r   r[   r\   rL   r  r   r   Zpermc_specsrI  r   r   r   r   test_sparse_solve_options/  s   "z-TestLinprogIPSparse.test_sparse_solve_options)r  r  r  r   r   r  r  r  r  xfail_on_32bitr  r>  r  rH  rJ  r1  r   r   r-  r   rA    s    
rA  c                       sj   e Zd ZdddZejjejdkddej	d fddZ
d	d
 Zejjdd fddZ  ZS )TestLinprogIPSparsePresolveT)rv   r   r  rC  r  rD  c                    r7  r   r<  r   r-  r   r   r  D  rE  z)TestLinprogIPSparsePresolve.test_bug_6139c                 C   r  )Nz6_sparse_presolve=True incompatible with presolve=Falser  r   r   r   r   r  N  r  zBTestLinprogIPSparsePresolve.test_enzo_example_c_with_infeasibilityrF  c                    r7  r   rG  r   r-  r   r   r  Q  r(  z)TestLinprogIPSparsePresolve.test_bug_6690)r  r  r  r   r   r  r  r  r  rK  r  r  r>  r  r1  r   r   r-  r   rL  A  s    

rL  c                   @   s8   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d ZdS )TestLinprogIPSpecificr   c                 C   s|   t rddd}ntrddd}ndddd}tdd\}}}t|||| j|d}t|||| jd}t|j|jdd	d
d d S )NTr>  FrB  rS   r   r   z(linprog default solver unexpected resultgV瞯<gV瞯<r'   )has_cholmodr  rQ   r	   r   r   r-   )r   r   rJ   rK   rL   r  r  r   r   r   test_solver_select]  s   

z(TestLinprogIPSpecific.test_solver_selectc                 C   s4   dg}dg}t ||| jdddd}t|dd d S )Nr3   rY  FT)r  r   r  r   )r	   r   r2   r  r   r   r   )test_unbounded_below_no_presolve_originall  s   z?TestLinprogIPSpecific.test_unbounded_below_no_presolve_originalc                 C   r   )NrS   r   Tr   r   r   r   r   r   r   r   test_choleskyu  r   z#TestLinprogIPSpecific.test_choleskyc              	   C   s   t dd\}}}t '}|td |td |td t|||| jdddd}W d    n1 s5w   Y  t|dd	 d S )
NrS   r}  r~  r  T)ipr   r   r   r   )	rQ   r   r   r  r
   r   r	   r   r2   )r   rJ   rK   rL   r   r   r   r   r   test_alternate_initial_point|  s   z2TestLinprogIPSpecific.test_alternate_initial_pointc                 C   s   dg}dgdgg}ddg}dgdgdgg}g d}t  "}|t |td t|||||t| jd	d
id}W d    n1 sAw   Y  t|j d d S )Nr$   r   r   r   ri   re   r  r~  r  Fr   r   )	r   r   r  r
   r	   r_   r   r   r   r  r   r   r   r    s   

z#TestLinprogIPSpecific.test_bug_8664N)	r  r  r  r   rO  rP  rQ  rS  r  r   r   r   r   rM  W  s    	rM  c                   @   sH   e Zd Zi Zdd Zdd Zdd Zdd Zd	d
 Zdd Z	dd Z
dS )TestLinprogRSCommonc                 C   r  r  r  r   r   r   r   rm    r  z%TestLinprogRSCommon.test_cyclic_blandc           	      C   P   t  \}}}}}}}t|||||t| j| j|d	}t|||d t|jd d S Nr   r   x0r   r   rm   r	   r_   r   r   r2   r   r   rp  r   r   r   "test_nontrivial_problem_with_guess     
z6TestLinprogRSCommon.test_nontrivial_problem_with_guessc           
      C   X   t  \}}}}}}}g d}t||||||| j| j|d	}	t|	||d t|	jd d S )N)rp   rp   rq   rp   rW  r   r   rm   r	   r   r   r2   r   r   
r   rL   rY   rZ   r[   r\   rk   rl   r_   r   r   r   r   0test_nontrivial_problem_with_unbounded_variables     
zDTestLinprogRSCommon.test_nontrivial_problem_with_unbounded_variablesc           
      C   r\  )N)rY  r   Nrq   r  g333333?rW  r   r   r]  r^  r   r   r   .test_nontrivial_problem_with_bounded_variables  r`  zBTestLinprogRSCommon.test_nontrivial_problem_with_bounded_variablesc           
      C   sp   t  \}}}}}}}dg}tg d}d}g d}t||||||| j| j|d	}	t|	||d t|	jd d S )Nr$   )gibl3g%G/?r   r  g@Q$@)rp   ra  rq   rb  rW  r   r   )	rm   r5   r?   r	   r   r   r2   r   r   r^  r   r   r   8test_nontrivial_problem_with_negative_unbounded_variable  s   
zLTestLinprogRSCommon.test_nontrivial_problem_with_negative_unbounded_variablec           
      C   J   t  \}}}}}}}g d}t|||||t| j| j|d	}	t|	jd d S N)r   r   r!   rU   rW  rd   rm   r	   r_   r   r   r   r   
r   rL   rY   rZ   r[   r\   rk   rl   Z	bad_guessr   r   r   r   &test_nontrivial_problem_with_bad_guess     
z:TestLinprogRSCommon.test_nontrivial_problem_with_bad_guessc           
      C   s   t d\}}}}}tjj|j }t :}|td |td |t	 t
|||| jd}t
|||| j|jd}t
|| ||| j|jd}	W d    n1 sPw   Y  t|dd t|jd t|	 t|	j|jk  d S )	Nr!   rn  rs  )r[   r\   r   )r[   r\   r   rX  r  r   r   )rM   r5   r6   rB   rC   r   r   r
   r  r   r	   r   r.   r2   r   r   r   )
r   rJ   rK   rL   r  r{  r   r   r  res3r   r   r   %test_redundant_constraints_with_guess  s   
z9TestLinprogRSCommon.test_redundant_constraints_with_guessN)r  r  r  r   rm  rZ  r_  rc  rd  ri  rl  r   r   r   r   rT    s    rT  c                   @      e Zd ZddiZdS )TestLinprogRSBlandpivotrk  Nr?  r   r   r   r   rn        rn  c                   @   s   e Zd ZdZi Zdd ZdS )TestLinprogHiGHSSimplexDualr   c                 C   s   t  \}}}}t||||| j| jd}t|jd t|jdu tt	|j
dk tt	|jdd |D k tt	|jdd |D k dS )z
        The scaled model should be optimal, i.e. not produce unscaled model
        infeasible.  See https://github.com/ERGO-Code/HiGHS/issues/494.
        )rY   rZ   r_   r   r   r   Ngưc                 S   s"   g | ]\}}|d u rt jn|qS r   r5   rV   r   r]   r^   r   r   r   
<listcomp>  s    zCTestLinprogHiGHSSimplexDual.test_lad_regression.<locals>.<listcomp>c                 S   s(   g | ]\}}|d u rt j n|d qS )Nrg  rr  rs  r   r   r   rt    s    )r}   r	   r   r   r   r   r   r.   r5   allr   )r   rL   rY   rZ   r|   r   r   r   r   test_lad_regression  s   
z/TestLinprogHiGHSSimplexDual.test_lad_regressionN)r  r  r  r   r   rv  r   r   r   r   rq    s    rq  c                   @   s   e Zd ZdZi ZdS )TestLinprogHiGHSIPMr=  N)r  r  r  r   r   r   r   r   r   rw   	  s    rw  c                   @   s   e Zd ZdZi Zejdejje	j
dk oe dkddddd	 Zd
d Zdd Zdd Zdd Zejjdd Zejjdd Zdd Zdd ZdS )TestLinprogHiGHSMIPr   rR   l        LinuxFzgh-16347)r6  runr  c              	   C   s  d}t |\}}}}}dgt| }dgt| }t|d |||| j|d}	| |	j |d ||}
tj|
dd}tj	
|jdd| tj	
|jdd| tj	
t| | tj	
t|d d d d df  | tj	j
|	jt|	jd	d
 d S )Nr$   r   r   r   rL   r[   r\   r_   r   r   r   r  r3   r  r  )rM   r   r	   r   r>   r.   r9   r5   r+  r   r   Zdiaground)r   rD   rJ   rK   rL   rG   rF   r_   r   r   sZsquarer   r   r   	test_mip1	  s   	* zTestLinprogHiGHSMIP.test_mip1c                 C   s   t ddgddgg}t ddg}t ddg }t dt jfgt| }t |}t||||| j|d	}t j|j	ddg t j|j
d
 d S )Nr   r   ri   rR   r3   r  r   r   r   rf   )r5   r?   rV   r   Z	ones_liker	   r   r   r   r.   r-   r   r   r   r   	test_mip2(	  s   

zTestLinprogHiGHSMIP.test_mip2c                 C   s   t ddgddgddgg}t g d}t ddg }dt jfgt| }dgt| }t||||| j|d}t j|jd t 	|j
ddgsUt 	|j
ddgsWJ d S d S )	Nr3   r   r!   r   r   r   r   r   )r5   r?   rV   r   r	   r   r   r   r-   r
  r.   r   r   r   r   	test_mip3;	  s   
0zTestLinprogHiGHSMIP.test_mip3c                 C   s   t ddgddgddgg}t g d}t ddg}dt jfgt| }ddg}t||||| j|d	}t j|jd
dg t j|j	d d S )Nr3   r   rj   r   r   )r3  irS   ra   r   r   g      @rh   ;   )
r5   r?   rV   r   r	   r   r   r   r.   r-   r   r   r   r   	test_mip4L	  s   
zTestLinprogHiGHSMIP.test_mip4c           	   
   C   s   t g dg}t dg}t g dg}t dg}t g d}dt jfdt jfdg}g d}t||||||| j|d	}t j|jg d
 t j|jd |	dd d us[J |	dd d useJ |	dd d usoJ d S )NrA  rh   )r$   r   r   r   )rf   r   r3   r   r   r  )rL   rY   rZ   r[   r\   r_   r   r   )r   rd   r   iZmip_node_countZmip_dual_boundmip_gap)
r5   r?   rV   r	   r   r   r   r.   r-   r  )	r   rY   rZ   r[   r\   rL   r_   r   r   r   r   r   	test_mip5\	  s    zTestLinprogHiGHSMIP.test_mip5c                 C   s   t g dg dg dg dg}t g d}t g d}dt jfgd }d	gd }t||||| j|d
}t j|jd d S )Nr  r  r  ry  r  r!   r3  r  '   rv  r  r  r  r  r  r  rw  r3  r  r  r  &   r  r  )   r  r  r_  rw  r  rv  r  i  i(  i:,  i/  r   rR   r  r  rh   r   rh   r!   r   ra   r   r{  i>  )r5   r?   rV   r	   r   r   r   r-   )r   r[   r\   rL   r_   r   r   r   r   r   	test_mip6t	  s   


zTestLinprogHiGHSMIP.test_mip6c                 C   s   t g dg dg dg dg}t g d}t g d}dt jfgd }d	gd }g d
}g }|D ]!}t|tt|||| j|d|id	}	|	d }
|
|ksPJ ||
 q4t t 	|}t 
|dksgJ t 
|dkrpJ d S )Nr  r  r  r  r  r  r   ra   r   )rU   g      ?g{Gz?r  mip_rel_gap)	rL   rY   rZ   r[   r\   r_   r   r   r   r  )r5   r?   rV   r	   rY   rZ   r   r=   diffflipru  )r   r[   r\   rL   r_   r   Zmip_rel_gapsZsol_mip_gapsr  r   Zfinal_mip_gapZ	gap_diffsr   r   r   test_mip_rel_gap_passdown	  s.   

z-TestLinprogHiGHSMIP.test_mip_rel_gap_passdownc                 C   st   t g d}t ddgddgddgddgg}t g d}t|||dd}t j|jg d |jdks8J d S )	N)r   r   r3   r3   rU   r  )r   r!   r   r!   r   )r_   r   r   )r   r   r  r   r   )r5   r?   r	   r   r   r.   r   )r   rL   r_   r   r   r   r   r   test_semi_continuous	  s   "z(TestLinprogHiGHSMIP.test_semi_continuousc                 C   sv   ddg}ddgg}dg}t |||ddgd}t |||tddgd}t |||dd}t|j|j t|j|j dS )z
        Test that when integrality is a list of all zeros, linprog gives the
        same result as when it is an array of all zeros / integrality=None
        r   r3   r   r   r   N)r	   r5   r  r   r.   )r   rL   rY   rZ   r  r  rk  r   r   r   test_bug_20584	  s   
z"TestLinprogHiGHSMIP.test_bug_20584N)r  r  r  r   r   r   r  r#  r>  r  maxsizer  systemr~  r  r  r  r  Zxslowr  r  r  r  r   r   r   r   rx  
	  s(    



%rx  c                   @   s&   e Zd ZddiZejZejZejZdS )AutoscaleTestsr  TN)r  r  r  r   r   r  r  r  r   r   r   r   r  	  s
    
r  c                   @   r  )TestAutoscaleIPr   c                 C   s   d| j d< t| S )Nr  r  )r   r  r  r   r   r   r   r  	  s   

zTestAutoscaleIP.test_bug_6139N)r  r  r  r   r  r   r   r   r   r  	  s    r  c                   @   r  )TestAutoscaleSimplexr   Nr  r   r   r   r   r  	  s    r  c                   @   s    e Zd ZdZdd Zdd ZdS )TestAutoscaleRSr   c           	      C   rU  rV  rY  rp  r   r   r   rZ  	  r[  z2TestAutoscaleRS.test_nontrivial_problem_with_guessc           
      C   re  rf  rg  rh  r   r   r   ri  	  rj  z6TestAutoscaleRS.test_nontrivial_problem_with_bad_guessN)r  r  r  r   rZ  ri  r   r   r   r   r  	  s    r  c                   @   s2   e Zd ZdZeZejZejZej	Z	ej
ZejZdS )RRTestsr   N)r  r  r  r   r   ZLCTro  Ztest_RR_infeasibilityr  r  r  Z	test_NFLCr  r   r   r   r   r  	  s    
r  c                   @   rm  )	TestRRSVDr   ZSVDNr?  r   r   r   r   r  
  rp  r  c                   @   rm  )TestRRPivotr   ro  Nr?  r   r   r   r   r  
  rp  r  c                   @   rm  )TestRRIDr   IDNr?  r   r   r   r   r  
  rp  r  )NNr&   r&   )r   )r   ra   re   rn   )Xr  r  r  numpyr5   Znumpy.testingr   r   r   r   r   r   r   r   r   Zscipy.optimizer	   r
   Zscipy.optimize._numdiffr   Zscipy.sparse.linalgr   Zscipy.linalgr   Zscipy._lib._utilr   Zscipy.sparseru   r  Zscikits.umfpackr   ImportErrorrN  ZsksparseZsksparse.cholmodr   Zcholmodr   r   r"   r%   r2   rM   rQ   r`   rm   r}   r   r  r  r   r   r   r   r   rY   rZ   r[   r\   r_   r   filterwarningsr  r  r  r  r$  r2  r4  r;  r=  r@  rA  rL  rM  rT  rn  rq  rw  rx  r  r  r  r  r  r  r  r  r   r   r   r   <module>   s     

B

#


	-          
S

 <.6
D?	
 
G
