3
���h�% �
@ s� d dl Zd dlZd dlmZ d dlmZmZmZ d dl Z d dl
Z
dd� Z G dd� d�Z G dd � d �Z
ejjd
ejejejejg�ejjd d d
g�ejjdejjejjejjejjejjejjg�dd� ���ZG dd� d�ZdS )� N)�random)�assert_array_equal�
assert_raises�assert_allclosec C sR t | �}dtj tj|�t|� }tj|�jdd�| }tj| tj|� dd�S )Ny @� )�axisy � ����)�len�np�piZarange�floatZreshape�sum�exp)�x�L�phase� r �?/tmp/pip-build-5_djhm0z/numpy/numpy/fft/tests/test_pocketfft.py�fft1 s r c @ s e Zd Zdd� ZdS )� TestFFTShiftc C s t ttjjdddgd� d S )Nr � � r )r �
ValueErrorr
�fft)�selfr r r �
test_fft_n s zTestFFTShift.test_fft_nN)�__name__�
__module__� __qualname__r r r r r r s r c @ s� e Zd Zdd� Zdd� Zejjdd*�dd � �Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zejjd"ejjejjejjejjg�d#d$� �Zd%d&� Z ejjd'ej ej ej ej g�d(d)� �Z!dS )+� TestFFT1Dc C s� d}t |�dt |� }t |�}xptd|�D ]b}ttjjtjj|d|� ��|d|� dd� ttjjtjj|d|� �|�|d|� dd� q,W d S )Ni y �?r r g�-���q=)�atol)r �ranger r
r �ifft�irfft�rfft)r �maxlenr Zxr�ir r r �
test_identity s & zTestFFT1D.test_identityc C sZ t d�dt d� }tt|�tjj|�dd� tt|�tjd� tjj|dd�dd� d S )N� y �?g���ư>)r �ortho)�norm)r r r r
r �sqrt)r r r r r �test_fft$ s zTestFFT1D.test_fftr* Nr) c C sh t d�dt d� }t|tjjtjj||d�|d�dd� tjtdd�� tjjg |d� W d Q R X d S )Nr( y �?)r* g���ư>)r z!Invalid number of FFT data points)�match)r r r
r r" �pytestZraisesr )r r* r r r r � test_ifft* s
zTestFFT1D.test_ifftc C sr t d �dt d � }ttjjtjj|dd�dd�tjj|�dd� ttjj|�tjd
� tjj|d d
�dd� d S )Nr( � y �?r )r r g���ư>)r r) )r* )r( r0 )r( r0 iX )r r r
r �fft2r+ )r r r r r � test_fft25 s
zTestFFT1D.test_fft2c C sr t d �dt d � }ttjjtjj|dd�dd�tjj|�dd� ttjj|�tjd
� tjj|d d
�dd� d S )Nr( r0 y �?r )r r g���ư>)r r) )r* )r( r0 )r( r0 iX )r r r
r r" �ifft2r+ )r r r r r �
test_ifft2< s
zTestFFT1D.test_ifft2c C s~ t d
�dt d� }ttjjtjjtjj|dd�dd�dd�tjj|�d d
� ttjj|�tjd� tjj|d d �d d
� d S )Nr( r0 �
y �?r )r r r g���ư>)r r) )r* )r( r0 r5 )r( r0 r5 iX ip )r r r
r �fftnr+ )r r r r r � test_fftnC s &zTestFFT1D.test_fftnc C s~ t d
�dt d� }ttjjtjjtjj|dd�dd�dd�tjj|�d d
� ttjj|�tjd� tjj|d d �d d
� d S )Nr( r0 r5 y �?r )r r r g���ư>)r r) )r* )r( r0 r5 )r( r0 r5 iX ip )r r r
r r" �ifftnr+ )r r r r r �
test_ifftnK s &zTestFFT1D.test_ifftnc C s� t d�}x�|jd|j gD ]~}xFd D ]>}ttjj|||d�d |d d � tjj|||d�dd� q$W ttjj||d�tj|� tjj||dd�dd� qW d S )
Nr( r r) )�nr* r g���ư>)r )r: )Nr) )r �sizer r
r r$ r+ )r r r: r* r r r � test_rfftS s
zTestFFT1D.test_rfftc C sP t d�}t|tjjtjj|��dd� t|tjjtjj|dd�dd�dd� d S )Nr( g���ư>)r r) )r* )r r r
r r# r$ )r r r r r �
test_irfft^ s zTestFFT1D.test_irfftc C sf t d�}ttjj|�d d �d d�f tjj|�dd� ttjj|�tjd � tjj|dd�dd� d S )
Nr( r0 � g���ư>)r r) )r* )r( r0 iX )r r r
r r1 �rfft2r+ )r r r r r �
test_rfft2d s .zTestFFT1D.test_rfft2c C sP t d�}t|tjjtjj|��dd� t|tjjtjj|dd�dd�dd� d S )Nr( r0 g���ư>)r r) )r* )r( r0 )r r r
r Zirfft2r? )r r r r r � test_irfft2j s zTestFFT1D.test_irfft2c C sl t d �}ttjj|�d d �d d �d d�f tjj|�dd� ttjj|�tjd � tjj|dd�dd� d S ) Nr( r0 r5 � g���ư>)r r) )r* )r( r0 r5 iX ip )r r r
r r6 �rfftnr+ )r r r r r �
test_rfftnp s 4zTestFFT1D.test_rfftnc C sP t d�}t|tjjtjj|��dd� t|tjjtjj|dd�dd�dd� d S ) Nr( r0 r5 g���ư>)r r) )r* )r( r0 r5 )r r r
r �irfftnrC )r r r r r � test_irfftnv s zTestFFT1D.test_irfftnc C s� t d�dt d� }tjt d�|t d�f�}tj||d d d � j� f�}ttjj|�tjj|�dd� ttjj|�tjd� tjj|dd�dd� d S )
N� y �?r g���ư>)r r( r) )r* r )r r
� concatenate�conjr r �hfftr+ )r r �x_hermr r r � test_hfft| s zTestFFT1D.test_hfftc C s� t d�dt d� }tjt d�|t d�f�}tj||d d d� j� f�}t|tjjtjj|��dd� t|tjjtjj|dd�dd�dd� d S ) NrG y �?r g���ư>)r r) )r* r )r r
rH rI r r �ihfftrJ )r r rK r r r �
test_ihttf� s zTestFFT1D.test_ihttf�opc C s\ t d
�}d d d
dddg}x>|D ]6}|tj||��}tj|||d�|�}t||dd � q W d S )Nr( r0 r5 r r r )�axesg���ư>)r )r( r0 r5 )r r r )r r r )r r r )r r r )r r r )r r r )r r
Z transposer )r rO r rP �aZop_trZtr_opr r r � test_axes� s
zTestFFT1D.test_axesc C s� t d�}tjj|�}|jd }tjjtjjftjjtjjftjj tjj
fg}xj|D ]b\}}xX|jd|j gD ]D}x>dD ]6}||||d�}||||d�}t |tjj|�dd� qvW qlW qRW d S )Nr( r r) )r: r* g���ư>)r )Nr) ) r r
Zlinalgr* r; r r" r$ r# rM rJ r ) r r Zx_normr: Z
func_pairsZforw�backr* �tmpr r r �test_all_1d_norm_preserving� s
z%TestFFT1D.test_all_1d_norm_preserving�dtypec C sN t d�j|�}ttjjtjj|��|dd� ttjjtjj|��|dd� d S )Nr( g���ư>)r )r �astyper r
r r" r# r$ )r rV r r r r � test_dtypes� s zTestFFT1D.test_dtypes)Nr) )"r r r r' r, r. �mark� parametrizer/ r2 r4 r7 r9 r< r= r@ rA rD rF rL rN r
r r6 r8 rC rE rR rU ZhalfZsingle�doubleZ
longdoublerX r r r r r s(
r rV �order�Fznon-contiguousr c C sB t jjd�}|jddd�j| dd�}dt jt j|j�� t j|j �j
}|dkr\t j |�}n"|d d d� }t j |d d d� �}|j
jd
�r�x�td �D ],}|||d �}|||d �} t|| ||d
� q�W nx|j
jd��r8dddg}
|j
jd�� r�|
jdddd g� x>|
D ].} ||| d�}||| d�} t|| ||d
� �qW nt� �d S ) N�* � � �
F)�copyg @r] r r r )r )r Zrtolr1 r6 r r )rP r r )r1 r6 )r r )r r )r r )r )r )r )r
r Z RandomStateZrandrW r+ �log2r; ZfinforV ZepsZasfortranarrayZascontiguousarrayr �endswithr! r �extendr ) rV r\ r �rng�XZ_tol�Yr ZX_resZY_resrP Zaxr r r �test_fft_with_order� s* $
ri c @ s< e Zd ZdZdZdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�TestFFTThreadSafe� � �� c s� �fdd��t j� ��� � }� ��fdd�t| j�D �}dd� |D � dd� |D � x&t| j�D ]}t�jdd�|d � q`W d S )
Nc s |j � | � � d S )N)�put)�args�q)�funcr r �worker� s z.TestFFTThreadSafe._test_mtsame.<locals>.workerc s g | ]}t j�� �fd ��qS ))�targetro )� threading�Thread)�.0r& )ro rp rr r r �
<listcomp>� s z2TestFFTThreadSafe._test_mtsame.<locals>.<listcomp>c S s g | ] }|j � �qS r )�start)rv r r r r rw � s c S s g | ] }|j � �qS r )�join)rv r r r r rw � s � )�timeoutz6Function returned wrong value in multithreaded context)�queue�Queuer! �threadsr �get)r rq ro �expected�tr& r )ro rq rp rr r � _test_mtsame� s z TestFFTThreadSafe._test_mtsamec C s( t j| j�d d }| jt jj|� d S )Nr y )r
�ones� input_shaper� r )r rQ r r r r, � s zTestFFTThreadSafe.test_fftc C s( t j| j�d d }| jt jj|� d S )Nr y )r
r� r� r� r r" )r rQ r r r r/ � s zTestFFTThreadSafe.test_ifftc C s t j| j�}| jt jj|� d S )N)r
r� r� r� r r$ )r rQ r r r r< � s zTestFFTThreadSafe.test_rfftc C s( t j| j�d d }| jt jj|� d S )Nr y )r
r� r� r� r r# )r rQ r r r r= � s z TestFFTThreadSafe.test_irfftN)rl rm )
r r r r~ r� r� r, r/ r< r= r r r r rj � s rj ) Znumpyr
r. Z numpy.randomr Z
numpy.testingr r r rt r| r r r rY rZ Zfloat32Zfloat64Z complex64Z
complex128r r1 r6 r" r3 r8 ri rj r r r r �<module> s$ $ |