← Back
Editing: test_extract.cpython-311.pyc
� d�c! � �P � d Z ddlmZ ddlmZ ddlZddlmZ G d� d� � ZdS )z)test sparse matrix construction functions� )� assert_equal)� csr_matrixN)�_extractc �& � e Zd Zd� Zd� Zd� Zd� ZdS )� TestExtractc �� � t ddgg� � t ddgg� � t ddgg� � t dgdgg� � t dgdgg� � t dgdgg� � t ddgddgg� � t ddgddgg� � t ddgddgg� � t ddgddgg� � t g d�g d�g d�g� � t g d�g d�g d�g� � j g | _ d S ) N� � r � � )r r r r r )r � r � � )r r � � r )r �T�cases)�selfs �A/usr/lib/python3/dist-packages/scipy/sparse/tests/test_extract.py� setup_methodzTestExtract.setup_method s � � ��1��w� � � ��1��w� � � ��1��w� � � ���Q�C�y� !� !� ���Q�C�y� !� !� ���Q�C�y� !� !� ��1��q��e�}� %� %� ��1��q��e�}� %� %� ��1��q��e�}� %� %� ��1��q��e�}� %� %� � � � �K�K�K� � � �<� =� =� � � � �K�K�K� � � �<� =� =� ?� �� � � � c � � | j D ]T}t j |� � \ }}}t |� � � t ||f|f|j �� � � � �Ud S )N)�shape)r r �findr �toarrayr r )r �A�I�J�Vs r r zTestExtract.find sh � ��� L� L�A� �M�!�$�$�E�A�a�� ������j�1�Q�%���!�'�&J�&J�&J� K� K� K� K� L� Lr c �� � | j D ]e}|� � � }dD ]L}t t j ||�� � � � � t j ||�� � � � �M�fd S �N)���������r r r r )�k)r r r r �tril�np�r r �Br% s r � test_trilzTestExtract.test_tril � � ��� M� M�A�� � � � �A�'� M� M�� �X�]�1�q� 1� 1� 1� 9� 9� ;� ;�R�W�Q��^�^�^�L�L�L�L� M� M� Mr c �� � | j D ]e}|� � � }dD ]L}t t j ||�� � � � � t j ||�� � � � �M�fd S r! )r r r r �triur' r( s r � test_triuzTestExtract.test_triu& r+ r N)�__name__� __module__� __qualname__r r r* r. � r r r r s[ � � � � � � � � � L� L� L� M� M� M� M� M� M� M� Mr r ) �__doc__� numpy.testingr � scipy.sparser �numpyr' r r r2 r r �<module>r7 s� �� /� /� &� &� &� &� &� &� #� #� #� #� #� #� � � � � !� !� !� !� !� !� M� M� M� M� M� M� M� M� M� Mr
Save File
Cancel