Edit File: test_fftpack.cpython-37.pyc
B ��Fd� � @ s� d dl mZmZmZ d dlZd dlmZ d dlm Z m Z m Z d dlm Z d dl Z d dlZejd dkrpd dlZnd dlZdd� ZG d d � d e �ZG d d � d e �ZG d d� de �Zedkr�e � dS )� )�division�absolute_import�print_functionN)�random)�TestCase�run_module_suite�assert_array_almost_equal)�assert_array_equal� c C sR t | �}dtj t�|�t|� }t�|��dd�| }tj| t�|� dd�S )Ny � ����� )�axis)�len�np�pi�arange�float�reshape�sum�exp)�x�L�phase� r ��/builddir/build/BUILDROOT/alt-python37-numpy-1.13.3-8.el8.x86_64/opt/alt/python37/lib64/python3.7/site-packages/numpy/fft/tests/test_fftpack.py�fft1 s r c @ s e Zd Zdd� ZdS )� TestFFTShiftc C s | � ttjjdddgd� d S )Nr � r r )� assertRaises� 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� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd S ) � TestFFT1Dc C sR t d�dt d� }tt|�tj�|�� tt|�t�d� tjj|dd�� d S )N� y �?�ortho)�norm)r r r r r �sqrt)r! r r r r �test_fft s zTestFFT1D.test_fftc C sT t d�dt d� }t|tj�tj�|��� t|tjjtjj|dd�dd�� d S )Nr' y �?r( )r) )r r r r �ifft)r! r r r r � test_ifft$ s zTestFFT1D.test_ifftc C sj t d�dt d� }ttjjtjj|dd�dd�tj�|�� ttj�|�t�d� tjj|dd�� d S ) N)r' � y �?r )r r iX r( )r) )r r r r �fft2r* )r! r r r r � test_fft2* s zTestFFT1D.test_fft2c C sj t d�dt d� }ttjjtjj|dd�dd�tj�|�� ttj�|�t�d� tjj|dd�� d S ) N)r' r. y �?r )r r iX r( )r) )r r r r r, �ifft2r* )r! r r r r � test_ifft21 s zTestFFT1D.test_ifft2c C sv t d�dt d� }ttjjtjjtjj|dd�dd�dd�tj�|�� ttj�|�t�d� tjj|dd �� d S ) N)r' r. � y �?r )r r r ip r( )r) )r r r r �fftnr* )r! r r r r � test_fftn8 s &zTestFFT1D.test_fftnc C sv t d�dt d� }ttjjtjjtjj|dd�dd�dd�tj�|�� ttj�|�t�d� tjj|dd �� d S ) N)r' r. r3 y �?r )r r r ip r( )r) )r r r r r, �ifftnr* )r! r r r r � test_ifftn@ s &zTestFFT1D.test_ifftnc C s� t d�}x�|jd|j gD ]v}xBdD ]:}ttjj|||d�d |d d � tjj|||d�� q$W ttjj||d�t�|� tjj||dd�� qW d S )Nr' r )Nr( )�nr) r )r8 r( )r �sizer r r �rfftr* )r! r r8 r) r r r � test_rfftH s zTestFFT1D.test_rfftc C sH t d�}t|tj�tj�|��� t|tjjtjj|dd�dd�� d S )Nr' r( )r) )r r r r �irfftr: )r! r r r r � test_irfftR s zTestFFT1D.test_irfftc C s^ t d�}ttj�|�d d �d d�f tj�|�� ttj�|�t�d� tjj|dd�� d S )N)r' r. � iX r( )r) )r r r r r/ �rfft2r* )r! r r r r � test_rfft2X s *zTestFFT1D.test_rfft2c C sH t d�}t|tj�tj�|��� t|tjjtjj|dd�dd�� d S )N)r' r. r( )r) )r r r r �irfft2r? )r! r r r r � test_irfft2^ s zTestFFT1D.test_irfft2c C sd t d�}ttj�|�d d �d d �d d�f tj�|�� ttj�|�t�d� tjj|dd�� d S )N)r' r. r3 � ip r( )r) )r r r r r4 �rfftnr* )r! r r r r � test_rfftnd s 0zTestFFT1D.test_rfftnc C sH t d�}t|tj�tj�|��� t|tjjtjj|dd�dd�� d S )N)r' r. r3 r( )r) )r r r r �irfftnrD )r! r r r r � test_irfftnj s zTestFFT1D.test_irfftnc C s� t d�dt d� }t�t d�|t d�f�}t�||d d d� �� f�}ttj�|�tj�|�� ttj�|�t�d� tjj|dd�� d S )N� y �?r r r' r( )r) )r r � concatenate�conjr r �hfftr* )r! r �x_hermr r r � test_hfftp s zTestFFT1D.test_hfftc C s� t d�dt d� }t�t d�|t d�f�}t�||d d d� �� f�}t|tj�tj�|��� t|tjjtjj|dd�dd�� d S )NrH y �?r r r( )r) )r r rI rJ r r �ihfftrK )r! r rL r r r � test_ihttfx s zTestFFT1D.test_ihttfc C s� t d�}tj�|�}|jd }tjjtjjftjjtjjftjj tjj fg}xf|D ]^\}}xT|jd|j gD ]@}x:dD ]2}||||d�}||||d�}t |tj�|�� qvW qlW qRW d S )Nr' r )Nr( )r8 r) ) r r �linalgr) r9 r r, r: r< rN rK r ) r! r Zx_normr8 Z func_pairsZforwZbackr) �tmpr r r �test_all_1d_norm_preserving� s z%TestFFT1D.test_all_1d_norm_preservingN)r# r$ r% r+ r- r0 r2 r5 r7 r; r= r@ rB rE rG rM rO rR r r r r r& s r& c @ s< e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd d � Z d S )�TestFFTThreadSafe� )i �� c s� �fdd��t �� ��� � }� ��fdd�t| j�D �}dd� |D � dd� |D � x&t| j�D ]}t�jdd�|d � q`W d S ) Nc s |� � | � � d S )N)�put)�args�q)�funcr r �worker� s z.TestFFTThreadSafe._test_mtsame.<locals>.workerc s g | ]}t j�� �fd ��qS ))�targetrW )� threading�Thread)�.0�i)rW rX rZ r r � <listcomp>� s z2TestFFTThreadSafe._test_mtsame.<locals>.<listcomp>c S s g | ] }|� � �qS r )�start)r^ r r r r r` � s c S s g | ] }|� � �qS r )�join)r^ r r r r r` � s � )�timeoutz6Function returned wrong value in multithreaded context)�queue�Queue�range�threadsr �get)r! rY rW �expected�tr_ r )rW rY rX rZ r � _test_mtsame� s z TestFFTThreadSafe._test_mtsamec C s( t �| j�d d }| �t jj|� d S )Nr y )r �ones� input_shaperl r )r! �ar r r r+ � s zTestFFTThreadSafe.test_fftc C s( t �| j�d d }| �t jj|� d S )Nr y )r rm rn rl r r, )r! ro r r r r- � s zTestFFTThreadSafe.test_ifftc C s t �| j�}| �t jj|� d S )N)r rm rn rl r r: )r! ro r r r r; � s zTestFFTThreadSafe.test_rfftc C s( t �| j�d d }| �t jj|� d S )Nr y )r rm rn rl r r< )r! ro r r r r= � s z TestFFTThreadSafe.test_irfftN) r# r$ r% rh rn rl r+ r- r; r= r r r r rS � s rS �__main__)� __future__r r r �numpyr Z numpy.randomr Z numpy.testingr r r r r\ �sys� version_infore rf r r r& rS r# r r r r �<module> s x'