← Back
Editing: test_csc.cpython-311.pyc
� d�cV � � � d dl Zd dlmZmZ d dlmZmZmZ d dl Z d� Z d� Z e j � d edd gd d gd dgg� � d d f edd gd d gd dgg� � dd f edd gd d gd dgg� � d d f eg d �g d�g d�g� � d dfg� � d� � � Ze j � dd� � d� � � ZdS )� N)�assert_array_almost_equal�assert_)� csr_matrix� csc_matrix� lil_matrixc � � d} t j � d� � t j � | | f� � }d||dk < t |� � }t | � � D ]m}|||dz �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 �=/usr/lib/python3/dist-packages/scipy/sparse/tests/test_csc.py�test_csc_getrowr s� � � �A��I�N�N�1���� � ���!�Q�� � �A��A�a�#�g�J� �a�=�=�D� �1�X�X� -� -���A�a�!�e�G�Q�Q�Q�J�-���+�+�a�.�.��!�'�7�?�?�+<�+<�=�=�=���W� � ��+�,�,�,�,� -� -� c � � d} t j � d� � t j � | | f� � }d||dk < t |� � }t | � � D ]m}|d d �||dz �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��I�N�N�1���� � ���!�Q�� � �A��A�a�#�g�J� �a�=�=�D� �1�X�X� -� -���A�A�A�q��Q��w�J�-���+�+�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 j d dz }|}|dz }|dk r1| ||�d d �f j j }| ||�d d �f j j }nm|dk r1| d d �||�f j j }| d d �||�f j j }n6|dk r0| ||�||�f j j }| ||�||�f j j }||k sJ �||k sJ �d S )Nr r r# )�A�shape)� matrix_input�axis�expected_shape�slice_1�slice_2�slice_3�actual_shape_1�actual_shape_2s r �test_csc_empty_slicesr0 % s � �& �n�"�1�%��)�G��G���k�G� �q�y�y�%�g�g�o�q�q�q�&8�9�;�A��%�g�g�o�q�q�q�&8�9�;�A��� ����%�a�a�a��� �&8�9�;�A��%�a�a�a��� �&8�9�;�A��� �� � �%�g�g�o�w�w��&F�G�I�O��%�g�g�o�w�w��&F�G�I�O�� �^� +� +� +� +� �^� +� +� +� +� +� +r �ax)������r r Nc �r � d}t |� � }d|d< d|d< t |� � }|� | �� � }| �||d z }||d z }nGt j |� � |j k sJ �t j |� � t j |� � }}|||f |d k sJ �d S )N)順 r5 �* )r2 r2 g�St$��?)���r7 )r) r )r r �argmaxr � count_nonzero�nnz�max)r1 �dimr& �idx�ii�jjs r �test_argmax_overflowr@ J s� � � �C��3���A��A�f�I��A�f�I��1� � �A� �(�(��(� � �C� �z� �3�q�6�\�� �C��F�]��� � ��$�$���-�-�-�-������b�i��n�n�B�� �R��V�9��&� � !� !� !� !� !� !r )�numpyr � numpy.testingr r � scipy.sparser r r �pytestr r �mark� parametrizer0 r@ � r r �<module>rH s� �� � � � � <� <� <� <� <� <� <� <� ;� ;� ;� ;� ;� ;� ;� ;� ;� ;� � � � � -� -� -� -� -� -� ����=��j�1�a�&��A���A��� � � ��� �j�1�a�&��A���A��� � � ��� �j�1�a�&��A���A��� � � �f�� �j�$�$�$�"�"�"�"�"�"�$� %� %� ����� �",� ,�#� �",�( ����� 3�4�4�"� "� 5�4�"� "� "r
Save File
Cancel