U
    SA{iû  ã                
   @   sº  d dl Z d dlZej ddd¡ dd¡Zej ddd¡ dd¡Zede  Zdd	d	gd	d	dggZdd	dgd	d	dggZ	ej
jeed
�Zej
jee	d
�Zej
jeed
�Zej ddd¡ dd¡Zej ddd¡ dd¡Zede  ZedkZedk Zej
jeed
�Zej
jeed
�Zej
jeed
�Zd4dd„Zdeeeefdd„Zdddeeeefdd„Zddeeeeeeeef
dd„Zedk�r¶eejƒ eejƒ eejƒ eejƒ eej ƒ eej!ƒ edddd� eddddd� ed ddd� ed!ddd� ed"d#dd� ed"d$dd� ed"d%dd� ed&d'dd	d(� ed&d)dd	d(� e"d*ƒ e"d+ƒ ed,d-dd� e"d*ƒ e"d.ƒ ed/d-dd� e"d*ƒ e"d0ƒ ed1d-dd� e"d*ƒ e"d2ƒ ed3d-dd� dS )5é    Néÿÿÿÿé   é   é   é   y              ð?TF)Úmaski'  éd   gš™™™™™é?gš™™™™™é¿Ú éô  c                 C   s¾   ddddg}ddddg}t d	|| f d
d� dd„ dD ƒ}dd |¡ }tj| |d�}t| ||¡ƒ| }	|	dkr”ttt t 	|	¡¡d ƒ dƒ}
nd}
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 ƒ ‘qS )z%ss,nm%ss,%sl,nm%slé   )Útuple)Ú.0Úx© r   ú2/tmp/pip-unpacked-wheel-m6chaliv/numpy/ma/bench.pyÚ
<listcomp>*   s     ztimer.<locals>.<listcomp>Zxyzz"from __main__ import numpy, ma, %sú,)ZstmtÚsetupg        r   z&%d loops, best of %d: %.*g %s per loop)
ÚprintÚjoinÚtimeitÚTimerÚminÚrepeatÚintÚnumpyÚfloorÚlog10)r   ÚvÚnloopZnrepZunitsZscalingÚvarnamesr   r   ÚbestÚorderr   r   r   Útimer&   s    "
ýr(   c           	      C   sl   | j }tdƒ t|› d�ƒ d\}}tdtƒ  d| |d� td| ƒ d\}}tdtƒ  d| |d� d S )	Nú2--------------------------------------------------ú on small arrays)únumpy.maÚnmxsú!%(module)s.%(funcname)s(%(data)s)ú%11s©r#   r$   ú%s on large arrays)r+   Únmxl©Ú__name__r   r(   Úlocals)	Úfuncr$   Úxsr,   Úxlr1   ÚfuncnameÚmoduleÚdatar   r   r   Úcompare_functions_1v8   s    r;   r   c	                 C   sn   t dƒ t | › d�ƒ d|› d�d }	}
tdtƒ  |
|d� t d|  ƒ d	| d }	}
tdtƒ  |
|d� d S )
Nr)   r*   ÚnmÚlr+   z!%(data)s.%(methodname)s(%(args)s)r/   r0   znm%sl)r   r(   r4   )Ú
methodnameÚargsÚvarsr$   Útestr6   r,   r7   r1   r:   Úverr   r   r   Úcompare_methodsE   s    rC   c                 C   sn   | j }tdƒ t|› d�ƒ d\}}tdtƒ  d| |d� t|› d�ƒ d\}}tdtƒ  d| |d� d S )	Nr)   r*   )r+   z	nmxs,nmysr-   r.   r/   z on large arrays)r+   z	nmxl,nmylr2   )r5   r$   rA   r6   r,   ÚysÚnmysr7   r1   ÚylÚnmylr8   r9   r:   r   r   r   Úcompare_functions_2vQ   s    rH   Ú__main__Zraveliè  )r$   Ú	conjugateÚzZ	transposeÚ
compressedÚ__getitem__Ú0z(0,0)z[0,-1]Ú__setitem__z0, 17)r$   rA   z	(0,0), 17r)   z__setitem__ on small arraysz(nmxs.__setitem__((-1,0),numpy.ma.masked)znumpy.ma   z__setitem__ on large arraysz(nmxl.__setitem__((-1,0),numpy.ma.masked)zwhere on small arraysz numpy.ma.where(nmxs>2,nmxs,nmys)zwhere on large arraysz numpy.ma.where(nmxl>2,nmxl,nmyl))r	   r
   r   )#r   r    ÚrandomÚuniformZreshaper6   rD   ZzsÚm1Úm2ÚmaÚarrayr,   rE   Znmzsr7   rF   ZzlZmaskxZmaskyr1   rG   Znmzlr(   r;   rC   rH   r3   ÚsinÚlogÚsqrtÚmultiplyÚdivideÚpowerr   r   r   r   r   Ú<module>   sˆ   	
   ÿ
   ÿ
    ü







