| Server IP : 209.209.40.120 / Your IP : 216.73.217.112 Web Server : Microsoft-IIS/10.0 System : Windows NT NEWWWW 10.0 build 17763 (Windows Server 2019) i586 User : NEWWWW$ ( 0) PHP Version : 8.3.30 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : OFF | Perl : OFF | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : C:/Python312/Lib/site-packages/numpy/fft/tests/__pycache__/ |
Upload File : |
�
�g� � �| � d Z ddlZddlmZ ddlmZmZ G d� d� Z G d� d� Z G d � d
� Z G d � d � Z
y)
zfTest functions for fftpack.helper module
Copied from fftpack.helper by Pearu Peterson, October 2005
� N)�assert_array_almost_equal)�fft�pic �* � e Zd Zd� Zd� Zd� Zd� Zd� Zy)� TestFFTShiftc � � g d�}g d�}t t j |� |� t t j |� |� g d�}g d�}t t j |� |� t t j |� |� y )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 r
r r r
�r r �fftshift� ifftshift)�self�x�ys �=C:\Python312\Lib\site-packages\numpy/fft/tests/test_helper.py�test_definitionz TestFFTShift.test_definition
s` � � +�� +��!�#�,�,�q�/�1�5�!�#�-�-��"2�A�6� /�� /��!�#�,�,�q�/�1�5�!�#�-�-��"2�A�6� c � � dD ]T }t j j |f� }t t j t j
|� � |� �V y )N)r
r
� �d �� )�np�randomr r r r )r �nr s r � test_inversezTestFFTShift.test_inverse sD � �$� I�A�� � � � �!��&�A� %�c�m�m�C�L�L��O�&D�a� H� Ir c � � g d�g d�g d�g}g d�g d�g d�g}t t j |d�� |� t t j |d �� t j |d
�� � t t j |d�� |� t t j |d �� t j |d
�� � t t j |� |� t t j |� |� y ) N)r r
r )r r
r )r r r )r r r )r r r
)r r r
�r r
��axesr �r r )r �freqs�shifteds r �test_axes_keywordz TestFFTShift.test_axes_keyword s� � ��J� �5�� ��J�7��!�#�,�,�u�6�"B�G�L�!�#�,�,�u�1�"=�"%�,�,�u�4�"@� B�!�#�-�-��f�"E�u�M�!�#�-�-��a�"@�"%�-�-��d�"C� E� "�#�,�,�u�"5�w�?�!�#�-�-��"8�%�@r c � � ddgddgddgg}ddgddgddgg}t t j |d�� |� t t j |d�� |� t t j |d�� |� t t j |dg�� |� ddgddgddgg}t t j |d�� |� t t j |d�� |� ddgddgddgg}t t j |d �� |� t t j |d �� |� t t j |ddg�� |� t t j |ddg�� |� t t j |d
�� |� t t j |d
�� |� t t j |� |� t t j |� |� y
) z1 Test 2D input, which has uneven dimension sizes r r
r r r
� r' r) r& Nr )r r* �
shift_dim0�
shift_dim1�shift_dim_boths r �test_uneven_dimsz TestFFTShift.test_uneven_dims) s� � � ��F�
��F�
��F�
�� ��F�
��F�
��F�
�
�
"�#�,�,�u�1�"=�z�J�!�#�-�-�
��"C�U�K�!�#�,�,�u�4�"@�*�M�!�#�-�-�
�!��"E�u�M� ��F�
��F�
��F�
�
�
"�#�,�,�u�1�"=�z�J�!�#�-�-�
��"C�U�K� ��F�
��F�
��F�
��
"�#�,�,�u�6�"B�N�S�!�#�-�-��V�"L�e�T�!�#�,�,�u�A�q�6�"B�N�S�!�#�-�-��a��V�"L�e�T� "�#�,�,�u�4�"@�.�Q�!�#�-�-��T�"J�E�R�!�#�,�,�u�"5�~�F�!�#�-�-��"?��Gr c
�r ��� �
� ddl m�m� m�m�
d��� �
fd� }d��� �
fd� }t d� D ]� }t d� D ]w }dD ]p }t j j ||� }t t j ||� |||� � t t j ||� |||� � �r �y �� y) z\ Test that the new (>=v1.15) implementation (see #10073) is equal to the original (<=v1.14) r )�asarray� concatenate�arange�takeNc � �� �
| � }|j }|�t t |� � }nt |t � r|f}|}|D ]: }|j
| }|dz dz } � � ||� � |� f� } � |||� }�< |S )z& How fftshift was implemented in v1.14r
r ��ndim�list�range�
isinstance�int�shape�
r r( �tmpr: r �kr# �p2�mylistr6 r4 r5 r7 s
����r �original_fftshiftz>TestFFTShift.test_equal_to_original.<locals>.original_fftshiftZ s� �� ��!�*�C��8�8�D��|��E�$�K�(���D�#�&� �w���A��
'���I�I�a�L���!�e��\��$�f�R��m�V�B�Z�%@�A����F�A�&��
'�
�Hr c � �� �
| � }|j }|�t t |� � }nt |t � r|f}|}|D ]= }|j
| }||dz dz z
} � � ||� � |� f� } � |||� }�? |S )z( How ifftshift was implemented in v1.14 r
r r9 r@ s
����r �original_ifftshiftz?TestFFTShift.test_equal_to_original.<locals>.original_ifftshiftj s� �� ��!�*�C��8�8�D��|��E�$�K�(���D�#�&� �w���A��
'���I�I�a�L���!�a�%�A� �%��$�f�R��m�V�B�Z�%@�A����F�A�&��
'�
�Hr � )r r
Nr) r& )N)
� numpy._corer4 r5 r6 r7 r< r! r" �randr r r r ) r rE rG �i�j� axes_keyword�inpr6 r4 r5 r7 s @@@@r �test_equal_to_originalz#TestFFTShift.test_equal_to_originalV s� �� �B�B� � � � �$ �r�� U�A��2�Y�
U��$>� U�L� �)�)�.�.��A�.�C�-�c�l�l�3� �.M�.?��\�.R�T� .�c�m�m�C� �.N�.@��l�.S�U�
U�
U� Ur N)�__name__�
__module__� __qualname__r r$ r, r2 rO � r r r r s � �7�I�
A�+H�Z/Ur r c � � e Zd Zd� Zy)� TestFFTFreqc �T � g d�}t dt j d� z |� t dt z t j dt � z |� g d�}t dt j d� z |� t dt z t j dt � z |� y )Nr r r �
)r r �fftfreqr �r r s r r zTestFFTFreq.test_definition� sv � � +��!�!�C�K�K��N�"2�A�6�!�!�B�$�s�{�{�1�b�'9�"9�1�=� /��!�"�S�[�[��_�"4�a�8�!�"�R�%�� � �B��(;�";�Q�?r N�rP rQ rR r rS r r rU rU � s � �@r rU c � � e Zd Zd� Zy)� TestRFFTFreqc �T � g d�}t dt j d� z |� t dt z t j dt � z |� g d�}t dt j d� z |� t dt z t j dt � z |� y )N)r r
r r r
r )r r
r r r
r. rW )r r �rfftfreqr rY s r r z TestRFFTFreq.test_definition� sw � � ��!�!�C�L�L��O�"3�Q�7�!�!�B�$�s�|�|�A�r�':�":�A�>� ��!�"�S�\�\�"�%5�"5�q�9�!�"�R�%�� � �R��(<�"<�a�@r NrZ rS r r r\ r\ � s � �Ar r\ c � � e Zd Zd� Zy)�
TestIRFFTNc � � t j j d� \ }}|d|z z }d}t j ||�� y )N)r rH � � y �?)r r' )r! r" r �irfftn)r �ar�ai�ar( s r �test_not_last_axis_successz%TestIRFFTN.test_not_last_axis_success� s< � ����!�!�.�1���B� ��B��J���� �
�
�1�4� r N)rP rQ rR rh rS r r r` r` � s � �!r r` ) �__doc__�numpyr! �
numpy.testingr r r r rU r\ r` rS r r �<module>rl sF ���
� 3� �zU� zU�z@� @�A� A� !� !r