
    Џkh                     d   d dl Zd dlmZmZ d dlmZmZmZ d dl	Z	d Z
d Ze	j                  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                  j                  dd      d        Zy)    N)assert_array_almost_equalassert_)
csr_matrix
csc_matrix
lil_matrixc                  t   d} t         j                  j                  d       t         j                  j                  | | f      }d||dkD  <   t        |      }t	        |       D ]U  }|||dz   d d f   }|j                  |      }t        ||j                                t        t        |      t        u        W y N
   r   gffffff?   )nprandomseedr   rangegetrowr   toarrayr   typer   )NXXcsciarr_rowcsc_rows         W/var/www/teggl/fontify/venv/lib/python3.12/site-packages/scipy/sparse/tests/test_csc.pytest_csc_getrowr      s    
AIINN1
		!Q AAa#gJa=D1X -Aa!eGQJ-++a.!'7??+<=W+,-    c                  t   d} t         j                  j                  d       t         j                  j                  | | f      }d||dkD  <   t        |      }t	        |       D ]U  }|d d ||dz   f   }|j                  |      }t        ||j                                t        t        |      t        u        W y r	   )
r   r   r   r   r   getcolr   r   r   r   )r   r   r   r   arr_colcsc_cols         r   test_csc_getcolr       s    
AIINN1
		!Q AAa#gJa=D1X -Aq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                 4   | j                         j                  d   dz
  }|}|dz
  }|dk(  rG| ||d d f   j                         j                  }| ||d d f   j                         j                  }n|dk(  rG| d d ||f   j                         j                  }| d d ||f   j                         j                  }nK|dk(  rF| ||||f   j                         j                  }| ||||f   j                         j                  }|k(  sJ |k(  sJ y )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_slicesr/   %   s5   & ""$**1-1GGkGqy%ggoq&89AACII%ggoq&89AACII	%a&89AACII%a&89AACII	%ggoww&FGOOQWW%ggoww&FGOOQWW^+++^+++r   ax)r   r   Nc                 L   d}t        |      }d|d<   d|d<   t        |      }|j                  |       }| ||d   z  }||d   z  }nNt        j                  |      |j
                  k(  sJ t        j                  |      t        j                  |      }}|||f   |d   k(  sJ y )N)順 r4   *   )r1   r1   gSt$?)r6   )r(   r   )r   r   argmaxr   count_nonzeronnzmax)r0   dimAidxiijjs         r   test_argmax_overflowr@   J   s     C3AAfIAfI1A
(((
C	z
 3q6\CF]
 $---biinBRV9&	!!!r   )numpyr   numpy.testingr   r   scipy.sparser   r   r   pytestr   r    markparametrizer/   r@    r   r   <module>rH      s    < ; ; -- =1a&AA   1a&AA   1a&AA  f $""$ % ",#",( 34" 5"r   