o
    tBhK                    @   s  d 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ZddlZd	Zzdd
lmZ W n eyW   dZY nw d	ZzddlZddlm Z! W n eyq   dZY nw dd Z"dd Z#dd Z$dd Z%		d_ddZ&dd Z'dd Z(d`ddZ)dd Z*dad#d$Z+d%d& Z,d'd( Z-d)d* Z.dZ/dZ0dZ1dZ2dZ3G d+d, d,Z4G d-d. d.e4Z5G d/d0 d0e4Z6G d1d2 d2e4Z7G d3d4 d4e4Z8G d5d6 d6e5Z9G d7d8 d8e5Z:G d9d: d:e5Z;G d;d< d<e6Z<erG d=d> d>e6Z=erG d?d@ d@e6Z>G dAdB dBe6Z?G dCdD dDe6Z@G dEdF dFZAG dGdH dHe7ZBG dIdJ dJe7ZCG dKdL dLe8ZDG dMdN dNe8ZEG dOdP dPZFG dQdR dReFZGG dSdT dTeFZHG dUdV dVeFZIG dWdX dXZJG dYdZ dZeJZKG d[d\ d\eJZLG d]d^ d^eJZMdS )bz"
Unit test for Linear Programming
    N)assert_assert_allcloseassert_equalassert_array_lessassert_warnssuppress_warnings)raises)linprogOptimizeWarning)approx_derivative)MatrixRankWarning)LinAlgWarningT)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   x/var/www/html/riverr-enterprise-integrations-main/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_sol1   s   r$   :0yE>c                 C   sj   | j sd| j| j}t|t| jd |d ur#t| j|d||d |d ur3t| j|d||d d S d S )Nz linprog status {0}, message: {1}r   z*converged to an unexpected objective valueerr_msgrtolatolz#converged to an unexpected solution)	r   formatr   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}
t j|	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dtype)nprandomseedarangereshapezerosrangecopyappendflattenarrayvstackfloatrandshape)nMnumbersr;   A_listb_listiA_rowjAbcr   r   r   magic_squareP   sT   ((0
""
rP   c                 C   s   t jd t jj| |fd }t | | | f}t | }t| D ]}|d }d|||| || f< ||  ||< q"t || | f}t |}t|D ]}|d }d|||d|f< d||< qMt ||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)	r6   r7   r8   exponentialr;   r<   rA   hstackravel)mrE   rO   ArowbrowrL   j1AcolbcolrM   rN   r   r   r   lpgen_2d   s"   


r\   c                 C   s   t j|  d\}}}t j|d }t j||d }t j|d }t j||d }t j|d }t j| }	t j|}
t j |	|	t j  k < t j|
|
t j k< t |	|
fj}||||||fS )N)
      2         ?)r6   r7   r8   rC   infrA   T)r8   m_eqm_ubrE   rO   A_ubb_ubA_eqb_eqlbubboundsr   r   r   very_random_gen   s   
rl   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#   )ro      	   )r   r3   r   )r]   ro      )rp   r3   r    r#   )rr   rp   rn   rp   )rs   r   r   )gk椀?gs?r   gkL?gqY:^@r   )rO   re   rf   rg   rh   x_starf_starr   r   r   nontrivial_problem   s   
ry   rm   rq   d   c                 C   sn  t j|  t jjdd||fd}t j|}t jjdd||fd}|||  t j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 }
qP|	
 }	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   rQ   h㈵>Nr   r3   NNr   N)r6   r7   r8   normalrandnonesscipysparse
lil_matrixr<   tocscr;   )r8   rV   drE   phiw_truer.   yrO   re   idxiirf   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+   completephaser   r   r   r#   nitr.   r-   slackcon)popr   r<   
isinstanceintboolstr)r   r+   r   last_cbr   r   cb   s   

z!generic_callback_test.<locals>.cbrr   r   r   r   r]   rm   r#   re   rf   callbackmethod      2rp   r/   r0   r-   r.   r   r   )r6   r@   r	   r   r2   r   selfr   rO   re   rf   r   r   r   r   generic_callback_test   s   r   c                  C   sp   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 ttt| ||d
did d S )Nrr   r   r   r   r   r   ekki-ekki-ekkire   rf   r   z
highs-ekki	rr_methodre   rf   options)r6   r@   assert_raises
ValueErrorr	   )rO   re   rf   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 )Nrr   r   r   r   r   r   highsr   r   rp   r   )r6   r@   r	   r2   )rO   re   rf   r   r   r   r   test_choose_solver  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
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Z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 Z-dVdW Z.dXdY Z/dZd[ Z0d\d] Z1d^d_ Z2d`da Z3dbdc Z4ddde Z5dfdg Z6dhdi Z7djdk Z8dldm Z9dndo Z:dpdq Z;drds Z<dtdu Z=dvdw Z>dxdy Z?dzd{ Z@d|d} ZAd~d ZBdd ZCdd ZDdd ZEdd ZFdd ZGdd ZHdd ZIdd ZJdd ZKdd ZLdd ZMdd ZNeOjPjQeRjSdkdddd ZTdS )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 )Nr^   dispTre   rf   r   r   T*Pr/   r\   r	   r   r2   r   rM   rN   rO   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#   rr   r   r   rp   r|   rr   N)re   rf   rk   r   r   r   )r	   r   r   r2   )r   rO   rM   rN   	x0_bounds	x1_boundsr   r   r   r   test_docstring_exampleF  s   z)LinprogCommonTests.test_docstring_examplec              	   C   s0   dg}dgg}d}t tt|||| j| jd d S )Nr   hellorg   rh   r   r   )r   	TypeErrorr	   r   r   )r   rO   rg   rh   r   r   r   test_type_errorQ  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   rv   r   )r6   r@   r=   ra   r	   rg   rh   r   r   r2   r   )r   rO   re   	b_ub_origrf   rk   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   )r6   r@   r=   ra   r	   re   rf   r   r   r2   r   )r   rO   rg   	b_eq_origrh   rk   r   r   r   r   test_aliasing_b_eqh  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   rO   re   rf   rg   rh   rk   r   r   r   r   test_non_ndarray_argsu  s   z(LinprogCommonTests.test_non_ndarray_argsc                    sz   t ddg}ddgddgddgg}g d}d d d d d i f fdd	} fd	d
 jD }d|d< tt|||||d d S )Nrr   r   r   r   r   r   c              
      s   t | ||||| j|d d S Nr   r	   r   )rO   re   rf   rg   rh   rk   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>*   spamr   )r6   r@   r   r   r
   )r   rO   re   rf   r   or   r   r   test_unknown_options  s   
z'LinprogCommonTests.test_unknown_optionsc                    s  d fdd	}t t|g dddgd t t|g dg dd 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 rd S t t|d	d
gtdd	d d S )Nc              
          t | ||||| j jd d S r   r	   r   r   rO   re   rf   rg   rh   rk   r   r   r   r        
z1LinprogCommonTests.test_invalid_inputs.<locals>.fr   r   r    r   r    r#   rk   )r   r   )r    r#      r   r   r   r   )re   rf   )rg   rh   _sparse_presolve)r   r   r    )re   rh   NNNNN)r   r   r   r6   r;   )r   r   r   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 j|}t j|}|t j| }|t j| }t ||f}	|| }
 jdv rot	t
d j d	d
 t|||
|	 j jd W d    d S 1 shw   Y  d S i  j} jdv r}d|d< t|||
|	 j|d}|jsJ d S )Nc              
      r   r   r   r   r   r   r   r     r   z5LinprogCommonTests.test_sparse_constraints.<locals>.fr   rz      r`   >   simplexrevised simplexzMethod 'z.' does not support sparse constraint matrices.match)rO   rg   rh   rk   r   r   >   interior-pointTr   r   )r6   r7   r8   r   r   rC   r   column_stackr   r   r   r	   r   r   )r   r   rV   rE   rg   x_validrO   rj   ri   rk   rh   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#   rm   r    r   r   r   r   r   r    r3   r   r   r          @rm   r   r3   r   rm   r]   r#   r   r   r3         iX  r   r    r   r   )r6   r7   r8   r	   r   r   r   r   )r   rO   rM   rN   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   rk   r   r   r   r   ))r   r   )r3   r3   )r    r       )r   r3   r    )r   r    r   )r   r   r   r   r   r   )r   r   r   rt   ))r   )r   r]   )      @r  )rg   rh   rk   r   r      )r   rt   r  )r   getr	   r   r2   r   r   r   do_presolver   r   r   r   test_bounds_fixed  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_infeasible  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   )rO   rk   r   r   r   r   g#  s   z6LinprogCommonTests.test_bounds_infeasible_2.<locals>.gr   rO   rk   r   r   r   )r   r  r   r6   ra   pytestwarnsRuntimeWarningr   
IndexErrorr	   r   r   r   )r   r  simplex_without_presolverO   bounds_1bounds_2r  r   r   r   r   test_bounds_infeasible_2  s:   	"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   rO   r   r   r   r   test_empty_constraint_18  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)	r6   ra   r	   r   r   r!   r  r   r   r   rO   rk   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/   )r6   ra   r	   r   r   r2   r  r   r   r   test_empty_constraint_3G  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   irp   r   	r6   r@   r	   rg   rh   rk   r   r   r2   r   rO   re   rf   r   r   r   r   test_inequality_constraintsN  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 )Nrp   r    r   r3   r   r   )r   r3   r3   r   r   gUUUUUU?gUUUUUU?r   r	   rg   rh   rk   r   r   r2   r  r   r   r   test_inequality_constraints2Z  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	   re   rf   rg   rh   r   r   r2   r  r   r   r   test_bounds_simpleg  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   r6   r@   r	   re   rf   r   r   r2   r   rO   rg   rh   rk   r   r   r   r   test_bounded_below_only_1s     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   r`   r   r  r6   r   eyer@   ra   r	   re   rf   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#   rr   r   r   r4   rp   r   gm۶m&g$I$Ig%I$I@r   )
r6   r@   float64ra   r	   rg   rh   r   r   r2   )r   rO   re   rf   r   r   rk   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 Nrv   r   rt   r   r   r   )   r   r    r   r   r}   r   )r	   rg   rh   r   r   r   r   rO   re   rf   rk   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 r0  )r	   re   rf   r   r   r   r$  r   r   r   !test_bounds_equal_but_infeasible2  r5  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#   rm   r   r2  c                    r   r   r   r   r   r   r   r     r   zDLinprogCommonTests.test_bounds_equal_no_presolve.<locals>.<dictcomp>Fr   r   )r   r	   rg   rh   r   r   r   rO   re   rf   rk   r   r   r   r   r   test_bounds_equal_no_presolve  s   z0LinprogCommonTests.test_bounds_equal_no_presolvec           
   
   C   s   d\}}t jd t j|}d|d< t j||}d|d d df< t j|}g dg}d}g d}t||||||| j| jd}	t|	dd	 d S )
Nr   r   r   )r   r   r   r   r    )rs   r]   r9  )rs   Nr|   r   gyMj#r   )r6   r7   r8   rC   r	   r   r   r2   
r   rV   rE   rO   rg   rh   re   rf   rk   r   r   r   r   test_zero_column_1  s   
z%LinprogCommonTests.test_zero_column_1c           
   
   C   s   t jd d\}}t j|}d|d< t j||}d|d d df< t j|}t j||}d|d d df< t j|}d}t||||||| j| jd}	t|	 | jddrat	|	j
d d S d S )	Nr   r"   r3   r   r|   r   r   T)r6   r7   r8   rC   r	   r   r   r!   r  r   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	   re   rf   rk   r   r   r2   r   rO   rg   rh   r   r   r   r   test_zero_row_1  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=  r>  r?  r   r   r   r   r  )r   re   rf   rO   r   r   r   r   test_zero_row_2  s   z"LinprogCommonTests.test_zero_row_2c              
   C   s   d\}}t j|}t j||}d|dd d f< t j|}t|tt||t| j| jd}t	| | j
ddr@t|jd d S d S Nr"   r   r   r   T)r6   r7   rC   r	   re   rf   rk   r   r   r   r  r   r   )r   rV   rE   rO   rg   rh   r   r   r   r   test_zero_row_3  s   z"LinprogCommonTests.test_zero_row_3c              
   C   s   d\}}t j|}t j||}d|dd d f< t j| }t|||ttt| j| jd}t	| | j
ddrAt|jd d S d S rD  )r6   r7   rC   r	   rg   rh   rk   r   r   r   r  r   r   )r   rV   rE   rO   re   rf   r   r   r   r   test_zero_row_4  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	   re   rf   rk   r   r   r   r  r   r   rA  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 )	NrG  rH  rI  rJ  )r   r   r   r#   r   r#   r   r@  rA  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 )NrG  rH  rI  r3   r   r   r   rJ  )r   r   r   r#   r|   r}   r}   r}   r   r   Tr   	r	   rg   rh   r   r   r   r  r   r   r3  r   r   r   test_singleton_row_ub_14  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 )NrG  rH  rI  rM  rJ  )r   r         r#   rN  r   r`   r   )r	   rg   rh   r   r   r2   r3  r   r   r   test_singleton_row_ub_2A  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	   rg   rh   rk   r   r   r   r  r   r   r   test_infeasibleJ  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   rp   r   r   Tr   rO  r3  r   r   r   !test_infeasible_inequality_boundsU  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   )	r6   r@   r	   rg   rh   rk   r   r   r!   r  r   r   r   test_unboundedb  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   p  r   zQLinprogCommonTests.test_unbounded_below_no_presolve_corrected.<locals>.<dictcomp>Fr   r  r   r   )r   r	   r   r   r   r!   )r   rO   rk   r   r   r   r   r   *test_unbounded_below_no_presolve_correctedl  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   )r|   r|   r|   r  r  r}   r   r   r3   N$   $The problem is (trivially) unboundedr6   r@   r	   rg   rh   r   r   r!   lower
startswithr   r.   ra   r+   r3  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 )rX  )r   r   r   r   r3   r   rY  rZ  )r   r   r   r   r   r   r[  )r|   r|   r|   r  r  )Nr   r   r   r3   Nr\  r]  r^  r3  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)rz   r]   r   r3   r  )r^   r   r   )   r^   r   )r   rz   '  r   )r   r   rd  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)rs   9   rq   g      8@)r`   g            rq   )r`   g      rQ  r   rH  )r   r   r   rz   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   )r6   r@   r   itemsr	   rg   rh   rk   r   r  r   r2   )r   rO   re   rf   r   r   r   r   r   r   test_cyclic_bland  s   
z$LinprogCommonTests.test_cyclic_blandc                 C   s   d\}}t j|}t j||}t j|}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 sQw   Y  t| d S )N)r]   r]   r   r   r3   A_eq does not appear...r   )r6   r7   rC   r   filterr
   r	   re   rf   rk   r   r   r   )r   rV   rE   rO   rg   rh   supr   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   )ry   r	   rk   r   r   r2   	r   rO   re   rf   rg   rh   rw   rx   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 )Nr^   $Solving system with option 'sym_pos'invalid value encounteredr   r   r   )r\   r   rn  r
   r  r   r	   rg   rh   rk   r   r   r2   )r   re   rf   rO   ro  r   r   r   r   test_lpgen_problem  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#   rq      r#   r    rm   rt   r   r        r  r   )r      i!   r   r   ir   i  ư>re  )r/   r)   r(   )
r   rn  r   r	   re   rf   rk   r   r   r2   )r   rO   rE   prg   rh   ro  r   r   r   r   test_network_flow  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  )rv   r   r   r#   scipy.linalg.solve
Ill...rm  Solving system with option...r   r1  r   )r   has_umfpackrn  r   r  r
   r   r	   re   rf   r   r   r2   )	r   rO   rk   rE   r|  rg   rh   ro  r   r   r   r   "test_network_flow_limited_capacity  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   rr   rv   )r    r   r   )r   r   r    r]   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*>re  )r/   r0   r)   r(   r@  rA  r   r   r   test_enzo_example0  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)gffffff@g333333@g%@gffffff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   )rQ  皙?333333?r  r  rm  r   gRQ)r  g?        r  g?r  r   )
r   rn  r
   r	   re   rf   rk   r   r   r2   )r   rO   rg   rh   ro  r   r   r   r   test_enzo_example_bD  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 )Nr^   r   r   r   r   r   )r6   r   pir9   rA   cossinr	   re   rf   rk   r   r   r2   r;   r   rV   rO   tmprg   rh   r   r   r   r   #test_enzo_example_c_with_degeneracyU  s   "z6LinprogCommonTests.test_enzo_example_c_with_degeneracyc              
   C   sz   d}t | }dt j t | |d  }t t |d t |f}ddg}t|tt	||t
| j| jd}t| d S )Nr_   r   r   r   r   )r6   r   r  r9   rA   r  r  r	   re   rf   rk   r   r   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 )	Nr_   r   r   c                    r   r   r   r   r   r   r   r   s  r   zMLinprogCommonTests.test_enzo_example_c_with_infeasibility.<locals>.<dictcomp>Fr   r   )r6   r   r  r9   rA   r  r  r   r	   re   rf   rk   r   r   )r   rV   rO   r  rg   rh   r   r   r   r   r   &test_enzo_example_c_with_infeasibilityk  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)   )r6   r@   r	   rk   r   r   r2   
zeros_like)r   rO   re   rf   rg   rh   r   r   r   r   test_basic_artificial_varsz  s   
z-LinprogCommonTests.test_basic_artificial_varsc           	   
   C   s   t d\}}}}}}t||||||| j| jd}t|j t|j t|j  t|jdk t	||j
 |j t	|||j
  |jdd t	|||j
  |jdd dD ] }|| v rtt|| d tjshJ t|| d tjstJ qTd S )	Nr   re   rf   rg   rh   rk   r   r   z%Optimization terminated successfully.gdy=r)   )eqlinineqlinr_  upper	marginalsresidual)rl   r	   r   r   r   r   r   r   r+   r   r.   r-   r   r   keysr   r6   ndarray)	r   rO   re   rf   rg   rh   rk   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)r}   r   rz   r  r  r  r  r  r     r  r  r  r  r  r}   r}   r}   r}   r}   r}   gqq?     g)r   gQr   r   rr   r   r   r   r3   r3   r   r3   r3   r   r   r   r   r   r   r   r   g333333rr   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   rz   rz   rz   rz   rz   rz   r  r  r  r  r  r  r   r   r   r   r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rs  rt  r   g^Zr   )r6   r@   r   rn  r
   r  r   r	   rg   rh   r   r   r2   )
r   rk   r   r  hre   rf   rO   ro  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 )Nr>  )r   r  r  )     @r  r  g      @r  )r  g    .Ag    .Ag    cAr|   r   gfffff-@)r   g@r   r   r6   r@   r	   r   r   r2   )r   rO   rg   rh   re   rf   rk   r   r   r   r   test_bug_6139  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  r`   g)\(?g(\?{Gz?g333333ÿ)
g=
ףp=gffffff?g?g(\g(\gp=
ף?g
ףp=
@g{Gz?gQ?gQ?z%Solving system with option 'cholesky'rs  rt  r   gV)
r  r  r  r  r  r`   r  r  r  r  r   r{  r   r   )r6   r@   rb   r   r  rn  r   r
   r  r   r	   r   r   r2   r   r.   )r   rg   rh   re   rf   rk   rO   ro  r   r/   r0   r)   r   r   r   test_bug_6690  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 s5w   Y  d}t||d t||j| tt|jjd |j d S )Nr    rm  rt  r   -U?r   r{   )rP   r   rn  r
   r  r   r	   re   rf   rk   r   r   r2   r   dotr.   r   r6   r;   rR   )r   rg   rh   rO   Nro  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   r|   rq   r   ghb$[@r{  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   r   rt  r   tolg&.>绽|=r   g r{  r  )r6   r@   r;   rD   r   r   rn  r  r   r	   rg   rh   r   r   r  r$   r2   )r   re   rf   rO   rk   ro  r   r   r   r   test_bug_8174T  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   rt   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  rz   r   rm  r   g?E@r   )r6   r@   identityr   r  rn  r   r
   r	   rk   r   r   r2   r   rO   re   rf   rg   rh   ro  r   r   r   r   test_bug_8174_2k  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)rt   r   rv         ?r  )r#         @r  r         )r   rh  r   r   r`   )r    rQ  r#   g      )ro   )r3   g      @r   r  r   )r  r   g      r3   g      #@)r   r   r   r   r   )re   rf   r   r   )r   r   ry  gUUUUUU@gUUUUUU#@r   )r6   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)rs   r]   rp   r    )rm   ru   rv   rp   )ru   rm   r#   rn   )rv   r#   rm   rv   )r    rr   rr   rs   )rq   r  rv   )r}   r}   r}   r}   g      B@rt  r   )r   r   r3   r   r   r|   r   r{   )r/   r(   )r   r  rn  r   r  r   r	   rg   rh   r   r   r>   r2   )
r   rO   re   rf   rk   r/   ro  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   ro   rn   r}   r|   r   g۶m۶m?gI$I$@r  )r	   re   rf   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   ru   rq   r    r   r]   r  r   )	r   rn  r  r
   r	   rk   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   )r|   r|   r|   r  r  r   r   N)r6   r@   r	   rg   rh   r   r   r2   r   r.   r-   r3  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   r|   r   r   r  N)	r6   r;   r@   r	   rg   rh   r   r   r2   r3  r   r   r   test_bug_8973_2  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   r|   r3   r#   rr   r   rp   r   r   Tr   r]   r   r  N)
r6   r;   r@   updater   r	   rg   rh   r   r2   r7  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   )      r]         f   r   r   r#   N)axisrm  r   )   \   r  r  r   r]   r  r  )r6   r@   concatenater;   r   rn  r
   r	   re   rf   rk   r   r   r2   )r   rg   rh   rO   ro  r   r   r   r   test_bug_10349  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   rn  r
   r  r   r  r   r	   re   rf   rk   r   r-   )r   rO   rg   rh   r   ro  r   r   r   r   test_bug_10466  s>   
z!LinprogCommonTests.test_bug_10466N)U__name__
__module____qualname____doc__r   r   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  r6  r8  r;  r<  rB  rC  rE  rF  rK  rL  rP  rR  rS  rT  rU  rW  ra  rb  rf  rl  rp  rr  ru  r}  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  markskipifsysplatformr  r   r   r   r   r   /  s    
! &
								
7,)$r   c                   @      e Zd ZdZdS )LinprogSimplexTestsr   Nr  r  r  r   r   r   r   r   r  J      r  c                   @   r  )LinprogIPTestsr   Nr  r   r   r   r   r  N  r  r  c                   @   s(   e Zd ZdZdd Zdd Zdd ZdS )	LinprogRSTestsr   c                 C      t d d S Nz Intermittent failure acceptable.r  skipr   r   r   r   r  e     zLinprogRSTests.test_bug_5400c                 C   r  r  r  r   r   r   r   r  h  r  zLinprogRSTests.test_bug_8662c                 C   r  r  r  r   r   r   r   r}  k  r  z LinprogRSTests.test_network_flowN)r  r  r  r   r  r  r}  r   r   r   r   r  R  s
    r  c                   @   sz   e Zd Zdd Z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dd Z	dd Z
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   <lambda>r  s    z1LinprogHiGHSTests.test_callback.<locals>.<lambda>rr   r   r   r   r   r   r   r   r   rp   r   )r6   r@   r   NotImplementedErrorr	   r   r2   r   r   r   r   r   p  s   zLinprogHiGHSTests.test_callbackr   r   r3   r   r   
time_limitdual_feasibility_toleranceprimal_feasibility_toleranceipm_optimality_tolerance!simplex_dual_edge_weight_strategy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   s^   t ddgd }t ddgg}dg}t|||ttt| j| jd}t|j	dk| jdk d S )Nr   r3   r  r   	highs-ipm)
r6   r@   r	   rg   rh   rk   r   r   r   crossover_nitr  r   r   r   test_crossover  s   
z LinprogHiGHSTests.test_crossoverc           
   
      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   r8   r  c              	      s   t |  jdjS Nr   r	   r   r-   r.   )rg   re   rh   rk   rO   r   r   r   f_bub  
   z/LinprogHiGHSTests.test_marginals.<locals>.f_bubz3-point)r   f0c              	      s   t  | jdjS r  r  r	  )rg   re   rf   rk   rO   r   r   r   f_beq  r  z/LinprogHiGHSTests.test_marginals.<locals>.f_beqc              	      s*   t | gj}t |jdjS r  r6   r@   rb   r	   r   r-   r.   rk   )rg   re   rh   rf   rO   r   rj   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  )rg   re   rh   rf   rO   ri   r   r   r   f_ub  r  z.LinprogHiGHSTests.test_marginals.<locals>.f_ub)rl   r	   r   r   rb   r   r-   r   r  r  r  r6   errstateisfiniter_  r  )
r   r   r
  dfdbubr  dfdbeqr  dfdlbr  dfdubr   )	rg   re   rh   rf   rk   rO   ri   r   rj   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  )rl   r	   r   r   rb   r  r  r  r  r_  r   )	r   rO   re   rf   rg   rh   rk   r   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   )	rl   r	   r   r   r6   allcloser  r  r.   r   r   r   r   test_complementary_slackness  s
   
&z.LinprogHiGHSTests.test_complementary_slacknessN)r  r  r  r   r  r  parametrizer  r  r  r  r   r   r   r   r   r  o  s$    

	1r  c                       s4   e Zd Zdd Zdd Zdd Z 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-  __classcell__r   r   r+  r   r"    s
    r"  c                       s,   e Zd Zdd Zdd Z fddZ  ZS )TestLinprogSimplexBlandc                 C      ddi| _ d S )Nrj  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   r+  r   r0    s    r0  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 fddZdd Zdd Z  ZS )TestLinprogSimplexNoPresolvec                 C   r1  )Nr   Fr   r   r   r   r   r#    r  z)TestLinprogSimplexNoPresolve.setup_methodr   rm   linuxz"Fails with warning on 32-bit linux)	conditionr  c                       t    d S r   )r*  r  r   r+  r   r   r       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-  $  r.  z2TestLinprogSimplexNoPresolve.test_bug_8174_low_tolc                 C   r  Nz#Tests behavior specific to presolver  r   r   r   r   ra  +  r  zGTestLinprogSimplexNoPresolve.test_unbounded_no_nontrivial_constraints_1c                 C   r  r8  r  r   r   r   r   rb  .  r  zGTestLinprogSimplexNoPresolve.test_unbounded_no_nontrivial_constraints_2)r  r  r  r#  r6   intpitemsize	is_32_bitr  r  r`  is_linuxr  r  xfailr  r7  r%  r-  ra  rb  r/  r   r   r+  r   r2  
  s    	r2  c                   @      e Zd ZddiZdS )TestLinprogIPDenser   FNr  r  r  r   r   r   r   r   r?  7      r?  c                   @   s   e Zd ZdddZdS )TestLinprogIPSparseCholmodTr   r   Nr@  r   r   r   r   rB  <  s    rB  c                   @   s&   e Zd ZdddZdd Zdd ZdS )	TestLinprogIPSparseUmfpackTFrC  c                 C   r  )Nz0Autoscale doesn't fix everything, and that's OK.r  r   r   r   r   r  D  r  z)TestLinprogIPSparseUmfpack.test_bug_10466c                 C   r  )Nz2Failing due to numerical issues on some platforms.r  r   r   r   r   r  G  r  z=TestLinprogIPSparseUmfpack.test_network_flow_limited_capacityN)r  r  r  r   r  r  r   r   r   r   rD  A  s    
rD  c                       s^   e Zd ZddddZ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r   r   sym_posrThis test is sensitive to machine epsilon level perturbations in linear system solution in _linprog_ip._sym_solve.c                    r5  r   r*  r  r   r+  r   r   r  N  r6  z!TestLinprogIPSparse.test_bug_6139Fails with ATLAS, see gh-7877r  c                    r5  r   r*  r  r   r+  r   r   r  T  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 sFw   Y  t|d
d d S )Nr    r  zMatrix is exactly singularr  c                    r   r   r   r   r   r   r   r   d  r   zLTestLinprogIPSparse.test_magic_square_sparse_no_presolve.<locals>.<dictcomp>Fr   r   r  r   )rP   r   r  rn  r   r   r
   r   r	   re   rf   r   r2   )	r   rg   rh   rO   r  rk   ro  r   r   r   r   r   $test_magic_square_sparse_no_presolveY  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 sKw   Y  d S )Nr    rm  zInvalid permc_spec optionc                    r   r   r   r   r   r   r   r   q  r   zATestLinprogIPSparse.test_sparse_solve_options.<locals>.<dictcomp>)NATURALMMD_ATAMMD_AT_PLUS_ACOLAMDr   
permc_specr   r  r   )rP   r   rn  r
   r   r	   re   rf   rk   r   r2   )
r   rg   rh   rO   r  ro  r   permc_specsrQ  r   r   r   r   test_sparse_solve_optionsk  s   "z-TestLinprogIPSparse.test_sparse_solve_options)r  r  r  r   r  r  xfail_on_32bitr  r=  r  rL  rS  r/  r   r   r+  r   rE  K  s    
rE  c                       sT   e Zd ZdddZejd fddZdd Zejj	dd	 fd
dZ
  ZS )TestLinprogIPSparsePresolveT)r   r   rH  c                    r5  r   rI  r   r+  r   r   r    r6  z)TestLinprogIPSparsePresolve.test_bug_6139c                 C   r  )Nz6_sparse_presolve=True incompatible with presolve=Falser  r   r   r   r   r    r  zBTestLinprogIPSparsePresolve.test_enzo_example_c_with_infeasibilityrJ  r  c                    r5  r   rK  r   r+  r   r   r    r&  z)TestLinprogIPSparsePresolve.test_bug_6690)r  r  r  r   r  r  rT  r  r  r=  r  r/  r   r   r+  r   rU  }  s    

rU  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 )NTrC  FrF  r^   r   r   z(linprog default solver unexpected resultgV瞯<r&   )has_cholmodr  r\   r	   r   r   r-   )r   r   rM   rN   rO   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   rV  FT)r   r   r  r   )r	   r   r2   r  r   r   r   )test_unbounded_below_no_presolve_original  s   z?TestLinprogIPSpecific.test_unbounded_below_no_presolve_originalc                 C   r   )Nr^   r   Tr   r   r   r   r   r   r   r   test_cholesky  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 )
Nr^   r~  r  r  T)ipr   r   r   r   )	r\   r   rn  r  r
   r   r	   r   r2   )r   rM   rN   rO   ro  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   ru   rq   r  r  r   Fr   r   )	r   rn  r  r
   r	   rk   r   r   r   r  r   r   r   r    s   

z#TestLinprogIPSpecific.test_bug_8664N)	r  r  r  r   rX  rY  rZ  r\  r  r   r   r   r   rV    s    	rV  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   rl    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   ry   r	   rk   r   r   r2   r   r   rq  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)r|   r|   r}   r|   r`  r   r   ry   r	   r   r   r2   r   r   
r   rO   re   rf   rg   rh   rw   rx   rk   r   r   r   r   0test_nontrivial_problem_with_unbounded_variables     
zDTestLinprogRSCommon.test_nontrivial_problem_with_unbounded_variablesc           
      C   re  )N)rV  r   Nr}   r  g333333?r`  r   r   rf  rg  r   r   r   .test_nontrivial_problem_with_bounded_variables  ri  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$@)r|   rj  r}   rk  r`  r   r   )	ry   r6   r@   r	   r   r   r2   r   r   rg  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    r`   r`  rp   ry   r	   rk   r   r   r   r   
r   rO   re   rf   rg   rh   rw   rx   	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 sOw   Y  t|dd t|jd t|	 t|	j|jk  d S )	Nr    rm  rt  )rg   rh   r   )rg   rh   r   ra  r  r   r   )rP   r6   r7   rC   rD   r   rn  r
   r  r   r	   r   r.   r2   r   r   r   )
r   rM   rN   rO   r  r|  ro  r   r  res3r   r   r   %test_redundant_constraints_with_guess  s   
z9TestLinprogRSCommon.test_redundant_constraints_with_guessN)r  r  r  r   rl  rc  rh  rl  rm  rs  rv  r   r   r   r   r]    s    r]  c                   @   r>  )TestLinprogRSBlandpivotrj  Nr@  r   r   r   r   rw    rA  rw  c                   @   s   e Zd ZdZi Zdd ZdS )TestLinprogHiGHSSimplexDualzhighs-dsc                 C   s   t  \}}}}t||||| j| jd}t|jd td|jv  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 )	zAThe scaled model should be optimal but unscaled model infeasible.)re   rf   rk   r   r   r#   z5An optimal solution to the scaled model was found butNgưc                 S   s"   g | ]\}}|d u rt jn|qS r   r6   ra   r   lur   r   r   
<listcomp>)  s   " zCTestLinprogHiGHSSimplexDual.test_lad_regression.<locals>.<listcomp>c                 S   s$   g | ]\}}|d u rt j n|qS r   rz  r{  r   r   r   r~  *  s   $ )r   r	   r   r   r   r   r   r+   r.   r6   allr   )r   rO   re   rf   r   r   r   r   r   test_lad_regression  s   
"z/TestLinprogHiGHSSimplexDual.test_lad_regressionN)r  r  r  r   r   r  r   r   r   r   ry    s    ry  c                   @   s   e Zd ZdZi ZdS )TestLinprogHiGHSIPMr  N)r  r  r  r   r   r   r   r   r   r  2  s    r  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                   @   s   e Zd ZdZdd ZdS )TestAutoscaleIPr   c                 C   s   d| j d< t| S )Nr  r  )r   r  r  r   r   r   r   r  G  s   

zTestAutoscaleIP.test_bug_6139N)r  r  r  r   r  r   r   r   r   r  D  s    r  c                   @   r  )TestAutoscaleSimplexr   Nr  r   r   r   r   r  L  r  r  c                   @   s    e Zd ZdZdd Zdd ZdS )TestAutoscaleRSr   c           	      C   r^  r_  rb  rq  r   r   r   rc  S  rd  z2TestAutoscaleRS.test_nontrivial_problem_with_guessc           
      C   rn  ro  rp  rq  r   r   r   rs  Z  rt  z6TestAutoscaleRS.test_nontrivial_problem_with_bad_guessN)r  r  r  r   rc  rs  r   r   r   r   r  P  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   LCTrp  test_RR_infeasibilityr  r  r  	test_NFLCr  r   r   r   r   r  g  s    
r  c                   @   r>  )	TestRRSVDr   SVDNr@  r   r   r   r   r  r  rA  r  c                   @   r>  )TestRRPivotr   rx  Nr@  r   r   r   r   r  v  rA  r  c                   @   r>  )TestRRIDr   IDNr@  r   r   r   r   r  z  rA  r  )NNr%   r%   )r   )r   rm   rq   rz   )Nr  r  numpyr6   numpy.testingr   r   r   r   r   r   r  r   r   scipy.optimizer	   r
   scipy.optimize._numdiffr   scipy.sparse.linalgr   scipy.linalgr   scipy.sparser   r  scikits.umfpackr   ImportErrorrW  sksparsesksparse.cholmodr   cholmodr   r   r!   r$   r2   rP   r\   rl   ry   r   r   r   r   re   rf   rg   rh   rk   r   r  r  r  r  r"  r0  r2  r?  rB  rD  rE  rU  rV  r]  rw  ry  r  r  r  r  r  r  r  r  r  r   r   r   r   <module>   s     

B

#
          %u-
2D?	
