3
���hr< � @ s� d dl Z d dlmZ d dlZd dlmZ d dljjZd dl m
Z
ej dd� ej Z G dd� d�Z
ed kr�d
Zd e Zd\ZZxxed
d�D ]jZde Ze jee�jeed �Zeje�Zede dd � eede �� eded ed
f � q�W dS )� N)�reduce)�float_)�
build_err_msg�ignore)�allc @ sx e Zd Zdd� Zd dd�Zd dd�Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd � Zd S ) � ModuleTesterc C s
|| _ |j| _|j| _|j| _|j| _|j| _|j| _|j| _|j| _|j | _ t
| _
|j | _ |j | _ |j
| _
|j| _|j| _|j| _|j| _|j| _|j| _|j| _|j| _|j| _|j| _|j| _|j| _|j| _y |j| _W n t k
r� |j j| _Y nX g | _ d S )N) �module�allequal�arange�array� concatenate�count�equal�filled�getmaskZ getmaskarray�id�inner� make_mask�masked� masked_array�
masked_values�mask_or�nomask�ones�outer�repeat�resize�sort�take� transpose�zerosZMaskType�umath�AttributeError�coreZ testnames)�selfr � r% �:/tmp/pip-build-5_djhm0z/numpy/numpy/ma/timer_comparison.py�__init__ s@ zModuleTester.__init__� Tc C s, | j |�}| j |�}| j| j|�| j|��} | j | j|| d�|�}| j | j|| d�|�}|jjdkr�|jt�}t|t j
�r�|j dkr�d|t j |�< nt j |�r�d}|jjdkr�|jt�}t|t j
�r�|j dkr�d|t j |�< nt j |�r�d}�y|j
f k�p|j
f k�p|j
|j
k}
|
�sLt||g|d|j
|j
f |d
d�} |
�sLt| ��|||�} | | jk �rv|�rv| j| | d�} t| t��r�| }
dg}
n| j� }
|
j� }
|
j� }
|
�s�d d
|
jd� t|
� }t||g|d |f |dd�} |
�s�t| ��W n2 tk
�r& t||g||dd�} t| ��Y nX d S )zZ
Assert that a comparison of two masked arrays is satisfied elementwise.
)�mask�O� r z
(shapes %s, %s mismatch)�x�y)�header�names�d g Y@z
(mismatch %s%%)N)r, r- )r, r- )r, r- )r r r r �dtype�char�astyper �
isinstance�npZndarray�size�isnan�shaper �AssertionErrorr �boolZravelr �tolistr
�len�
ValueError)r$ Z
comparisonr, r- �err_msgr. Z
fill_value�xfZyf�mZcond�msg�valZreduced�matchr% r% r& �assert_array_compare3 s^
$
z!ModuleTester.assert_array_comparec C s | j | j|||dd� dS )zH
Checks the elementwise equality of two masked arrays.
zArrays are not equal)r> r. N)rD r )r$ r, r- r> r% r% r& �assert_array_equalk s z ModuleTester.assert_array_equalc
C s\ t jdddd td ddd ddddg �}ddddddddddddg }| j||d �}|d d
S )
z!
Tests creation
g �?g @g @g @g $@g @r+ r )r) Ng �g $�)r5 r �pir )r$ r, r@ �xmr% r% r& �test_0s s & zModuleTester.test_0c
C sf t jddddtd dddddddg �}t jddddddddddddg �}dd d d d d dd d d d d g }d d dd d ddd d d d dg }| j||d
�}| j||d
�}t j|d |�}|jd � || jd �j� s�t�|j }|j
t d d
� |�ks�t�| j |�t
|�t dd
� |� k�st�xRd D ]J}||_ ||_ ||_ ||_ ||_ | j |�t
|�t dd
� |� k�st��qW dS ) z!
Tests creation
g �?g @g @g @g $@g @g r+ r )r) g@��x �Dc S s | | S )Nr% )r, r- r% r% r& �<lambda>� s z%ModuleTester.test_1.<locals>.<lambda>c S s | | S )Nr% )r, r- r% r% r& rI � s � � � � c S s | | S )Nr% )r, r- r% r% r& rI � s Ng �g $�g �g �g $��rJ rK �rL rM )rN rO )r5 r rF r �whereZset_fill_valuer �anyr9 r8 r6 r r
r< ) r$ r, r- �m1�m2rG Zymr? �sr% r% r& �test_1} s& &"
&
zModuleTester.test_1c C s� t jddddg�}| j|ddddgd�}| j|ddddgd�}| j|�}t|� t|� t|d �t|d �kstt�|d |d ks�t�d|d< d|d< | j||� d|dd�< d|dd�< | j|d< | j|dd�< ||d d �< | j|d< | jddddgddddg�|d d �< | jddddgddddg�|d d �< t j d
�d }| j
|d �}| jdd
ddgt �}t jdd
ddgt �}|d |d |dd� j dk�s�t�dddddg}| j
|�}| j
|�}||k�s�t�| j
|dd�}||k �s�t�d S )z2
Tests conversions and indexing.
r+ rM rJ rK r )r) � �c N� g �?g @Zhello)�copy)r )r5 r �str�repr�typer9 rE r r r
r �objectr8 r ) r$ Zx1Zx2Zx3�x4�nr@ rS �m3r% r% r& �test_2� s@
$$
zModuleTester.test_2c
C s� | j d�}| j|d< | j|d �}| j| j||g�|�s:t�| j| j|�ddddddddg�s`t�| j|d
dd�}| j|ddddddddg� | j|ddd�}| j||�s�t�|jd dd�}| j||�s�t�|jdd�}| j||�s�t�dS ) z&
Tests resize/repeat
rJ rM � r r+ )�axisrK N)rb )rM rM rM rM )rM rM rM rM ) r
r r r r r9 r r rE )r$ r^ Zy4Zy5Zy6Zy7Zy8r% r% r& �test_3� s
& zModuleTester.test_3c C sN | j d�}tj d�}| j|dd�< |jddd�}|jddd�}| jtj|d �| j|d
��s^t�| jtj|dd�| j|dd��s�t�| jtj| j |d�| j |d��| j||��s�t�| jtj
| j |d�| j |d��| j
||��s�t�| j d dd
ddgt �}| j|d< | j|dddg�}|d d k�s&t�|d dk�s8t�|d dk�sJt�d S )zB
Test of take, transpose, inner, outer products.
� rX rL rM rK rJ r r+ �abc�defN)rM r r+ )rM r r+ )rM r r+ )rM r r+ )
r
r5 r �reshaper r r9 r r r r r r] )r$ r, r- �tr% r% r& �test_4� s"
$
zModuleTester.test_4c C s | j d�}| j d�}| j d�}| j|d< |d7 }| j||d �sDt�|d7 }| j||d �s`t�| j d�}| j d�}| j|d< |d8 }| j||d �s�t�|d8 }| j||d �s�t�| j d�d }| j d�d }| j|d< |d9 }| j||d �s�t�|d9 }| j||d ��st�| j d�d }| j d�d }| j|d< |d }| j||��sVt�|d }| j||��spt�| j d�d }| j d�d }| j|d< |d }| j||d ��s�t�|| j d� }| j|| jd�� | j d�jt�}| j d�}| j|d< |d7 }| j||d ��st�dS )z*
Tests inplace w/ scalar
�
rM r+ g �?g @N)rk )r
r r r9 rE r r3 r )r$ r, r- rG r% r% r& �test_5� sR
zModuleTester.test_5c C s6 | j dtd�}| j d�}| j dtd�}| j|d< |j}| j dtd�}| j|d< ||7 }||7 }| j||| �srt�| j||| �s�t�| j|j| j||j��s�t�| j dtd�}| j dtd�}| j|d< |j}| j dtd�}| j|d< ||8 }||8 }| j||| ��s t�| j||| ��s"t�| j|j| j||j���s@t�| j dtd�}| j dtd�}| j|d< |j}| j dtd�}| j|d< ||9 }||9 }| j||| ��s�t�| j||| ��s�t�| j|j| j||j���s�t�| j dtd�}| j dtd�}| j|d< |j}| j dtd�}| j|d < || }|| }dS )
z)
Tests inplace w/ array
rk )r1 rM r+ N���rm rm rm )r
r r r) r r9 r )r$ r, r- rG r@ �ar% r% r& �test_6 sT
zModuleTester.test_6c
C s� | j ddd
td gd ddgdgd d�| j dddtd gd ddgdgd d�f}x�dD ]�}yt| j|�}W n tk
r� tt|�}Y nX t| j|�}|d |j� }||� }||� }| j|j d�|j d�|� | j|j
|j
� q^W d S )z Tests ufuncg �?r r+ rM rL )r) �sqrt�log�log10�exp� conjugateNrm rm )rp rq rr rs rt ) r rF �getattrr! r" � fromnumericr ZninrE r Z_mask)r$ �d�fZufZmf�args�ur�mrr% r% r& �test_7S s *.
zModuleTester.test_7c C s( | j ddddgddddgd�}| jd| j|dd�� | jd| j|ddddgd �� | j|ddddgdd
�\}}| jd|� |d ks�t�| j|d d �< | j|dd�| jks�t�| j ddddgddddgd�}|jd d �}| j|d d �df< | j| j|dd�ddg� | j|dd�d | jk�s t�| jddg| j|dd�� | j|ddd
�\}}| j|ddg� ddddddg}ddddddgddddddgg}| jd�}| j| j|dd�d� | j| j|d|d�d� | j | jd�d| jd� g�}| j| j|d �tjj tjd��d d � | j| j|dd�tjd�d d � | j| j|dd�| j|dd�| j|dd�d g� | j| j|d |d �d� | j| j|d|d�ddddd dg� | j| j|dd�| j|dd�| j|dd�d g� | j
d�}ddddddg} ddddddgddddddgg}
| j d�} ddddddg} | j| j| j ||�dd�d� | j| j| j || �dd�d� | j| j| j || �dd�d� | j| j
| j| j || �dd��d� | j ||
�}
| j| j|
d �d� | j| j|
dd�ddddd dg� | j| j|
dd�ddg� | j| j|
d|d�ddddd dg� d S )Ng g �?g @g @r+ r )r) )rc )�weights)r} �returnedg @rM )rc r~ rL g @)rc r} g (@g 4@g @g $@g �X@g @g @g������
@g������
@)r rE Zaverager9 r rh r
r5 �addr r r r r
)r$ Zott�resultZwtsZw1Zw2r, r- rR rS r` Zm4Zm5�zr% r% r& �test_99u sT
*$2$2
$ "zModuleTester.test_99c C s* | j d�}| j|dd�< |jddd�}d S )Nre rX rL rM rK rJ )r
r rh )r$ r, r% r% r& �test_A� s
zModuleTester.test_AN)r( r( T)r( )�__name__�
__module__� __qualname__r' rD rE rH rU ra rd rj rl ro r| r� r� r% r% r% r& r s "
7
*43"-r �__main__zNfrom __main__ import ModuleTester
import numpy
tester = ModuleTester(module)
z import numpy.ma.core as module
rk r+ rb ztester.test_%i()z#%i�2 �.z ModuleTester.test_%i.__doc__zcore_current : %.3f - %.3f)rk rk ) Ztimeit� functoolsr Znumpyr5 r Znumpy.core.fromnumericr# rv Z
numpy.testingr ZseterrrF r r� Z
setup_baseZ setup_curZnrepeatZnloop�range�i�func�Timerr �curr �print�evalr% r% r% r& �<module> s,
|