Current File : //lib64/python3.9/site-packages/numpy/ma/__pycache__/timer_comparison.cpython-39.opt-1.pyc
a
z[yc*= � @ s� d dl Z d dlmZ d dlZd dlmZ d dlm mZ d dl m
Z
ej Z G dd� d�Z e
dkr�dZd e Zd
\ZZed d �D ]nZd
e Ze �ee��eed �Ze�e�Zede d � eede �� eded d�ded d��� q�dS )� N)�reduce)�float_)�
build_err_msgc @ s� e Zd Zdd� Zd dd�Zd!dd�Zejd d
�d d � �Zejd d
�d
d� �Z ejd d
�dd� �Z
ejd d
�dd� �Z ejd d
�dd� �Z ejd d
�dd� �Z
ejd d
�dd� �Zejd d
�dd� �Zejd d
�dd � �Zejd d
�d d � �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| _z |j| _W n t y� |j j| _Y n0 g | _ d S )N) �moduleZallequal�arange�arrayZ concatenate�count�equal�filled�getmaskZ getmaskarray�id�inner� make_mask�masked� masked_array�
masked_values�mask_or�nomask�onesZouter�repeat�resize�sort�takeZ transpose�zerosZMaskType�umath�AttributeError�coreZ testnames)�selfr � r �?/usr/lib64/python3.9/site-packages/numpy/ma/timer_comparison.py�__init__ s@ zModuleTester.__init__� Tc
C s& | � |�}| � |�}| �| �|�| �|��} | � | j|| d�|�}| � | j|| d�|�}|jjdkr�|�t�}t|t j
�r�|j dkr�d|t � |�< nt � |�r�d}|jjdkr�|�t�}t|t j
�r�|j dkr�d|t � |�< nt � |�r�d}z�|j
dk�p|j
dk�p|j
|j
k}
|
�sBt||g|d|j
� d|j
� d� |d d
�} |||�} | | ju�rl|�rl| j| | d�} t| t��r�| }
dg}
n| �� }
|
�� }
|
�� }
|
�s�d d |
�d� t|
� }t||g|d
|f |d d
�} W nD t�y } z*t||g||d d
�} t| �|�W Y d}~n
d}~0 0 dS )zZ
Assert that a comparison of two masked arrays is satisfied elementwise.
��mask�O� r r z
(shapes z, z
mismatch))�x�y)�header�names�d g Y@z
(mismatch %s%%)N)r r r r �dtype�char�astyper �
isinstance�npZndarray�sizeZisnan�shaper r �boolZravel�all�tolistr �len�
ValueError)r Z
comparisonr' r( �err_msgr) Z
fill_value�xfZyf�mZcond�msg�valZreduced�match�er r r �assert_array_compare0 sb
$��
��z!ModuleTester.assert_array_comparec C s | j | j|||dd� dS )zH
Checks the elementwise equality of two masked arrays.
zArrays are not equal)r8 r) N)r? r
)r r' r( r8 r r r �assert_array_equalg s �z ModuleTester.assert_array_equal�ignore)r4 c C sH t �ddddtd dddddddg �}g d �}| j||d
�}|d d S )
�!
Tests creation
� �?� �� @� @� @� $�� $@� @� r& r r r r r r& r r r r r r# r N)r0 r �pir )r r' r: �xmr r r �test_0o s &zModuleTester.test_0c C s� t �ddddtd dddddddg �}t �g d ��}g d
�}g d �}| j||d �}| j||d �}t �|d
|�}|�d
� |j}dD ]"}||_||_||_||_||_q�dS )rB rC rD rE rF rG rH rI rJ ) rG � rJ rE g �g �rO rH rI rC rO rJ rK ) r r r& r r r& r& r r r r r& r# g@��x �D))� � )� � N)r0 r rL r �whereZset_fill_valuer2 ) r r' r( �m1�m2rM Zymr9 �sr r r �test_1z s &
zModuleTester.test_1c C s^ t �g d��}| j|g d�d�}| j|g d�d�}| �|�}t|� t|� d|d< d|d< | �||� d|dd �< d|dd �< | j|d< | j|dd �< ||d
d
�< | j|d< | �g d �g d ��|d
d
�< | �g d �g d ��|d
d
�< t �d
�d }| �|d�}| �g d�t �}t �g d�t �}|d |d g d�}| �
|�}| �
|�}| j
|dd�}d
S )z2
Tests conversions and indexing.
)r& rS rP rQ �r& r r r r# )r r& r r& � rS �c r& rQ N)r& rS rQ rP )r r& r& r � rC rJ )r& ZhellorS rQ )r r r& r r )�copy) r0 r �str�reprr@ r r r r �objectr ) r Zx1Zx2Zx3�x4�nr: rV �m3r r r �test_2� s8
zModuleTester.test_2c C sn | � d�}| j|d< | �|d�}| j|ddd�}| �|g d�� | j|ddd�}|jddd�}|�dd�}dS ) z&
Tests resize/repeat
rP rS )� )rS rS rS rS r ��axis)r r r& r& rS rS rQ rQ N)r r r r r@ )r ra Zy4Zy5Zy6Zy7Zy8r r r �test_3� s
zModuleTester.test_3c C sl | � d�}t� d�}| j|dd�< |�ddd�}|�ddd�}| �g d�t�}| j|d< | �|g d��}d S )
zB
Test of take, transpose, inner, outer products.
� r\ rR rS rQ rP )�abcr& �defrS rQ )r rQ rP N)r r0 r �reshaper r` r )r r' r( �tr r r �test_4� s
zModuleTester.test_4c C sP | � d�}| � d�}| � d�}| j|d< |d7 }|d7 }| � d�}| � d�}| j|d< |d8 }|d8 }| � d�d }| � d�d }| j|d< |d9 }|d9 }| � d�d }| � d�d }| j|d< |d }|d }| � d�d }| � d�d }| j|d< |d }|| � d� }| �|| �d�� | � d��t�}| � d�}| j|d< |d7 }dS )z*
Tests inplace w/ scalar
�
rS r& rC rE )ro N)r r r@ r r. r )r r' r( rM r r r �test_5� s@
zModuleTester.test_5c C s^ | j dtd�}| � d�}| j dtd�}| j|d< |j}| j dtd�}| j|d< ||7 }||7 }| j dtd�}| j dtd�}| j|d< |j}| j dtd�}| j|d< ||8 }||8 }| j dtd�}| j dtd�}| j|d< |j}| j dtd�}| j|d< ||9 }||9 }| j dtd�}| j dtd�}| j|d< |j}| j dtd�}| j|d< || }|| }dS )z)
Tests inplace w/ array
ro )r, rS ���N)r r r r$ )r r' r( rM r: �ar r r �test_6" sB
zModuleTester.test_6c C s� | j dddtd gd ddgdgd d�| j dddtd gd ddgdgd d�f}dD ]�}zt| j|�}W n ty� tt|�}Y n0 t| j|�}|d |j� }||� }||� }| �|� d�|� d�|� | �|j
|j
� q\d S )
z Tests ufuncrC r rq rS r& rR r# )Zsqrt�logZlog10Zexp� conjugateN) r rL �getattrr r � fromnumericr Zninr@ r Z_mask)r �d�fZufZmf�args�ur�mrr r r �test_7V s **� zModuleTester.test_7c C s~ | j g d�g d�d�}| �d| j|dd�� | �d| j|g d�d�� | j|g d�d d
�\}}| �d|� | j|d d �< | j g d�g d�d�}|�d d �}| j|d d �d f< | �| j|dd�dd g� | �dd g| j|dd�� | j|dd d
�\}}| �|dd g� g d�}g d�g d�g}| �d�}| �| j|dd�d� | �| j|d|d�d� | � | �d�d| �d� g�}| �| �|d �tj�t�d��d d � | �| j|dd�t�d�d d � | �| j|d d�| j|dd�| j|dd�d g� | �| j|d |d�d� | �| j|d|d�g d�� | �| j|d d�| j|dd�| j|dd�d g� | � d�}g d�} g d�g d�g}
| �
d�} g d�} | �| j| � ||�dd�d� | �| j| � || �dd�d� | �| j| � || �dd�d � | �| � | j| � || �dd��d� | � ||
�}
| �| �|
d �d� | �| j|
dd�g d�� | �| j|
d d�ddg� | �| j|
d|d�g d �� d S ) N)rO rC rE rJ rY r# rE r rf )rC rC rE rC )�weightsr& )r~ �returnedrS rO )rg r rC )r r& r& r& r& r )r& r r r r r& rR g @)rg r~ rJ g (@g������
@)rO rC rE rJ rF rI )r r r& r& r r )r r& r& r& r& r& )rO rC � �X@r� rF g @rG )rO rC r� r� rF rI )
r r@ Zaverager rl r r0 �addr r r r r )r Zott�resultZwtsZw1Zw2r' r( rU rV rc Zm4Zm5�zr r r �test_99y sN
*$2 2
$ zModuleTester.test_99c C s* | � d�}| j|dd�< |�ddd�}d S )Nri r\ rR rS rQ rP )r r rl )r r' r r r �test_A� s
zModuleTester.test_AN)r" r" T)r" )�__name__�
__module__� __qualname__r! r? r@ r0 ZerrstaterN rX rd rh rn rp rs r} r� r� r r r r r
s2 " �
7
*
4
3
"
-
r �__main__zNfrom __main__ import ModuleTester
import numpy
tester = ModuleTester(module)
z import numpy.ma.core as module
)ro ro r& re ztester.test_%i()ro z#%iz2..................................................z ModuleTester.test_%i.__doc__zcore_current : z.3fz - )Ztimeit� functoolsr Znumpyr0 r Znumpy.core.fromnumericr rw Z
numpy.testingr rL r r� Z
setup_baseZ setup_curZnrepeatZnloop�range�i�func�Timerr Zcurr �print�evalr r r r �<module> s* $
Mini Shell