관리-도구
편집 파일: test_helper.cpython-311.pyc
� �܋f � � � d Z ddlZddlmZ ddlmZmZ G d� d� � Z G d� d� � Z G d � d � � Z G d� d� � Z dS ) 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� ZdS )�TestFFTShiftc �6 � g d�}g d�}t t j |� � |� � t t j |� � |� � g d�}g d�}t t j |� � |� � t t j |� � |� � 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 r r r r �r r �fftshift� ifftshift)�self�x�ys �n/builddir/build/BUILD/cloudlinux-venv-1.0.6/venv/lib64/python3.11/site-packages/numpy/fft/tests/test_helper.py�test_definitionzTestFFTShift.test_definition s� � �+�+�+��+�+�+��!�#�,�q�/�/�1�5�5�5�!�#�-��"2�"2�A�6�6�6�/�/�/��/�/�/��!�#�,�q�/�/�1�5�5�5�!�#�-��"2�"2�A�6�6�6�6�6� c � � dD ]V}t j � |f� � }t t j t j |� � � � |� � �Wd S )N)r r � �d �� )�np�randomr r r r )r �nr s r �test_inversezTestFFTShift.test_inverse s[ � �$� I� I�A�� � � �!��&�&�A�%�c�m�C�L��O�O�&D�&D�a�H�H�H�H� I� 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 |� � |� � d S )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_keywordzTestFFTShift.test_axes_keyword s � ����J�J�J����5���<�<����J�J�J�7��!�#�,�u�6�"B�"B�"B�G�L�L�L�!�#�,�u�1�"=�"=�"=�"%�,�u�4�"@�"@�"@� B� B� B�!�#�-��f�"E�"E�"E�u�M�M�M�!�#�-��a�"@�"@�"@�"%�-��d�"C�"C�"C� E� E� E� "�#�,�u�"5�"5�w�?�?�?�!�#�-��"8�"8�%�@�@�@�@�@r c �P � 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 |� � |� � d S )z1 Test 2D input, which has uneven dimension sizes r r r r r � r'