U
    ?h                     @   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   ddddddddd	g	}dddd	dddddg	}t t|| t t|| dddddd
dddd	g
}d
dddd	dddddg
}t t|| t t|| d S )Nr               )r   r   r   )selfxy r   Q/var/www/html/venv/lib/python3.8/site-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                   @   s   e Zd Zdd ZdS )TestFFTFreqc              
   C   s   ddddddddd	g	}t d
td
 | t d
t td
t | dddddddddd	g
}t dtd | t dt tdt | d S )Nr   r
   r   r   r   r   r   r   r   r   r   
   )r   r   r   r   r   r   r   r   r   %   s    zTestFFTFreq.test_definitionNr   r   r    r   r   r   r   r   r!   #   s   r!   c                   @   s   e Zd Zdd ZdS )TestRFFTFreqc              
   C   s   dddddddddg	}t dtd | t dt tdt | ddddddddddg
}t dtd | t dt tdt | d S )	Nr   r
   r   r   r   r      r"   )r   r   r   r#   r   r   r   r   0   s    zTestRFFTFreq.test_definitionNr$   r   r   r   r   r%   .   s   r%   N)Z	__usage__Znumpy.testingr   Zscipy.fftpackr   r   r   r   numpyr   r   r	   r!   r%   r   r   r   r   <module>   s   	