o
    i                     @   s   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d Z
d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d Ze	jdddd ZdS )    N)assert_array_almost_equalassert_)
csr_matrix
csc_matrix
lil_matrixc                  C   s   d} t jd t j| | f}d||dk< t|}t| D ]"}|||d d d f }||}t||  tt	|t
u  qd S N
   r   gffffff?   )nprandomseedr   rangegetrowr   toarrayr   typer   )NXXcsciarr_rowcsc_row r   X/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/scipy/sparse/tests/test_csc.pytest_csc_getrow      
r   c                  C   s   d} t jd t j| | f}d||dk< t|}t| D ]"}|d d ||d f }||}t||  tt	|tu  qd S r   )
r
   r   r   r   r   getcolr   r   r   r   )r   r   r   r   arr_colcsc_colr   r   r   test_csc_getcol   r   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                 C   s   |   jd d }|}|d }|dkr.| ||d d f   j}| ||d d f   j}n=|dkrM| d d ||f   j}| d d ||f   j}n|dkrk| ||||f   j}| ||||f   j}||ksqJ ||kswJ d S )Nr   r	   r!   )r   shape)matrix_inputaxisexpected_shapeslice_1slice_2slice_3actual_shape_1actual_shape_2r   r   r   test_csc_empty_slices%   s   r,   ax)r   r	   Nc                 C   s   d}t |}d|d< d|d< t|}|j| d}| d u r)||d  }||d  }nt||jks3J t|t|}}|||f |d ksJJ d S )N)順 r0   *   )r.   r.   gSt$?)r2   )r%   r   )r   r   argmaxr
   count_nonzeronnzmax)r-   dimAidxiijjr   r   r   test_argmax_overflowJ   s   r<   )numpyr
   numpy.testingr   r   scipy.sparser   r   r   pytestr   r   markparametrizer,   r<   r   r   r   r   <module>   sH    
