/ lib / numpy / ma / timer_comparison.pyc
timer_comparison.pyc
  1  o

  2  [��c�>�@s�ddlZddlmZddlZddlmZddlmmZddl	m
  3  Z
  4  ejZGdd�d�Ze
dkrxdZd	eZd
  5  \ZZedd�D]9Zd
eZe�ee��eed�Ze�e�Zeded�eede��ededd�dedd���q@dSdS)�N)�reduce)�float_)�
build_err_msgc@s�eZdZdd�Z		d dd�Zd!dd�Zejd	d
  6  �dd��Zejd	d
  7  �d
d��Z	ejd	d
  8  �dd��Z
  9  ejd	d
 10  �dd��Zejd	d
 11  �dd��Zejd	d
 12  �dd��Z
ejd	d
 13  �dd��Zejd	d
 14  �dd��Zejd	d
 15  �dd��Zejd	d
 16  �dd��ZdS)"�ModuleTestercCs||_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j	|_	t
 17  |_
 18  |j|_|j|_|j
|_
|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_|j|_z|j|_Wnty~|jj|_Ynwg|_dS�N)�module�allequal�arange�array�concatenate�count�equal�filled�getmask�getmaskarray�id�inner�	make_mask�masked�masked_array�
masked_values�mask_or�nomask�ones�outer�repeat�resize�sort�take�	transpose�zeros�MaskType�umath�AttributeError�coreZ	testnames)�selfr�r&��C:\Users\Jacks.GUTTSPC\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.10_qbz5n2kfra8p0\LocalCache\local-packages\Python310\site-packages\numpy\ma\timer_comparison.py�__init__sB�
 19  zModuleTester.__init__�Tc
 20  Cs(|�|�}|�|�}|�|�|�|�|��}	|�|j||	d�|�}|�|j||	d�|�}|jjdkrQ|�t�}t|t	j
 21  �rJ|jdkrJd|t	�|�<nt	�|�rQd}|jjdkrv|�t�}t|t	j
 22  �ro|jdkrod|t	�|�<nt	�|�rvd}z�|j
dkp�|j
dkp�|j
|j
k}
 23  |
 24  s�t||g|d|j
�d|j
�d�|d	d
 25  �}|
 26  s�J|��|||�}|	|jur�|r�|j||	d�}t|t�r�|}
 27  dg}
n|��}
|
��}
 28  |
��}
|
 29  s�dd|
�d�t|
�}t||g|d
|f|d	d
 30  �}|
 31  s�J|��WdSWdSt�y}zt||g||d	d
 32  �}t|�|�d}~ww)zZ
 33          Assert that a comparison of two masked arrays is satisfied elementwise.
 34  
 35          ��mask�O�rr&z	
 36  (shapes z, z
 37   mismatch)��x�y)�header�names�dgY@z
 38  (mismatch %s%%)N)rrrr�dtype�char�astyper�
 39  isinstance�np�ndarray�size�isnan�shaperr�bool�ravel�all�tolistr�len�
 40  ValueError)r%�
 41  comparisonr/r0�err_msgr1�
 42  fill_value�xf�yf�m�cond�msg�val�reduced�match�er&r&r'�assert_array_compare0sl
 43  
 44  
 45  
 46  
 47  
 48   ��
 49  
 50  ���
 51  ��z!ModuleTester.assert_array_comparecCs|j|j|||dd�dS)zH
 52          Checks the elementwise equality of two masked arrays.
 53  
 54          zArrays are not equal)rDr1N)rOr
)r%r/r0rDr&r&r'�assert_array_equalgs
 55  �zModuleTester.assert_array_equal�ignore)r?cCsHt�ddddtddddddddg�}gd	�}|j||d
 56  �}|ddS)
�!
 57          Tests creation
 58  
 59          ��?���@�@�@�$��$@�@�r-rrrrrr-rrrrrr*rN)r8r
 60  �pir)r%r/rH�xmr&r&r'�test_0os&zModuleTester.test_0c	Cs t�ddddtddddddddg�}t�gd	��}gd
 61  �}gd�}|j||d�}|j||d�}t�|d
|�}|�d
�||�d���sGJ�|j}|j	t
 62  dd�|�ksVJ�|�|�t|�t
 63  dd�|�kshJ�dD]#}||_||_||_||_||_|�|�t|�t
 64  dd�|�ks�J�qjdS)rRrSrTrUrVrWrXrYrZ)rW�rZrUg�g�r_rXrYrSr_rZr[)rrr-rrr-r-rrrrr-r*g@��x�DrcSs||Srr&r.r&r&r'�<lambda>��z%ModuleTester.test_1.<locals>.<lambda>cS�||Srr&r.r&r&r'r`�ra))��)��cSrbrr&r.r&r&r'r`�raN)
r8r
 65  r\r�where�set_fill_valuer�anyr<r:rrrA)	r%r/r0�m1�m2r]�ymrF�sr&r&r'�test_1zs(&
 66  $&�zModuleTester.test_1c	Cs�t�gd��}|j|gd�d�}|j|gd�d�}|�|�}t|�t|�t|d�t|d�us4J�|d|dks>J�d|d<d|d<|�||�d|dd	�<d|dd	�<|j|d<|j|dd	�<||d
 67  d
 68  �<|j|d<|�gd�gd��|d
 69  d
 70  �<|�gd�gd��|d
 71  d
 72  �<t�d
�d}|�	|d�}|�gd�t
 73  �}t�gd�t
 74  �}|d|d|dd�jdks�J�gd�}|�|�}|�|�}||us�J�|j|dd�}||us�J�d
 75  S)z2
 76          Tests conversions and indexing.
 77  
 78          )r-rfrcrd�r-rrrr*)rr-rr-r-�	rf�crdN)r-rfrdrc)rr-r-r�rSrZ)r-�hellorfrd)r)rrr-rr)�copy)
r8r
 79  �str�repr�typerPrrr	r�objectr<r)	r%�x1�x2�x3�x4�nrHrk�m3r&r&r'�test_2�s@
 80  
 81  
 82  
 83  
 84  zModuleTester.test_2cCs�|�d�}|j|d<|�|d�}|�|�||g�|�sJ�|�|�|�gd��s*J�|j|ddd�}|�|gd��|j|ddd�}|�||�sJJ�|jddd�}|�||�sYJ�|�dd�}|�||�sgJ�d	S)
 85  z&
 86          Tests resize/repeat
 87  
 88          rcrf)�)rrr-rrrr-r)rfrfrfrfr��axis)rrr-r-rfrfrdrdN)r	rrrrrrrP)r%r|�y4�y5�y6�y7�y8r&r&r'�test_3�s
 89  
 90  zModuleTester.test_3c	Cs@|�d�}t�d�}|j|dd�<|�ddd�}|�ddd�}|�t�|d�|�|d��s/J�|�t�|dd�|�|dd��sAJ�|�t�|�|d	�|�|d	��|�||��sYJ�|�t�	|�|d	�|�|d	��|�	||��sqJ�|�
 91  gd
 92  �t�}|j|d<|�|gd��}|d	dks�J�|ddks�J�|ddks�J�d
S)zB
 93          Test of take, transpose, inner, outer products.
 94  
 95          �rrrerfrdrc)rfrr-r-r)�abcr-�defrfrd)rrdrcr�N)r	r8r�reshaperrrrrrr
 96  rx)r%r/r0�tr&r&r'�test_4�s&
 97  
 98   $
 99  �
100  �
101  zModuleTester.test_4cCs|�d�}|�d�}|�d�}|j|d<|d7}|�||d�s"J�|d7}|�||d�s0J�|�d�}|�d�}|j|d<|d8}|�||d�sMJ�|d8}|�||d�s[J�|�d�d}|�d�d}|j|d<|d9}|�||d�s|J�|d9}|�||d�s�J�|�d�d}|�d�d}|j|d<|d}|�||�s�J�|d}|�||�s�J�|�d�d}|�d�d}|j|d<|d}|�||d�s�J�||�d�}|�||�d��|�d��t�}|�d�}|j|d<|d7}|�||d��sJ�dS)z*
102          Tests inplace w/ scalar
103  
104105  rfr-rSrU)r�N)r	rrrPrr6r)r%r/r0r]r&r&r'�test_5�sR
106  
107  
108  
109  
110  
111  
112  
113  
114  
115  
116  
117  zModuleTester.test_5cCs*|jdtd�}|�d�}|jdtd�}|j|d<|j}|jdtd�}|j|d<||7}||7}|�|||�s9J�|�|||�sCJ�|�|j|�||j��sQJ�|jdtd�}|jdtd�}|j|d<|j}|jdtd�}|j|d<||8}||8}|�|||�s�J�|�|||�s�J�|�|j|�||j��s�J�|jdtd�}|jdtd�}|j|d<|j}|jdtd�}|j|d<||9}||9}|�|||�s�J�|�|||�s�J�|�|j|�||j��s�J�|jdtd�}|jdtd�}|j|d<|j}|jdtd�}|j|d<||}||}dS)z)
118          Tests inplace w/ array
119  
120          r�)r4rf�����N)r	rrr+rr)r%r/r0r]rH�ar&r&r'�test_6"sT
121  
122  
123  
124  
125  
126  
127  
128  
129  zModuleTester.test_6c	Cs�|jdddtdgdddgdgdd�|jdddtdgdddgdgdd�f}dD]C}zt|j|�}WntyFtt|�}Ynwt|j|�}|d	|j�}||�}||�}|�|�	d�|�	d�|�|�|j
130  |j
131  �q.d	S)
132  zTests ufuncrSrr�rfr-rer*)�sqrt�log�log10�exp�	conjugateN)r
133  r\�getattrr"r#�fromnumericr�ninrPr�_mask)r%�d�f�uf�mf�args�ur�mrr&r&r'�test_7Vs **���zModuleTester.test_7c	Cs�|jgd�gd�d�}|�d|j|dd��|�d|j|gd�d��|j|gd�d	d
134  �\}}|�d|�|dks;J�|j|dd�<|j|dd�|jusNJ�|jgd�gd�d�}|�dd�}|j|dd�d	f<|�|j|dd�dd
g�|j|d	d�d|jus�J�|�dd
g|j|dd��|j|dd	d�\}}|�|dd
g�gd�}gd�gd�g}|�d�}|�|j|dd�d�|�|j|d|d�d�|�|�d�d|�d�g�}|�|�|d�tj�t�d��dd�|�|j|dd�t�d�dd�|�|j|d	d�|j|dd�|j|dd�dg�|�|j|d|d�d�|�|j|d|d�gd��|�|j|d	d�|j|dd�|j|dd�dg�|�	d�}gd�}	gd�gd�g}
135  |�
136  d�}gd�}|�|j|�||�dd�d�|�|j|�||	�dd�d�|�|j|�||�dd�d
�|�|�|j|�||�dd��d�|�||
137  �}
|�|�|
d�d�|�|j|
dd�gd��|�|j|
d	d�ddg�|�|j|
d|d�gd��dS)N)r_rSrUrZror*rUrr�)rSrSrUrS)�weightsr-)r��returnedrVrfr_)r�r�rS)rr-r-r-r-r)r-rrrrr-reg@)r�r�rZg(@g������
138  @)r_rSrUrZrVrY)rrr-r-rr)rr-r-r-r-r-)r_rS��X@r�rVg@rW)r_rSr�r�rVrY)
r
139  rP�averagerr�r	r8�addrr rrr)r%�ott�result�wts�w1�w2r/r0rjrkr~�m4�m5�zr&r&r'�test_99ysT
140  *$22
141  
142  $ zModuleTester.test_99cCs*|�d�}|j|dd�<|�ddd�}dS)Nr�rrrerfrdrc)r	rr�)r%r/r&r&r'�test_A�s
143  zModuleTester.test_AN)r)r)T)r))�__name__�
144  __module__�__qualname__r(rOrPr8�errstater^rnrr�r�r�r�r�r�r�r&r&r&r'r
s4"
145146  7
147  
148  
149  
150  
151  
152  
153  
154  *
155  
156  
157  
158  
159  
160  4
161  
162  3
163  
164  "
165  
166  -r�__main__zNfrom __main__ import ModuleTester 
167  import numpy
168  tester = ModuleTester(module)
169  zimport numpy.ma.core as module
170  )r�r�r-r�ztester.test_%i()r�z#%iz2..................................................zModuleTester.test_%i.__doc__zcore_current : z.3fz - )�timeit�	functoolsr�numpyr8r�numpy.core.fromnumericr$r��
numpy.testingrr\rr��
171  setup_base�	setup_cur�nrepeat�nloop�range�i�func�Timerr�curr�print�evalr&r&r&r'�<module>s0$
172  "�