Submit
Path:
~
/
/
proc
/
thread-self
/
root
/
opt
/
alt
/
python35
/
lib64
/
python3.5
/
site-packages
/
numpy
/
fft
/
tests
/
__pycache__
/
File Content:
test_helper.cpython-35.pyc
X��f� � @ s� d Z d d l m Z m Z m Z d d l Z d d l m Z m Z m Z d d l m Z d d l m Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z e d k r� e � d S)zfTest functions for fftpack.helper module Copied from fftpack.helper by Pearu Peterson, October 2005 � )�division�absolute_import�print_functionN)�TestCase�run_module_suite�assert_array_almost_equal)�fft)�pic @ s4 e Z d Z d 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 j | � | � t t j | � | � d d d d d d d d d d g } d d d d d d d d d d g } t t j | � | � t t j | � | � d S)Nr � � � � � ������������r r r r ���r r r r r r r r r )r r �fftshift� ifftshift)�self�x�y� r �/test_helper.py�test_definition s !!$$zTestFFTShift.test_definitionc C sX xQ d d d d d g D]: } t j j | f � } t t j t j | � � | � q Wd S)Nr r � �d �� )�np�randomr r r r )r �nr r r r �test_inverse s zTestFFTShift.test_inversec C s� d d d g d d d g d d d g g } d d d g d d d g d d d g g } t t j | d d �| � t t j | d d �t j | d d �� t t j | d d �| � t t j | d d �t j | d d �� d S)Nr r r r r �axesr r r r r r r r )r r )r )r r )r )r r r r )r ZfreqsZshiftedr r r �test_axes_keyword s **zTestFFTShift.test_axes_keywordN)�__name__� __module__�__qualname__r 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 j d � | � t d t t j d t � | � d d d d d d d d d d g } t d t j d � | � t d t t j d t � | � d S)Nr r r r r r r � r r r r r r r r r )r r Zfftfreqr )r r r r r r , s !!$zTestFFTFreq.test_definitionN)r&