o
    Eb                     @   sf   d Z ddlmZ ddlmZmZmZmZ ddlm	Z	m
Z
 G dd dZG dd dZG d	d
 d
ZdS )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                   @   s   e Zd Zdd Zdd ZdS )TestFFTShiftc                 C   s\   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y r   A/usr/lib/python3/dist-packages/scipy/fftpack/tests/test_helper.pytest_definition   s   zTestFFTShift.test_definitionc                 C   s,   dD ]}t  |f}ttt|| qd S )N)r   r   	   d      )r   r   r   r   )r   nr   r   r   r   test_inverse   s   zTestFFTShift.test_inverseN)__name__
__module____qualname__r   r   r   r   r   r   r	      s    
r	   c                   @      e Zd Zdd ZdS )TestFFTFreqc                 C   h   g d}t dtd | t dt tdt | g d}t dtd | t dt tdt | d S )Nr
   r   r   
   )r   r   r   r   r   r   r   r   r   %      zTestFFTFreq.test_definitionNr    r!   r"   r   r   r   r   r   r$   #       r$   c                   @   r#   )TestRFFTFreqc                 C   r%   )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   0   r(   zTestRFFTFreq.test_definitionNr)   r   r   r   r   r+   .   r*   r+   N)Z	__usage__Znumpy.testingr   Zscipy.fftpackr   r   r   r   Znumpyr   r   r	   r$   r+   r   r   r   r   <module>   s   	