3
���h �
@ s� d dl Z d dlZejjd3dd�jdd�Zejjd4dd�jdd�Zede ZdddgdddggZdddgdddggZ ej
j eed �Z ej
j ee d �Z
ej
j eed �Zejjd5dd6�jd
d
�Zejjd7dd8�jd
d
�Zede Zed kZed9k Zej
j eed �Zej
j eed �Zej
j eed �Zd:dd�Zd
ee eefdd�Zdd
dee eefdd�Zd
dee ee
eeeef
dd�Zedk�r�eej � eej � eej � eej � eej � eej!� edd dd� edd d dd� ed d dd� ed d dd� ed d dd� ed d!dd� ed d"dd� ed#d$ddd%� ed#d&ddd%� e"d'd( � e"d)� ed*d+d,d� e"d'd( � e"d-� ed.d+d,d� e"d'd( � e"d/� ed0d+dd� e"d'd( � e"d1� ed2d+d
d� dS );� N� � � � y �?TF)�mask�d g�������?� �� c C s� ddddg}ddddg}t d || f d
d � d d
� dD �}ddj|� }tj| |d�}t|j||��| } | dkr�tttjtj | ��d � d�}
nd}
t d||d| ||
||
f � d S )N�s�msu µs�nsr g @�@g ��.Ag e��Az
%s : %-50s : � )�endc S s g | ]}d t |d � �qS )z%ss,nm%ss,%sl,nm%sl� )�tuple)�.0�x� r �//tmp/pip-build-5_djhm0z/numpy/numpy/ma/bench.py�
<listcomp>+ s ztimer.<locals>.<listcomp>Zxyzz"from __main__ import numpy, ma, %s�,)Zstmt�setupg r z&%d loops, best of %d: %.*g %s per loop)
�print�join�timeit�Timer�min�repeat�int�numpy�floor�log10) r
�v�nloopZnrepZunitsZscalingZvarnamesr r �best�orderr r r �timer' s "
r& c C sn | j }tdd � td| � d \}}tdt� d| |d� td | � d \}}tdt� d| |d� d S )
N�-�2 z%s on small arrays�numpy.ma�nmxsz!%(module)s.%(funcname)s(%(data)s)z%11s)r" r# z%s on large arrays�nmxl)r) r* )r) r+ )�__name__r r& �locals) �funcr# �xsr* �xlr+ �funcname�module�datar r r �compare_functions_1v9 s r4 r c C sl t dd � t d| � d| d } }
tdt� |
|d� t d| � d | d } }
tdt� |
|d� d S )
Nr' r( z%s on small arraysznm%ssznumpy.maz!%(data)s.%(methodname)s(%(args)s))r" r# z%s on large arraysznm%sl)r r& r- ) �
methodname�args�varsr# �testr/ r* r0 r+ r3 �verr r r �compare_methodsF s r: c C sn | j } tdd � td| � d \} }
tdt� d| |d� td | � d \} }
tdt� d| |d� d S )
Nr' r( z%s on small arrays�numpy.ma� nmxs,nmysz!%(module)s.%(funcname)s(%(data)s)z%11s)r" r# z%s on large arrays� nmxl,nmyl)r; r< )r; r= )r, r r& r- )r. r# r8 r/ r* �ys�nmysr0 r+ �yl�nmylr1 r2 r3 r r r �compare_functions_2vR s rB �__main__Zraveli� )r# � conjugate�zZ transpose�
compressed� __getitem__�0z(0,0)z[0,-1]� __setitem__z0, 17)r# r8 z (0,0), 17r' r( z__setitem__ on small arraysz(nmxs.__setitem__((-1,0),numpy.ma.masked)z numpy.ma i' z__setitem__ on large arraysz(nmxl.__setitem__((-1,0),numpy.ma.masked)zwhere on small arraysz numpy.ma.where(nmxs>2,nmxs,nmys)zwhere on large arraysz numpy.ma.where(nmxl>2,nmxl,nmyl)���rJ rJ i' rJ i' g�������)r r r )#r r �random�uniformZreshaper/ r> Zzs�m1�m2�ma�arrayr* r? Znmzsr0 r@ ZzlZmaskxZmaskyr+ rA Znmzlr& r4 r: rB r, �sin�log�sqrt�multiply�divide�powerr r r r r �<module> sn
|