← Back
Editing: test_csr.cpython-311.pyc
� d�c � �R � d dl Zd dlmZmZ d dlmZmZ d dlZd� Z d� Z d� Z d� Z ej � d eg d �g d �g d �g� � d d f eg d �g d �g d �g� � d df eg d �g d �g d �g� � ddf eg d�g d�g d�g� � d dfg� � d� � � Zd� Zd� ZdS )� N)�assert_array_almost_equal�assert_)� csr_matrix�hstackc �� � || |f }|| |f }t ||� � � d � � t t |� � t u � � d S )Nr )r �toarrayr �typer )�i�sl�X�Xcsr�np_slice� csr_slices �=/usr/lib/python3/dist-packages/scipy/sparse/tests/test_csr.py�_check_csr_rowslicer s\ � ���B��x�H��Q��U� �I� �h� �(9�(9�(;�(;�A�(>�?�?�?� �D��O�O�z� )�*�*�*�*�*� c � � d} t j � d� � t j � | | f� � }d||dk < t |� � }t d d d � � t d d d� � t ddd� � t ddd� � g}t | � � D ]}|D ]}t ||||� � ��d S )N� r �ffffff�?���� ���� )�np�random�seedr �slice�ranger )�Nr r �slicesr r s r �test_csr_rowslicer! s� � � �A��I�N�N�1���� � ���!�Q�� � �A��A�a�#�g�J� �a�=�=�D��D�$��%�%��D�$��#�#��A�r�1�o�o��B��2� � � �F� �1�X�X� 0� 0��� 0� 0�B� ��2�q�$� /� /� /� /� 0�0� 0r c � � d} t j � d� � t j � | | f� � }d||dk < t |� � }t | � � D ]m}|||dz �d d �f }|� |� � }t ||� � � � � t t |� � t u � � �nd S �Nr r r r ) r r r r r �getrowr r r r )r r r r �arr_row�csr_rows r �test_csr_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� � ��+�,�,�,�,� -� -r 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�csr_cols r �test_csr_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_shape)r r r r )r r r r )r r � r )r � r )r- r �both)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_csr_empty_slicesr< < 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 c �` � t g d�g d�g d�g� � } g d�}t j |� � }g d�g d�g d�g}t j |� � }g d�g d�f}t j |d � � t j |d � � f}| | � � � }| | � � � }| | } | | } | | } | | } ||k � � � sJ �| | k � � � sJ �| | k � � � sJ �d S )N)r r r )r- r. r0 )� � � )FTFr r )r r �arrayr �all) �data� list_indices1�array_indices1� list_indices2�array_indices2� list_indices3�array_indices3� slice_list1� slice_array1� slice_list2� slice_array2� slice_list3� slice_array3s r �test_csr_bool_indexingrP a s[ � � �y�y�y�)�)�)�Y�Y�Y�7� 8� 8�D�(�(�(�M��X�m�,�,�N�)�)�)�+?�+?�+?�AU�AU�AU�V�M��X�m�,�,�N�)�)�)�+?�+?�+?�@�M��h�}�Q� /�0�0�"�(�=��;K�2L�2L�M�N��}�%�-�-�/�/�K���'�/�/�1�1�L��}�%�K���'�L��}�%�K���'�L� �<� '� ,� ,� .� .�.�.�.� �<� '� ,� ,� .� .�.�.�.� �<� '� ,� ,� .� .�.�.�.�.�.r c �T � t j t j � � j } dg}dg}| dz }d}|dz g}|dz g}t |||ff� � }t |||ff� � }t |dz |dz � � | k sJ �|j j |j j cxk rt j k sn J �|j j |j j cxk rt j k sn J �t ||gd�� � } | j � � � ||z dz k sJ �||z dz | k sJ �| j j | j j cxk rt j k sn J �t |j � � } t |j � � } t | | gd�� � } | j | j k sJ �| j j t j k sJ �| |z dz g} t ||| ff� � }t ||gd�� � }|j j t j k sJ �|j � � � | dz k sJ �dS )z� Tests if hstack properly promotes to indices and indptr arrays to np.int64 when using np.int32 during concatenation would result in either array overflowing. g �?r r r- �csr)�formatN) r �iinfo�int32�maxr �indices�dtype�indptrr �int64r3 )� max_int32rC �row� max_indices_1� max_indices_2�col_1�col_2�X_1�X_2�X_hs� X_1_empty� X_2_empty� X_hs_empty�col_3�X_3�X_hs_32s r �test_csr_hstack_int64rj t sy � � ����"�"�&�I� �5�D� �#�C� ��M�M��M� �Q� � �E� �Q� � �E� �d�S�%�L�)� *� *�C� �d�S�%�L�)� *� *�C� �}�q� �-�!�"3� 4� 4�y� @� @� @� @� �;� �� � 0� <� <� <� <�B�H� <� <� <� <� <� <� �;� �� � 0� <� <� <� <�B�H� <� <� <� <� <� <� �3��*�U� +� +� +�D� �<� � � � � � �!>��!B� B� B� B� B� �=� (�1� ,�y� 8� 8� 8� 8� �<� ���!2� >� >� >� >�b�h� >� >� >� >� >� >� �3�9�%�%�I��3�9�%�%�I���I�.�u�=�=�=�J� � �t�z� )� )� )� )� � � #�r�x� /� /� /� /� � � &�� *� +�E� �d�S�%�L�)� *� *�C��c�3�Z��.�.�.�G� �?� �B�H� ,� ,� ,� ,� �?� � � � �I��M� 1� 1� 1� 1� 1� 1r )�numpyr � numpy.testingr r � scipy.sparser r �pytestr r! r' r, �mark� parametrizer< rP rj � r r �<module>rr s� �� � � � � <� <� <� <� <� <� <� <� +� +� +� +� +� +� +� +� � � � �+� +� +�0� 0� 0�" -� -� -� -� -� -� ����=��j�,�,�,� � � � � � � � � � ��� �j�,�,�,� � � � � � � � � � ��� �j�,�,�,� � � � � � � � � � �f�� �j�/�/�/� ��� ���!� "� "� ����� �",� ,�#� �",�(/� /� /�&52� 52� 52� 52� 52r
Save File
Cancel