Submit
Path:
~
/
/
proc
/
thread-self
/
root
/
opt
/
alt
/
python35
/
lib64
/
python3.5
/
site-packages
/
numpy
/
fft
/
tests
/
__pycache__
/
File Content:
test_fftpack.cpython-35.opt-1.pyc
X��f� � @ s d d l m Z m Z m Z d d l Z d d l m Z d d l m Z m Z m Z d d l m Z d d l Z d d l Z e j d d k r� d d l Z n d d l Z d d � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z e d k re � d S)� )�division�absolute_import�print_functionN)�random)�TestCase�run_module_suite�assert_array_almost_equal)�assert_array_equal� c C so t | � } d t j t j | � t | � } t j | � j d d � | } t j | t j | � d d �S)Ny @� �axisy � ����)�len�np�piZarange�floatZreshape�sum�exp)�x�LZphase� r �/test_fftpack.py�fft1 s $r c @ s e Z d Z d d � Z d S)�TestFFTShiftc C s) | j t t j j d d d g d � d S)Nr � r r )ZassertRaises� 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 Z d Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d S)� TestFFT1Dc C so t d � d t d � } t t | � t j j | � � t t | � t j d � t j j | d d �� d S)N� y �?�norm�ortho)r r r r r �sqrt)r r r r r �test_fft s zTestFFT1D.test_fftc C st t d � d t d � } t | t j j t j j | � � � t | t j j t j j | d d �d d �� d S)Nr# y �?r$ r% )r r r r �ifft)r r r r r � test_ifft$ s %zTestFFT1D.test_ifftc C s� t d � d t d � } t t j j t j j | d d �d d �t j j | � � t t j j | � t j d � t j j | d d �� d S)Nr# � y �?r r r r$ r% )r# r* )r# r* iX )r r r r �fft2r&