← Back
Editing: test_helper.cpython-311.pyc
� d�c� � � � d Z ddlmZ ddlmZmZmZmZ ddlm Z m Z G d� d� � Z G d� d� � Z G d � d � � Z dS )z� Build fftpack: python setup_fftpack.py build Run tests if scipy is installed: python -c 'import scipy;scipy.fftpack.test(<level>)' Run tests if fftpack is not installed: python tests/test_helper.py [<level>] � )�assert_array_almost_equal)�fftshift� ifftshift�fftfreq�rfftfreq)�pi�randomc � � e Zd Zd� Zd� ZdS )�TestFFTShiftc � � g d�}g d�}t t |� � |� � t t |� � |� � g d�}g d�}t t |� � |� � t t |� � |� � 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 r )�self�x�ys �A/usr/lib/python3/dist-packages/scipy/fftpack/tests/test_helper.py�test_definitionzTestFFTShift.test_definition s� � �#�#�#��#�#�#��!�(�1�+�+�a�0�0�0�!�)�A�,�,�q�1�1�1�&�&�&��&�&�&��!�(�1�+�+�a�0�0�0�!�)�A�,�,�q�1�1�1�1�1� c � � dD ]A}t j |f� � }t t t |� � � � |� � �Bd S )N)r r � �d �� )r r r r )r �nr s r �test_inversezTestFFTShift.test_inverse sQ � � � @� @�A�� �q�d�#�#�A�%�i�����&<�&<�Q�?�?�?�?� @� @r N)�__name__� __module__�__qualname__r r# � r r r r s7 � � � � � �2� 2� 2�@� @� @� @� @r r c � � e Zd Zd� ZdS )�TestFFTFreqc �N � g d�}t dt d� � z |� � t dt z t dt � � z |� � g d�}t dt d� � z |� � t dt z t dt � � z |� � d S )Nr r r � )r r r �r r s r r zTestFFTFreq.test_definition% s� � �#�#�#��!�!�G�A�J�J�,�q�1�1�1�!�!�B�$�w�q��}�}�"4�Q�7�7�7�&�&�&��!�"�W�R�[�[�.��3�3�3�!�"�R�%���2���"6�q�9�9�9�9�9r N�r$ r% r&