← Back
Editing: test_array_tools.cpython-311.pyc
� d�c0 � �` � d dl Zd dlmZ d dlmZ d dlmZm Z m Z m Z m Z m Z G d� d� � ZdS )� N)�assert_array_equal)�raises)� axis_slice� axis_reverse�odd_ext�even_ext� const_ext�zero_extc �2 � e Zd Zd� Zd� Zd� Zd� Zd� Zd� ZdS )�TestArrayToolsc � � t j d� � � dd� � }t |ddd�� � }t ||dd�d d �f � � t |dd�� � }t ||dd �d d �f � � t |ddd�� � }t ||d d �dd�f � � t |dd�� � }t ||d d �dd �f � � t |dd d� � � }t ||d d d �d d �f � � t |dd d� � � }t ||d d �d d d �f � � d S ) N� � � r � )�start�stop�axis���)r r � )r �stepr )�np�arange�reshaper r )�self�a�ss �E/usr/lib/python3/dist-packages/scipy/signal/tests/test_array_tools.py�test_axis_slicez TestArrayTools.test_axis_slice si � � �I�b�M�M� !� !�!�Q� '� '�� �q���� 2� 2� 2���1�a��!��Q�Q�Q��i�(�(�(� �q��� +� +� +���1�a����Q�Q�Q��i�(�(�(� �q���� 2� 2� 2���1�a����1�Q�3��i�(�(�(� �q��� +� +� +���1�a����2�3�3��i�(�(�(� �q���� 2� 2� 2���1�a���!��Q�Q�Q��i�(�(�(� �q���� 2� 2� 2���1�a����3�3�Q�3��i�(�(�(�(�(� c � � t j d� � � dd� � }t |d�� � }t ||d d d�d d �f � � t |d�� � }t ||d d �d d d�f � � d S )Nr r r r �r r r )r r r r r )r r �rs r �test_axis_reversez TestArrayTools.test_axis_reverse! s� � � �I�b�M�M� !� !�!�Q� '� '�� ��� #� #� #���1�a���"��a�a�a��j�)�)�)� ��� #� #� #���1�a����4�4�R�4��j�)�)�)�)�)r c � � t j g d�g d�g� � }t |dd�� � }t j g d�g d�g� � }t ||� � t |dd�� � }t j g d �g d�g d�g d �g� � }t ||� � t t t |dd�� � t t t |d d�� � d S ) N�r r r r � �� � � � r' r r r" ) r r r r r r r' r, r+ ) � � r) r* r+ r, r' r r r )i�������r r r' )� � r- r* r' r' )r �arrayr r � assert_raises� ValueError)r r �odd�expecteds r � test_odd_extzTestArrayTools.test_odd_ext* s � � �H�o�o�o�%�o�o�'� (� (�� �a���#�#�#���8� 9� 9� 9� :� :� :� <� =� =���3��)�)�)��a���#�#�#���8� /� /� /� ,�_�_� ,�_�_� /� /� /� 1� 2� 2�� �3��)�)�)��j�'�1�a�a�8�8�8�8��j�'�1�a�a�8�8�8�8�8�8r c � � t j g d�g d�g� � }t |dd�� � }t j g d�g d�g� � }t ||� � t |dd�� � }t j g d�g d�g d�g d�g� � }t ||� � t t t |dd�� � t t t |d d�� � d S ) Nr& r( r r r" ) r r r r r r r' r r ) r+ r* r) r* r+ r, r' r, r+ r r' )r r2 r r r3 r4 )r r �evenr6 s r � test_even_extz TestArrayTools.test_even_ext= s � � �H�o�o�o�%�o�o�'� (� (�� ��1�1�%�%�%���8� 8� 8� 8� 8� 8� 8� :� ;� ;���4��*�*�*���1�1�%�%�%���8�_�_�_� ,�_�_� ,�_�_� ,�_�_� .� /� /�� �4��*�*�*��j�(�A�q�q�9�9�9�9��j�(�A�q�q�9�9�9�9�9�9r c �6 � t j g d�g d�g� � }t |dd�� � }t j g d�g d�g� � }t ||� � t |dd�� � }t j g d�g d�g d�g d�g� � }t ||� � d S ) Nr& r( r r r" ) r r r r r r r' r' r' ) r) r) r) r* r+ r, r' r' r' r )r r2 r r )r r �constr6 s r �test_const_extz TestArrayTools.test_const_extP s� � � �H�o�o�o�%�o�o�'� (� (�� �!�Q�Q�'�'�'���8� 8� 8� 8� 8� 8� 8� :� ;� ;���5�(�+�+�+��!�Q�Q�'�'�'���8�_�_�_� ,�_�_� ,�_�_� ,�_�_� .� /� /�� �5�(�+�+�+�+�+r c �6 � t j g d�g d�g� � }t |dd�� � }t j g d�g d�g� � }t ||� � t |dd�� � }t j g d �g d�g d�g d �g� � }t ||� � d S ) Nr& r( r r r" ) r r r r r r r' r r ) r r r) r* r+ r, r' r r r )r r r r r )r r2 r r )r r �zeror6 s r � test_zero_extz TestArrayTools.test_zero_ext` s� � � �H�o�o�o�%�o�o�'� (� (�� ��1�1�%�%�%���8� 8� 8� 8� 8� 8� 8� :� ;� ;���4��*�*�*���1�1�%�%�%���8�_�_�_� ,�_�_� ,�_�_� ,�_�_� .� /� /�� �4��*�*�*�*�*r N) �__name__� __module__� __qualname__r r$ r7 r: r= r@ � r r r r sn � � � � � �)� )� )�**� *� *�9� 9� 9�&:� :� :�&,� ,� ,� +� +� +� +� +r r )�numpyr � numpy.testingr �pytestr r3 �scipy.signal._arraytoolsr r r r r r r rD r r �<module>rI s� �� � � � � ,� ,� ,� ,� ,� ,� *� *� *� *� *� *�-� -� -� -� -� -� -� -� -� -� -� -� -� -� -� -�d+� d+� d+� d+� d+� d+� d+� d+� d+� d+r
Save File
Cancel