
    -ii                         d dl Z d dlZd dlZd dlZd dlmZmZ d dlm	Z	m
Z
mZmZmZ d dlZej        dk    o ej                    dk    Zd Zd Zej                            d e
d	d gd d gd d
gg          d df e
d	d gd d gd d
gg          d	df e
d	d gd d gd d
gg          ddf e
g dg dg dg          d dfg          d             Zej                            dd          d             Zej                            e d          ej                            d
          ej                            de j        e j        e j        e j        e j        e j        d d f          d                                     ZdS )    N)assert_array_almost_equalassert_)
csr_matrix
csc_matrix
lil_matrix	csr_array	csc_arraylinuxx86_64c                     d} t           j                            d           t           j                            | | f          }d||dk    <   t          |          }t	          |           D ]m}|||dz   d d f         }|                    |          }t          ||                                           t          t          |          t          u            nd S N
   r   gffffff?   )nprandomseedr   rangegetrowr   toarrayr   typer   )NXXcsciarr_rowcsc_rows         a/var/www/html/bet.cuttalo.com/ml/venv/lib/python3.11/site-packages/scipy/sparse/tests/test_csc.pytest_csc_getrowr      s    
AINN1
	!Q  AAa#gJa==D1XX - -Aa!eGQQQJ-++a..!'7??+<+<===W+,,,,- -    c                     d} t           j                            d           t           j                            | | f          }d||dk    <   t          |          }t	          |           D ]m}|d d ||dz   f         }|                    |          }t          ||                                           t          t          |          t          u            nd S r   )
r   r   r   r   r   getcolr   r   r   r   )r   r   r   r   arr_colcsc_cols         r   test_csc_getcolr$      s    
AINN1
	!Q  AAa#gJa==D1XX - -AAAqQwJ-++a..!'7??+<+<===W+,,,,- -r   z"matrix_input, axis, expected_shaper      )r   r%   )   r   both)r   r   )r   r   r   r   r   r   )r   r   r   r   r   r   )r   r   r%   r&   r   r   )r      c                 \   |                                  j        d         dz
  }|}|dz
  }|dk    rK| ||d d f                                          j        }| ||d d f                                          j        }n|dk    rK| d d ||f                                          j        }| d d ||f                                          j        }nP|dk    rJ| ||||f                                          j        }| ||||f                                          j        }||k    sJ ||k    sJ d S )Nr   r   r'   )r   shape)matrix_inputaxisexpected_shapeslice_1slice_2slice_3actual_shape_1actual_shape_2s           r   test_csc_empty_slicesr3   -   s[   & ""$$*1-1GGkGqyy%ggoqqq&89AACCI%ggoqqq&89AACCI	%aaa&89AACCI%aaa&89AACCI	%ggoww&FGOOQQW%ggoww&FGOOQQW^++++^++++++r   ax)r   r   Nc                 r   d}t          |          }d|d<   d|d<   t          |          }|                    |           }| ||d         z  }||d         z  }nGt          j        |          |j        k    sJ t          j        |          t          j        |          }}|||f         |d         k    sJ d S )N)順 r8   *   )r5   r5   gSt$?)r:   )r,   r   )r   r   argmaxr   count_nonzeronnzmax)r4   dimAidxiijjs         r   test_argmax_overflowrD   R   s     C3AAfIAfI1A
(((

C	z
 3q6\CF]
 $$----binnBRV9&	!!!!!!r   z(avoid variations due to OS, see gh-23826)reasonopc                 ,    |                      |          S N)minimumxys     r   <lambda>rM   r   s    QYYq\\ r   c                 ,    |                      |          S rH   )maximumrJ   s     r   rM   rM   r   s    		RS r   c                    t          j        t           j                  j        dz   }t          j        |dz
  g          }t          j        dg          }||f}||f}t          t          j        dg          |f|df          }|                                } | ||           t          t          j        dg          |fd|f          }|                                }	 | |	|           d S )Nr   r      )r*   )r   iinfouint32r>   arrayr	   copyr   )
rF   num_minor_axisminor_axis_indexmajor_axis_indexrow_colscol_rowsr   X_2ZZ_2s
             r   #test_compressed_rc_conversion_mixupr^   m   s     Xbi((,q0Nx!!3 455x~~ "23H "23H28RD>>8,^R4HIIIA
&&((CBsAJJJ28RD>>8,R4HIIIA
&&((CBsAJJJJJr   ) operatorsysplatformnumpyr   numpy.testingr   r   scipy.sparser   r   r   r   r	   pytestmachineLINUX_INTELr   r$   markparametrizer3   rD   skipiftimeoutneltgtaddsubmulr^    r   r   <module>rs      s    



      < < < < < < < < Q Q Q Q Q Q Q Q Q Q Q Q Q Q  |w&L-=X-=-?-?8-K- - -- - - =j1a&AA    j1a&AA    j1a&AA   f j$$$""""""$ % %  ", ,# ",( 344" " 54"4 O,VWWQX[(+ (hlHL 9 9;T;T V W W 	W W  XW  r   