fume-manage-python.git

U  
P±då>ã@sôddlZddlmZddlZddlmZddlmmZddl    m
Z
ejZGdddZe dkrðdZd    eZd
\ZZeddD]nZd eZe ee¡ eed¡Ze e¡ZededeedeededddeddqdS)éN)Úreduce)Úfloat_)Ú build_err_msgc@sðeZdZddZd ddZd!ddZejd    d
ddZejd    d
d dZ    ejd    d
ddZ
ejd    d
ddZejd    d
ddZejd    d
ddZ ejd    d
ddZejd    d
ddZejd    d
ddZejd    d
ddZdS)"ÚModuleTestercCs
||_|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|_Wntk
rþ|jj|_YnXg|_dS©N)ÚmoduleÚallequalÚarangeÚarrayÚconcatenateÚcountÚequalÚfilledÚgetmaskZgetmaskarrayÚ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$úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/ma/timer_comparison.pyÚ__init__s@zModuleTester.__init__ÚTc
CsB| |¡}| |¡}| | |¡| |¡¡}    | |j||    d|¡}| |j||    d|¡}|jjdkr¢| t¡}t|t    j
r|jdkrd|t     |¡<nt     |¡r¢d}|jjdkrì| t¡}t|t    j
rÞ|jdkrÞd|t     |¡<nt     |¡rìd}z|j dkp|j dkp|j |j k}
|
sRt||g|d|j d|j d|d    d
}|
sRt||||}|    |jk    r||r||j||    d}t|tr|}
dg} n| ¡} |  ¡}
|  ¡} |
sødd|  d¡t| }t||g|d |f|d    d
}|
søt|WnBtk
r<}z"t||g||d    d
}t||W5d}~XYnXdS)zZ
        Assert that a comparison of two masked arrays is satisfied elementwise.
 
        ©ÚmaskÚOérr$z    
(shapes z, z
 mismatch)©ÚxÚy)ÚheaderÚnamesédgY@z
(mismatch %s%%)N)rrrrÚdtypeÚcharÚastyperÚ
isinstanceÚnpZndarrayÚsizeÚisnanÚshaperÚAssertionErrorrÚboolZravelÚallÚtolistrÚlenÚ
ValueError)r#Z
comparisonr-r.Úerr_msgr/Z
fill_valueÚxfZyfÚmZcondÚmsgÚvalZreducedÚmatchÚer$r$r%Úassert_array_compare0sd
 
 
 
 
 
$ÿü
ÿüz!ModuleTester.assert_array_comparecCs|j|j|||dddS)zH
        Checks the elementwise equality of two masked arrays.
 
        zArrays are not equal)r@r/N)rGr )r#r-r.r@r$r$r%Úassert_array_equalgsÿzModuleTester.assert_array_equalÚignore)r<cCs\t ddddtddddddddg¡}d    d
d
d
d
d
d    d
d
d
d
d
g}|j||d}|d
dS) ú!
        Tests creation
 
        çð?çÀç@ç@ç@ç$Àç$@ç@r+rr(N)r6r
Úpir)r#r-rBÚxmr$r$r%Útest_0os&zModuleTester.test_0c    Csbt ddddtddddddddg¡}t dd    ddd
dd    dddd    dg¡}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 |d|¡}| d¡|| d ¡ ¡sÊt|j    }|j
tdd|ksèt| |¡t |tdd|kstdD]J}||_    ||_    ||_    ||_    ||_    | |¡t |tdd|kstqdS)rJrKrLrMrNrOrPrQrRçgð¿gÀr+rr(g@µx¯DcSs||Srr$r,r$r$r%Ú<lambda>óz%ModuleTester.test_1.<locals>.<lambda>cSs||Srr$r,r$r$r%rWrX))éé)éécSs||Srr$r,r$r$r%rWrXN)r6r
rSrÚwhereZset_fill_valuerÚanyr:r9r7rrr>)    r#r-r.Úm1Úm2rTZymrAÚsr$r$r%Útest_1zs&&"
&zModuleTester.test_1c    Csìt ddddg¡}|j|ddddgd}|j|ddddgd}| |¡}t|t|t|dt|dkstt|d|dkstd|d<d|d<| ||¡d|dd<d|dd<|j|d<|j|dd<||d    d    <|j|d<| ddddgddddg¡|d    d    <| ddddgddddg¡|d    d    <t     d
¡d}| 
|d¡}| dd ddgt¡}t dd ddgt¡}|d|d|ddjdkstdddddg}|  |¡}|  |¡}||ksÌt|j |dd}||k    sètd    S)z2
        Tests conversions and indexing.
 
        r+r\rYrZrr(é    écNérKrRZhello)r)Úcopy)r6r
ÚstrÚreprÚtyper:rHrrr    rÚobjectr9r)    r#Úx1Zx2Zx3Úx4ÚnrBr`Úm3r$r$r%Útest_2s@
 
 
$$
 
zModuleTester.test_2cCsê| d¡}|j|d<| |d¡}| | ||g¡|¡s:t| | |¡ddddddddg¡s`t|j|ddd}| |ddddddddg¡|j|ddd}| ||¡s¬t|jddd}| ||¡sÊt| dd¡}| ||¡sætd    S)
z&
        Tests resize/repeat
 
        rYr\)érr+)r\r\r\r\©ÚaxisrZN)    r    rrrrr:rrrH)r#rlZy4Zy5Zy6Zy7Zy8r$r$r%Útest_3Ás
 
&zModuleTester.test_3c    CsN| d¡}t d¡}|j|dd<| ddd¡}| ddd¡}| t |d¡| |d¡¡s^t| t |dd¡| |dd¡¡st| t |     |d    ¡|     |d    ¡¡| ||¡¡s²t| t 
|     |d    ¡|     |d    ¡¡| 
||¡¡sât| d
ddddgt¡}|j|d<| |d    ddg¡}|d    d
ks&t|ddks8t|ddksJtdS) zB
        Test of take, transpose, inner, outer products.
 
        érer[r\rZrY)r\rr+r+rÚabcÚdefN) r    r6rÚreshaperrr:rrrrr
rj)r#r-r.Útr$r$r%Útest_4Õs&
 
 $
ÿ
ÿ
zModuleTester.test_4cCs| d¡}| d¡}| d¡}|j|d<|d7}| ||d¡sDt|d7}| ||d¡s`t| d¡}| d¡}|j|d<|d8}| ||d¡st|d8}| ||d¡s¶t| d¡d}| d¡d}|j|d<|d9}| ||d¡søt|d9}| ||d¡st| d¡d}| d¡d}|j|d<|d}| ||¡sVt|d}| ||¡spt| d¡d}| d¡d}|j|d<|d}| ||d¡s´t|| d¡}| || d¡¡| d¡ t¡}| d¡}|j|d<|d7}| ||d¡stdS)z*
        Tests inplace w/ scalar
 
        é
r\r+rKrM)rzN)r    rrr:rHrr4r)r#r-r.rTr$r$r%Útest_5ísR
 
 
 
 
 
 
 
 
 
 
 
zModuleTester.test_5cCs6|jdtd}| d¡}|jdtd}|j|d<|j}|jdtd}|j|d<||7}||7}| |||¡srt| |||¡st| |j| ||j¡¡s¢t|jdtd}|jdtd}|j|d<|j}|jdtd}|j|d<||8}||8}| |||¡st| |||¡s"t| |j| ||j¡¡s@t|jdtd}|jdtd}|j|d<|j}|jdtd}|j|d<||9}||9}| |||¡sªt| |||¡sÀt| |j| ||j¡¡sÞt|jdtd}|jdtd}|j|d<|j}|jdtd}|j|d<||}||}dS)z)
        Tests inplace w/ array
 
        rz)r2r\éÿÿÿÿN)r    rrr)rr:r)r#r-r.rTrBÚar$r$r%Útest_6"sT
 
 
 
 
 
 
 
 
zModuleTester.test_6c    Csê|jdddtdgdddgdgdd|jdddtdgdddgdgddf}dD]}zt|j|}Wntk
rtt|}YnXt|j|}|d    |j}||}||}| |     d¡|     d¡|¡| |j
|j
¡q\d    S)
zTests ufuncrKrr|r\r+r[r()ÚsqrtÚlogÚlog10ÚexpÚ    conjugateN)r
rSÚgetattrr r!ÚfromnumericrZninrHrZ_mask)r#ÚdÚfZufZmfÚargsÚurÚmrr$r$r%Útest_7Vs**ÿzModuleTester.test_7c    Cs(|jddddgddddgd}| d|j|dd¡| d|j|ddddgd    ¡|j|ddddgdd
\}}| d|¡|dkst|j|dd<|j|dd|jks¬t|jddddgddddgd}| dd¡}|j|dddf<| |j|ddddg¡|j|ddd|jks t| ddg|j|dd¡|j|ddd \}}| |ddg¡ddddddg}ddddddgddddddgg}| d¡}| |j|ddd¡| |j|d|dd¡| | d¡d| d¡g¡}| | |d¡tj     t d¡¡dd¡| |j|ddt d¡dd¡| |j|dd|j|dd|j|dddg¡| |j|d|d    d¡| |j|d|dddddddg¡| |j|dd|j|dd|j|dddg¡| 
d¡}ddddddg}    ddddddgddddddgg}
| d¡}ddddddg}| |j| ||¡ddd¡| |j| ||    ¡ddd¡| |j| ||¡ddd¡| |  |j| ||¡dd¡d¡| ||
¡} | | | d¡d¡| |j| ddddddddg¡| |j| ddddg¡| |j| d|dddddddg¡dS)NrVrKrMrRr+rr(rq)Úweights)rÚreturnedrNr\)rrrr[g@)rrrg(@g«ªªªªª
@rQgÀX@g@rO)r
rHZaverager:rrwr    r6Úaddrrrrr)r#ZottÚresultZwtsZw1Zw2r-r.r_r`rnZm4Zm5Úzr$r$r%Útest_99ysT 
*$2$2
 
$"zModuleTester.test_99cCs*| d¡}|j|dd<| ddd¡}dS)Nrtrer[r\rZrY)r    rrw)r#r-r$r$r%Útest_A§s
zModuleTester.test_AN)r'r'T)r')Ú__name__Ú
__module__Ú__qualname__r&rGrHr6ZerrstaterUrbrorsryr{r~rrrr$r$r$r%r s2"ÿ
7
 
 
 
 
 
 
 
*
 
 
 
 
 
4
 
3
 
"
 
-
rÚ__main__zNfrom __main__ import ModuleTester 
import numpy
tester = ModuleTester(module)
zimport numpy.ma.core as module
)rzrzr+rpztester.test_%i()rzz#%iz2..................................................zModuleTester.test_%i.__doc__zcore_current : z.3fz - )ZtimeitÚ    functoolsrÚnumpyr6rZnumpy.core.fromnumericr"rZ numpy.testingrrSrrZ
setup_baseZ    setup_curZnrepeatZnloopÚrangeÚiÚfuncÚTimerrÚcurrÚprintÚevalr$r$r$r%Ú<module>s*$