Edit File: test_helper.cpython-37.pyc
B ��Fd� � @ s� d Z ddlmZmZmZ ddlZddlmZm Z m Z ddlm Z ddlm Z ddl mZ G dd � d e�ZG d d � d e�ZG d d � d e�ZG dd� de�ZG dd� de�Zedkr�e � dS )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)�pi)� _FFTCachec @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestFFTShiftc C s� ddddddddd g }dddd dddddg }t t�|�|� t t�|�|� dddddd dddd g }d dddd dddddg }t t�|�|� t t�|�|� d S ) Nr � � � � ���������������)r r �fftshift� ifftshift)�self�x�y� 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_helper.py�test_definition s z TestFFTShift.test_definitionc C s6 x0dD ](}t j�|f�}tt�t�|��|� qW d S )N)r r � �d �� )�np�randomr r r r )r �nr r r r � test_inverse s zTestFFTShift.test_inversec C s� dddgdddgddd gg}d ddgdddgdddgg}t tj|d d �|� t tj|dd �tj|d d �� t tj|d d �|� t tj|dd �tj|d d �� d S ) Nr r r r r r r r r )r r )�axes)r )r r r r )r ZfreqsZshiftedr r r �test_axes_keyword s z TestFFTShift.test_axes_keywordN)�__name__� __module__� __qualname__r r# r% r r r r r s r c @ s e Zd Zdd� ZdS )� TestFFTFreqc C s� ddddddddd g }t d t�d � |� t d t t�d t� |� dddddd dddd 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 �fftfreqr )r r r r r r - s zTestFFTFreq.test_definitionN)r& r' r( r r r r r r) + s r) c @ s e Zd Zdd� ZdS )� TestRFFTFreqc C s~ dddddg}t dt�d� |� t dt t�dt� |� ddddddg}t dt�d� |� t dt t�dt� |� d S ) Nr r r r r r � r* )r r �rfftfreqr )r r r r r r 8 s z TestRFFTFreq.test_definitionN)r& r' r( r r r r r r, 6 s r, c @ s e Zd Zdd� ZdS )� TestIRFFTNc C s2 t j�d�\}}|d| }d}tj||d� d S )N)r � � � y �?)r )r$ )r r! r �irfftn)r �ar�ai�ar$ r r r �test_not_last_axis_successC s z%TestIRFFTN.test_not_last_axis_successN)r& r' r( r7 r r r r r/ A s r/ c @ s e Zd Zdd� Zdd� ZdS )� TestFFTCachec C s t ddd�}|�dtjdtjd�� |�dtjdtjd�� t|�d�tjdtjd�� t|�d�tjdtjd�� | �t |j �d� x@t d�D ]4}|�dtjdtjd�� |�dtjdtjd�� q�W t|�d�tjdtjd�� t|�d�tjdtjd�� | �t |j �d� d S )Nr r )�max_size_in_mb�max_item_countr )�dtyper ) r �put_twiddle_factorsr �ones�float32�zerosr �pop_twiddle_factors� assertEqual�len�_dict�range)r �c�_r r r �test_basic_behaviourO s z!TestFFTCache.test_basic_behaviourc C s8 t ddd�}|�dtjdtjd�� |�dtjdtjd�� | �t|j�� �ddg� |�dtjd tjd�� | �t|j�� �dddg� |�dtjd tjd�� | �t|j�� �dg� t ddd�}|�dt� d�� |�dt� d�� | �t|j�� �ddg� |�dt� d�� | �t|j�� �ddg� t dd d�}|�dtjd tjd�� |�dtjdtjd�� |�dtjdtjd�� |�dtjdtjd�� | �t|j�� �ddddg� x*t d �D ] }|�d tjd tjd�� �q�W | �t|j�� �dddd g� |�d tjd tjd�� | �t|j�� �d g� d S )Ng{�G�z�?r )r9 r: r �� )r; r r i� r- � i� i� ) r r< r r= r> rA �listrC �keys�emptyrD )r rE rF r r r �test_automatic_pruningj s2 z#TestFFTCache.test_automatic_pruningN)r& r' r( rG rM r r r r r8 M s r8 �__main__)�__doc__� __future__r r r �numpyr Z numpy.testingr r r r r Znumpy.fft.helperr r r) r, r/ r8 r& r r r r �<module> s N