U
|
O ±dÚ' ã @ sÎ d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlZ d dl
|
m Z d dl m Z mZmZ d dlmZmZmZmZmZmZmZmZmZmZmZ d dl m Z d dl m Z m Z! d dl"mZ# G d d
|
d
|
Z$G d d d Z%G d d dZ&G dd dZ'G dd dZ(G dd dZ)G dd dZ*G dd dZ+G dd dZ,G dd d e-Z.G d d d Z/G d d d Z0G d!d" d"Z1G d#d$ d$Z2G d%d& d&Z3G d'd( d(Z4d)d* Z5G d+d, d,Z6G d-d. d.Z7G d/d0 d0Z8G d1d2 d2Z9G d3d4 d4Z:G d5d6 d6Z;G d7d8 d8Z<G d9d: d:Z=G d;d< d<Z>G d=d> d>Z?G d?d@ d@Z@G dAdB dBZAG dCdD dDZBG dEdF dFZCG dGdH dHZDdIdJ ZEG dKdL dLZFG dMdN dNZGG dOdP dPZHG dQdR dRZIG dSdT dTZJdS )Ué N)ÚDecimal)Úumath)ÚrandÚrandintÚrandn) Úassert_Ú assert_equalÚ assert_raisesÚassert_raises_regexÚassert_array_equalÚassert_almost_equalÚassert_array_almost_equalÚ assert_warnsÚassert_array_max_ulpÚ HAS_REFCOUNTÚIS_WASM)Úrational)ÚgivenÚ
|
strategies)Únumpyc @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )Ú
|
TestResizec C sº t ddgddgg¡}t ddddgddddgg¡}tt |d¡| t ddgddgddgddgg¡}tt |d¡| t dddgdddgdddgdddgg¡}tt |d¡| d S )Né é é é ©r r ©r r ©r r ©ÚnpÚarrayr Úresize©ÚselfÚAZAr1ZAr2ZAr3© r% úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/core/tests/test_numeric.pyÚ test_copies s "*zTestResize.test_copiesc C s´ t dddg¡}t ddddgddddgg¡}tt |d¡| t ddgddgddgddgg¡}tt |d¡| t dddgdddgdddgdddgg¡}tt |d¡| d S )Nr r r r r r r r" r% r% r& Ú test_repeats# s "*zTestResize.test_repeatsc C st t ddgddgg¡}t |d¡}t|t g ¡ t|j|j t |d¡}t|jd t |d¡}t|jd d S )Nr r r r ©r ©r r ©r r )r r r! r r ÚdtypeÚshape©r# r$ ZArr% r% r& Útest_zeroresize. s zTestResize.test_zeroresizec C sH t jddt jfgd}t |d¡}t|t d|j¡ t|j|j d S )Nr Úa©r, ©r r )r ÚzerosÚfloat32r! r r, r r. r% r% r& Útest_reshape_from_zero: s z!TestResize.test_reshape_from_zeroc C sB t jddt jd}d}tjtdd t j||d W 5 Q R X d S )Nr é
|
r1 )éöÿÿÿéÿÿÿÿÚnegative©Úmatch)Ú new_shape)r Úaranger4 ÚpytestÚraisesÚ
|
ValueErrorr! )r# r$ r< r% r% r& Útest_negative_resizeA s z TestResize.test_negative_resizec C s G dd dt j}t dg¡ |¡}tt |d¡|ks<ttt |d¡|ksTtt g ¡ |¡}tt |d¡|ks|td S )Nc @ s e Zd ZdZdS )z)TestResize.test_subclass.<locals>.MyArrayç ð?N)Ú__name__Ú
|
__module__Ú __qualname__Z__array_priority__r% r% r% r& ÚMyArrayH s rF r é r )r Úndarrayr ÚviewÚtyper! ÚAssertionError)r# rF Zmy_arrr% r% r& Ú test_subclassG s zTestResize.test_subclassN) rC rD rE r' r( r/ r5 rA rL r% r% r% r& r s r c @ s~ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zej dejejejejejejejejej ej ej g ¡d d Z ej d ej d d ejj!d!d"d#d d$e" #e" $d ¡¡ fd e" #e" $d ¡¡ fg¡d%d& Z%d'd( Z&ejj!e'd)d*d+d, Z(d-d. Z)d/d0 Z*d1d2 Z+d3d4 Z,d5d6 Z-d7d8 Z.d9d: Z/d;d< Z0d=d> Z1d?d@ Z2dAS )BÚTestNonarrayArgsc C sJ dddgdddgdddgg}dddg}dddg}t ||¡}t|| d S ) Nr r r r r rG é é )r Úchooser )r# ÚchoicesÚtgtr0 Úoutr% r% r& Ú test_chooseU s þ
|
|
z TestNonarrayArgs.test_choosec C s@ dddddddg}t |dd¡}dddddddg}t|| d S ) Nr8 rG r r r6 éüÿÿÿi÷ÿÿÿrO ©r Úclipr )r# ÚarrrS rR r% r% r& Ú test_clip_ s zTestNonarrayArgs.test_clipc C sN dddddgdddd d
|
gg}dddd d
|
gg}t jddg|dd }t|| d S ) Nr r r r r rG rN rO é é ©Úaxis)r Úcompressr ©r# rX rR rS r% r% r& Ú test_compresse s ÿz TestNonarrayArgs.test_compressc C sF dddddgdddddgg}t ddg¡}t j|dd}t|| d S )Nr r rO r r é r\ )r r Ú count_nonzeror r_ r% r% r& Útest_count_nonzerol s ÿz#TestNonarrayArgs.test_count_nonzeroc C sB dddgdddgg}t t t |¡t dddddd g¡k¡ d S )
|
Nr r r r rG rN é éx iÐ )r r ÚallZ
|
cumproductr ©r# r$ r% r% r& Útest_cumproducts s z TestNonarrayArgs.test_cumproductc C sD ddddgddddgd d
|
d d gg}t |¡}ddd g}t|| d S ) Nr r r r r rG rN rO rZ r[ r6 é )r Zdiagonalr )r# r0 rS rR r% r% r& Ú test_diagonalw s
|
|
|
þ
|
|
z TestNonarrayArgs.test_diagonalc C sÀ dddgdddgg}t t |¡dk t t t |d¡t d dd
|
g¡k¡ t t t |d¡t d d g¡k¡ tjd d:}t ddt¡ t t t g ¡¡ t |d j
|
tk W 5 Q R X d S )Nr r r r rG rN g @r g @g @ç @ç @T©ÚrecordÚalwaysÚ ) r r Úmeanrf r ÚwarningsÚcatch_warningsÚfilterwarningsÚRuntimeWarningÚisnanÚcategory©r# r$ Úwr% r% r& Ú test_mean s &$zTestNonarrayArgs.test_meanc C s* dddddddg}t tj|dd d
|
d S ) Nr r rG r6 éýÿÿÿéûÿÿÿg @r r\ g .@)r r Zptp©r# r0 r% r% r& Útest_ptp s zTestNonarrayArgs.test_ptpc C sD ddddgddddgd dddgg}d
|
d d g}t tj|d d| d S )Nr r r r rG rN rO r[ r6 rd ib iX r8 r\ )r r Úprod©r# rX rR r% r% r& Ú test_prod s
|
|
|
þ
|
zTestNonarrayArgs.test_prodc C sT dddgdddgddd gd
|
d d gg}ddddddddd d
|
d d g }t t |¡| d S ) Nr r r r rG rN rO rZ r[ r6 ri é )r r Zravel)r# r0 rR r% r% r& Ú
|
test_ravel s $ zTestNonarrayArgs.test_ravelc C s4 dddg}ddddddg}t |d¡}t|| d S )Nr r r )r Úrepeatr )r# r0 rR rS r% r% r& Ú test_repeat s
|
z TestNonarrayArgs.test_repeatc C sZ dddgdddgddd gd
|
d d gg}ddddddgddd d
|
d d gg}t t |d ¡| d S )Nr r r r rG rN rO rZ r[ r6 ri r )r rN )r r Úreshaper r% r% r& Ú test_reshape£ s $ z TestNonarrayArgs.test_reshapec C s\ ddddg}ddddg}t tj|d d
|
| t d ¡}tt| ¡ tj t | ¡ d d S ) Ngö(\Âõø?gÃõ(\"R@gffffff@g
|
@gù?g R@g@g @r )ZdecimalsrB )r r ZaroundÚfloat64r Ú
|
isinstanceÚround)r# rX rR Úsr% r% r& Ú
|
test_round¨ s
|
zTestNonarrayArgs.test_roundr, c C sx |d}t tt|t t tt|d t t tt|d dt tt|d tt|d d tt|d dd d S )Nr ©Úndigits)r r r Úintr )r# r, r r% r% r& Útest_dunder_round° s z"TestNonarrayArgs.test_dunder_roundz val, ndigitsiÿÿÿr8 zOut of range of int32©Úreason)Zmarksr c C s t t||tt |¡| d S ©N)r r r Úint32)r# Úvalr r% r% r& Útest_dunder_round_edgecases¾ s z,TestNonarrayArgs.test_dunder_round_edgecasesc C sÐ t ddd ¡}ttt|dt j tt|dddd ttt|ddt j tt|ddddd t d¡}ttt|dt j tt|dd ttt|ddt j tt|ddd d S )
|
Nçffffff@r6 éI é·ÿÿÿrl r iõ éþÿÿÿiô )r r r r r r Úint64)r# ÚfÚir% r% r& Útest_dunder_round_accuracyÈ s
|
z+TestNonarrayArgs.test_dunder_round_accuracyzgh-15896)r? r c C s, ddd }t tt |¡dt|d d S )Nr r6 r r )r r r r )r# r r% r% r& Útest_round_py_consistencyÕ s z*TestNonarrayArgs.test_round_py_consistencyc C s* ddddddg}t |d¡}t|d d S )Niøÿÿÿr| r8 r rN r6 r )r Z searchsortedr )r# rX rS r% r% r& Útest_searchsortedÚ s z"TestNonarrayArgs.test_searchsortedc C sR dddgdddgg}t t |¡dk t t |d¡dk t t |d¡dk d S )Nr r r r rG rN r )r r Úsizerg r% r% r& Ú test_sizeß s zTestNonarrayArgs.test_sizec C s dddgdddgdddggg}t t |¡jd t t t d¡¡jd t tjt d¡ddjd t tjt d¡d
|
djd t tjt d¡ddjd t t t d ¡g¡jd t tjt d ¡gddjd t tjt d ¡gddjd t tjt d ¡gd
|
djd d S ) Nr r r ©r r )r r r )r r r\ ©r r r8 ©r r )r r Zsqueezer- r3 rg r% r% r& Ú test_squeezeå s z TestNonarrayArgs.test_squeezec C s® dddgdddgg}t t |¡d t t |d¡t d d d g¡ t t |d¡t d
|
d
|
g¡ tjd d :}t d dt¡ tt t g ¡¡ t|d j
|
tk W 5 Q R X d S )Nr r r r rG rN g bép@Sû?r g ø?gɪo½ ê?Trm ro rp ) r r Ústdr rr rs rt ru r rv rw rx r% r% r& Útest_stdñ s zTestNonarrayArgs.test_stdc C s\ ddgddggddgddggg}ddgddggddgddggg}t |dd¡}t|| d S ) Nr r r rN r rG r rO )r Úswapaxesr )r# rR r0 rS r% r% r& Ú test_swapaxesü s z TestNonarrayArgs.test_swapaxesc C sJ dddgdddgddd gg}dgd
|
gd gg}t j|dd d }t|| d S )Nr r r r rG rN rO rZ r[ é rd T©r] Úkeepdims)r Úsumr )r# ÚmrR rS r% r% r& Útest_sum s þzTestNonarrayArgs.test_sumc C s< dddg}dddg}dddddg}t ||¡}t|| d S )Nr r rG r r )r Ztaker )r# rR Úindicesr0 rS r% r% r& Ú test_take s
|
|
|
zTestNonarrayArgs.test_takec C s* ddgddgddgg}t t |¡d d S )Nr r r r rG rN )r r Útrace)r# Úcr% r% r& Ú
|
test_trace s zTestNonarrayArgs.test_tracec C s@ ddgddgddgg}dddgdddgg}t t |d¡| d S )Nr r r r rG rN ©r r )r r Z transposer r% r% r& Útest_transpose s z TestNonarrayArgs.test_transposec C sÔ dddgdddgg}t t |¡d t t |d¡t d d d g¡ t t |d¡t d
|
d
|
g¡ tjd d :}t d dt¡ tt t g ¡¡ t|d j
|
tk W 5 Q R X t d dg¡}d|d< t t |¡d d S )Nr r r r rG rN gUUUUUU@r g @gv WUUå?Trm ro rp ù ð?g Ð?) r r Úvarr rr rs rt ru r rv rw )r# r$ ry ÚBr% r% r& Útest_var s zTestNonarrayArgs.test_varN)3rC rD rE rT rY r` rc rh rj rz r~ r r r
r r r> ÚmarkÚ parametrizer Úint8Úint16r r Úuint8Úuint16Úuint32Úuint64Úfloat16r4 r r ÚparamÚxfailÚmathÚceilÚlog10r r rK r r r¢ r¦ r¨ rª r° r² rµ r· r» r% r% r% r& rM S s`
|
ý
|
ÿû
|
|
rM c @ s e Zd Zdd ZdS )Ú TestIsscalarc C s t t d¡ t t t d¡¡ t t d¡ t t d¡ t t dg¡ t t d ¡ ddlm} t t |dd¡ dd lm} t t | ¡ d S )
|
NgÍÌÌÌÌÌ@i90 Fr r )ÚFractionrG é )ÚNumber)r r Úisscalarr¿ Z fractionsrË ZnumbersrÍ )r# rË rÍ r% r% r& Ú test_isscalar- s zTestIsscalar.test_isscalarN)rC rD rE rÏ r% r% r% r& rÊ , s rÊ c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
ÚTestBoolScalarc C s4 t j}t j}d}t|o||k t|o(||k d S )NZxyz©r ÚFalse_ZTrue_r )r# r Útr r% r% r& Ú test_logical= s
|
zTestBoolScalar.test_logicalc C sP t j}t j}t||B |k t||B |k t||B |k t||B |k d S r rÑ ©r# r rÓ r% r% r& Útest_bitwise_orD s z TestBoolScalar.test_bitwise_orc C sP t j}t j}t||@ |k t||@ |k t||@ |k t||@ |k d S r rÑ rÕ r% r% r& Útest_bitwise_andL s z TestBoolScalar.test_bitwise_andc C sP t j}t j}t||A |k t||A |k t||A |k t||A |k d S r rÑ rÕ r% r% r& Útest_bitwise_xorT s z TestBoolScalar.test_bitwise_xorN)rC rD rE rÔ rÖ r× rØ r% r% r% r& rÐ < s rÐ c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
Ú TestBoolArrayc C s t jdgd tddd
| _t jdgd tddd
| _t jdgd tddd
| _| j ¡ | _| j ¡ | _d| jd< d| jd < d| jd< d| jd < d S )
|
NTé) r1 r Fé* r r r ) r r ÚboolrÓ r ÚoÚcopyÚnmÚim©r# r% r% r& Ú setup_method^ s
|
|
|
zTestBoolArray.setup_methodc C s t | j ¡ t | j ¡ t | j ¡ t | j ¡ t | j ¡ t | j ¡ t | j ¡ t | j ¡ tdD ]v}tj dgd t
|
ddd
}d||< t t |¡ tj dgd t
|
ddd
}d||< t t |¡ t || qt tddd d
|
d d g D ]l}tj dgd t
|
d}d||< t t |¡d| d tj dgd t
|
d}d||< t t |¡ d| d qd S )Néù Fé r1 rO Tr[ ip iû iT i¥_ r7 iË z%r©Úmsg) r rÓ rf Úanyr rß rà Úranger r rÜ r Úlist)r# r ÚdÚer% r% r& Ú test_all_anyj s. zTestBoolArray.test_all_anyc C sÖ t | j | j t t | j ¡| j t t | j ¡| j t t | j¡| j t t | j¡ | j t t | j¡ | j t t | j ¡| j tj| j| jd t | j| j tj| j| jd t | j| j d S )N©rS ) r rÓ r r Úabsrß rà Z logical_notrÝ rá r% r% r& Útest_logical_not_abs
s z"TestBoolArray.test_logical_not_absc C sB t | j| jB | j t | j| jB | j t | j| jB | j t | j| jB | j tj| j| j| jd t | j| j t | j| j@ | j t | j| j@ | j t | j| j@ | j t | j| j@ | j tj| j| j| jd t | j| j t | j| jA | j t | j| jA | j t | j| jA | j t | j| jA | j tj| j| j| jd t | j| j t | j| j@ | j t | j | j@ d t | jd@ | j t | j d@ | j t | j| jB | j t | j | jB | j t | jdB | j t | j dB | j t | j| jA | j t | j | jA | j t | jdA | j t | j dA | j d S )Nrí FT)
|
r rÓ r r Ú
|
logical_orrÝ Ú logical_andÚ logical_xorrß rà rá r% r% r& Útest_logical_and_or_xor s< z%TestBoolArray.test_logical_and_or_xorN)rC rD rE râ rì rï ró r% r% r% r& rÙ ] s rÙ c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestBoolCmpc sx t jdt jd| _t j| jjtd| _t jdt jd| _t j| jjtd| _ d}t
|
dD ]T fddt
|
dD | j||d
< fd dt
|
dD | j||d
< |d7 }qXd}t
|
d
|
D ]T fd dt
|
d D | j||d
< fd dt
|
d D | j ||d
< |d 7 }qº| j ¡ | _ | j ¡ | _ t j| j | j< t j| j | j < | j ¡ | _| j ¡ | _t j| jd d d
| jd d d
< t j| jd d d
| j d d d
< t j | jdd d
| jdd d
< t j | jdd d
| j dd d
< t j| jdd d
| jdd d
< t j| jdd d
| j dd d
< | j ¡ | _d| jdd d
< | j ¡ | _d| jdd d
< | j ¡ | _| j ¡ | _| j| j d9 < | j| j d9 < t j | jdd d
| jdd d
< t j | jdd d
| j dd d
< t j | jdd d
| jdd d
< t j | jdd d
| j dd d
< d| jd d d
| jd d d
< d| jd d d
| j d d d
< d S )Nrä r1 é r é c s g | ]} d | @ qS ©r r% ©Ú.0Úx©r r% r& Ú
|
<listcomp>½ s z,TestBoolCmp.setup_method.<locals>.<listcomp>rZ c s g | ]} d | @ dkqS r+ r% rø rû r% r& rü ¾ s é c s g | ]} d | @ qS r÷ r% rø rû r% r& rü Â s r c s g | ]} d | @ dkqS r+ r% rø rû r% r& rü Ã s r r r Fç ð¿rN g )r Úonesr4 r r¡ rÜ Úefr rê Úedrè rÞ ÚnfÚndÚnanÚinffÚinfdÚinfÚefnonanÚednonanÚsignfÚsignd)r# r r% rû r& râ µ sP $$
|
$$
|
""$$"" $$$$ zTestBoolCmp.setup_methodc C s@ t dD ]0}t| j|d
dk| j|d
t| j|d
d dk| j|d
t| j|d
dk| j|d
t| j|d
dk | j|d
t| j|d
d dk| j|d
| j|d
dk}t|| j|d
| j|d
t | j|d
¡k}d| j|d
k}t|| t|| t| tj¡| tj¡ t| tj¡| tj¡ t| tj¡| tj¡ tt | j
|
|d
¡| j|d
tt | j
|
|d
¡| j|d
tt | j |d
¡| j|d
tt | j |d
¡| j|d
tt | j|d
¡| j|d
qd S )Nr r r )rè r r r r Ú
|
zeros_likerI r¾ Úastyperv r Úisfiniter Úisinfr Úsignbitr
|
©r# r ÚrZr2Zr3r% r% r& Ú
|
test_floatã s( "&$$("
|
|
$&&$zTestBoolCmp.test_floatc C s@ t dD ]0}t| j|d
dk| j|d
t| j|d
d dk| j|d
t| j|d
dk| j|d
t| j|d
dk | j|d
t| j|d
d dk| j|d
| j|d
dk}t|| j|d
| j|d
t | j|d
¡k}d| j|d
k}t|| t|| t| tj¡| tj¡ t| tj¡| tj¡ t| tj¡| tj¡ tt | j
|
|d
¡| j|d
tt | j
|
|d
¡| j|d
tt | j |d
¡| j|d
tt | j |d
¡| j|d
tt | j|d
¡| j|d
qd S )Nr r r )rè r rê r r r rI r¾ r rv r r r r r r r r r% r% r& Ú test_doubleý s( "&$$("
|
|
$&&$zTestBoolCmp.test_doubleN)rC rD rE râ r r r% r% r% r& rô ´ s .rô c @ sl e Zd Zdd Zdd Zejjeddejje
|
¡ dkddd d
|
Z ejjeddd d Z d d Z dS )Ú
|
TestSeterrc C s" t ¡ }t|tddddd d S )NÚwarnÚignore)ÚdivideÚinvalidÚoverZunder)r Úgeterrr Údict)r# Úerrr% r% r& Ú test_default s ýÿzTestSeterr.test_defaultc C s t ¡ t ¡ }t jdd}t||k t ¡ }t|d dk t jdd tt ¡ d dk t|d dk t jf | tt ¡ |k W 5 Q R X d S )NÚprint©r r Úraise)r r )r ÚerrstateÚseterrr r )r# r ÚoldÚnewr% r% r& Útest_set" s
|
zTestSeterr.test_setú no wasm fp exception supportr Zarmv5telz See gh-413.c
|
C sl t jddV tt t dg¡t dg¡ W 5 Q R X t jdd t dg¡t dg¡ W 5 Q R X d S )Nr! r rB ç r )r r" r ÚFloatingPointErrorr r# rá r% r% r& Útest_divide_err/ s
|
|
" zTestSeterr.test_divide_errc
|
s6 t ¡ }d _ztjdd^}t d¡ t jdd< t ddd g¡ t d g¡t d
|
g¡ t t
|
|d W 5 Q R X W 5 Q R X fd d }t jd d. t dd|g¡ t d g¡t d
|
g¡ W 5 Q R X t jd t |¡ t jd d t j d d
|
dd|gd W 5 Q R X t jd W 5 t |¡ `X d S )Nr Trm ro r r i N r rB r( c s6 j d7 _ | }tt|dk td|d k d S )Nr r r r )Úcalledr Úlen)ÚargsZ
|
extobj_errrá r% r& Úlog_errE s z'TestSeterr.test_errobj.<locals>.log_errr r )Zextobjr ) r Ú geterrobjr+ Ú seterrobjrr rs Ú simplefilterr" r r r, r )r# Ú olderrobjry r. r% rá r& Ú test_errobj9 s(
|
" "
|
|
zTestSeterr.test_errobjc C s| t ¡ }zbt tjtjd d g¡ t t dg¡¡ tdD ]}t tjtjd g¡ q<t t dg¡¡ W 5 t |¡ X d S )Nr rN i' ) r r/ r0 r ZUFUNC_BUFSIZE_DEFAULTZ ERR_DEFAULTrv r rè )r# r2 r r% r% r& Útest_errobj_noerrmaskX s ÿ ÿz TestSeterr.test_errobj_noerrmaskN)rC rD rE r r& r> r¼ Úskipifr ÚplatformÚmachiner* r3 r4 r% r% r% r& r s
|
r c @ s` e Zd Zdd Zdd Zejjeddej de
|
j d ¡d d
|
Z ejjeddd d Z d S )ÚTestFloatExceptionsc
|
C sp t |}z ||| tdd||f W nB tk
|
rj } z$tt| |¡dkd||f W 5 d }~X Y nX d S )NFz%Type %s did not raise fpe error '%s'.r z$Type %s raised wrong fpe error '%s'.)rJ r r) ÚstrÚfind)r# ÚfpeerrÚfloprú ÚyÚftypeÚexcr% r% r& Úassert_raises_fpel s
|
|
ÿ
|
ÿz%TestFloatExceptions.assert_raises_fpec C sT | ||||¡ | |||d |¡ | ||||d ¡ | |||d |d ¡ d S )Nr% )r@ )r# r; r< Zsc1Zsc2r% r% r& Úassert_op_raises_fpev s z(TestFloatExceptions.assert_op_raises_fper' r ÚtypecodeZAllFloatc C s t jddü t |¡}t |¡jdkrRt |¡}|jj}|j}|j }d}d}n@t
|
|dj } t | ¡}||jj}||j}||j }d}d}d}
|
d } t |¡sÌ| |d
|
d ||¡ | |d d ||¡ | |
|
d d ||d¡ | |
|
dd ||d¡ | |
|
dd ||| ¡ | |
|
dd | || ¡ | |
|
t j|d|d|j ¡ | |dd |d|d¡ | | dd |t j|t j¡ | | dd |d|d¡ | | dd |t j|t j¡ | | dd |t j|t j ¡ | | dd |d|t j¡ W 5 Q R X d S )Nr! ©rf r Ú underflowúdivide by zeror rp Úoverflowr c S s | | S r r% ©r0 Úbr% r% r& Ú<lambda>¢ ó z>TestFloatExceptions.test_floating_exceptions.<locals>.<lambda>c S s | | S r r% rG r% r% r& rI ¤ rJ c S s | | S r r% rG r% r% r& rI ¦ rJ r c S s | | S r r% rG r% r% r& rI ¨ rJ ç à?c S s | | S r r% rG r% r% r& rI ª rJ c S s | | S r r% rG r% r% r& rI ¬ rJ c S s | | S r r% rG r% r% r& rI ° rJ r c S s | | S r r% rG r% r% r& rI ² rJ c S s | | S r r% rG r% r% r& rI µ rJ c S s | | S r r% rG r% r% r& rI · rJ c S s | | S r r% rG r% r% r& rI º rJ c S s | | S r r% rG r% r% r& rI ½ rJ )r r" Z
|
obj2sctyper, ÚkindÚfinfoZ_macharZtinyÚmaxZepsrJ Úrealrv r@ ÚpowerZnexpr ) r# rB r> ÚfiZft_tinyZft_maxZft_epsrD Z divbyzeroZrtyperF r r% r% r& Útest_floating_exceptions s®
|
|
|
|
|
|
ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ ÿ
|
ÿ ÿz,TestFloatExceptions.test_floating_exceptionsc
|
C s t jdd}t d¡ tjddâ t dd¡ tt|d tdt |d j
|
k t d
|
¡t d
|
¡ tt|d td t |d j
|
k t tj ¡t tj ¡ tt|d tdt |d j
|
k t d¡t d¡ tt|d tdt |d j
|
k W 5 Q R X W 5 Q R X d S )NTrm ro r rC r r( rE r gu <ä7~r rF r8 r z invalid valuegYóøÂ n¥r rD ) rr rs r1 r r" r r r, r r9 Úmessager r )r# ry r% r% r& Ú test_warnings¿ s
|
z!TestFloatExceptions.test_warningsN)rC rD rE r@ rA r> r¼ r5 r r½ r Ú typecodesrR rT r% r% r% r& r8 k s
|
:r8 c @ sº e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d d d g¡ej
|
d ddg¡dd Z ej
|
ddge d¡e d¡ge dg¡e dg¡gg¡dd Zej
|
ddge d¡e d¡ge dg¡e de d¡ ¡ fg¡ge d ¡e d ¡ge d ¡e d ¡gg¡d d Zej
|
d!ee jd" d#d$d%d&d'd(eg ¡d)d* Zej jej d+¡ej
|
ddgejee jd" d#d%d&d'd(eg d,d-¡d.d/ Zd0d1 Zd2d3 Zd4d5 Z d6d7 Z d8S )9Ú TestTypesc C s~ t d¡}t d¡t d¡t d¡t d¡f\}}}}t d¡t d¡t d¡t d¡f\}}} }
|
t
|
d¡t d¡t d¡ } } } t d¡t d¡t d¡ }}}t|||t t j¡ t|||t t j¡ t|||t t j¡ t||| t t j¡ t|| | t t j ¡ t|| | t t j ¡ t|| | t t j ¡ t|||t t j¡ t|||t t j¡ t||| t t j¡ t|||t t j¡ t|||t t j¡ t|||t t j¡ t|||t t j¡ t||| t t j¡ t||
|
|t t j ¡ t||| t t j ¡ t||| t t j ¡ t|| |t t j
|
¡ t|| | t t j ¡ t|| |t t j ¡ t||| t t j¡ t||| t t j¡ t|t |g¡|t t j¡ t|t |g¡|t t j¡ t|t |g¡|t t j¡ t|t |g¡| t t j¡ t|t |g¡|t t j¡ t||
|
t |g¡t t j¡ t||t | g¡t t j¡ t|t d¡t |
|
g¡t t j ¡ t|| t | g¡t t j
|
¡ t|| t | g¡t t j
|
¡ t|t | g¡| t t j ¡ t|| t |g¡t t j ¡ t||t | g¡t t j¡ t|t d¡t | g¡t t j¡ t|t |g¡| t t j ¡ t|t |g¡|t t j¡ t|t |g¡|
|
t t j ¡ t|t |g¡| t t j ¡ t|t |g¡| t t j ¡ t|t |g¡|t t j¡ t|t | g¡|t t j ¡ d S )Nr r8 y @)r Úbool_r¾ r¿ r r rÀ rÁ r rà r4 r Ú
|
longdoubleÚ complex64Ú
|
complex128Ú clongdoubler r, r )r# Z promote_funcrH Úi8Zi16Zi32Zi64Úu8Úu16Úu32Zu64Zf32Zf64ZfldZc64Zc128Zcldr% r% r& Úcheck_promotion_casesÔ sv
|
,,""
|
ÿ
|
ÿ
|
ÿ
|
ÿ
|
ÿz TestTypes.check_promotion_casesc C s dd }| |¡ t ddg¡tjddgtjdfD ]d}d| }t|jt d ¡d
|
|j t d¡| }t|jt tj¡d
|
|j t d¡| }t|jt d ¡d
|
|j t d¡| }t|jt d ¡d
|
|j t d¡| }t|jt d ¡d
|
|j d | }t|jt d¡d
|
|j t
|
d ¡| }t|jt tj
|
¡d
|
|j t d ¡| }t|jt d¡d
|
|j t d ¡| }t|jt d¡d
|
|j q4d S )Nc S s t | |¡jS r )r Úaddr, rG r% r% r& Úres_type s z)TestTypes.test_coercion.<locals>.res_typeTFr{ r r1 gX9´Èv¾ó?Úf8z array type %sÚf4Úf2y X9´Èv¾ó?Úc16Zc8) r` r r r¾ r r, rX r r4 rÄ r[ rZ rY )r# rb r0 rH r% r% r& Ú test_coercion s2
|
(ÿÿzTestTypes.test_coercionc C s( | tj¡ tt d ¡t d ¡k d S r )r` r Ú result_typer r, rá r% r% r& Útest_result_typeT s zTestTypes.test_result_typec C s t t dd¡t d¡ t t dd¡t d¡ t t dd¡t d¡ t t dd¡t d¡ t t dd¡t d¡ t t dd¡t d¡ t t dd¡t d ¡ t t d
|
d ¡t d ¡ t t dd¡t d ¡ t t d d
|
¡t d ¡ t t d d¡t d ¡ t t d d ¡t d ¡ t t dd¡t d¡ t t dd¡t d¡ t t dd¡t d¡ t t dd¡t d¡ d S )Nú<i8r\ ú>i8z>U16ÚU21z<U16z<S5z<U8ZU8z>S5z>U8z<U5z>U5z<M8ZM8z>M8z<m8Úm8z>m8)r r Ú promote_typesr, rá r% r% r& Útest_promote_types_endianX s z#TestTypes.test_promote_types_endianc
|
C sÌ dddddg}d}t t¡}|D ]2}t |¡}t ||¡s>tt ||¡|ks tq |D ]@}t |¡}t ||¡rvtt t¡ t ||¡ W 5 Q R X qXt d¡}t ||¡s´tt ||¡|ksÈtd S )Nr¾ r¿ r r rÜ Z BHILQPFDGmMfVÚdouble) r r, r Úcan_castrK rn r> r? Ú TypeError)r# Z valid_typesZ invalid_typesZ rational_dtZ numpy_dtypeZ double_dtr% r% r& Ú#test_can_cast_and_promote_usertypesn s
|
|
|
|
z-TestTypes.test_can_cast_and_promote_usertypesÚswaprp Ú string_dtypeÚUÚSc C s¦ |dkrdd }nt j}|}t|d|t |d ¡ t|d|t |d ¡ t|d|t |d ¡ t|d
|
|t |d ¡ t|d |t |d ¡ t|d |t |d ¡ t|d|t |d ¡ t|d|t |d ¡ t|d|t |d ¡ t|d|t |d ¡ t|d|d t |d ¡ t|d|d t |d ¡ t|d|d t |d ¡ t|d|d t |d ¡ t|d|d t |d ¡ t|d|d t |d ¡ t|d
|
|d t |d ¡ t|d
|
|d t |d ¡ t|d |d t |d ¡ t|d |d t |d ¡ t|d |d t |d ¡ t|d |d t |d ¡ t|d|d t d¡ d S )Nrt c S s t || ¡S r )r rn rG r% r% r& rI rJ z6TestTypes.test_promote_types_strings.<locals>.<lambda>rÜ Ú5rH Ú4Úu1Ú3Úu2Úu4Ú10r] Z20Úi1Úi2Ú6Úi4Z11r\ Z21Ú1Ú30ÚO)r rn r r, )r# rt ru rn rw r% r% r& Útest_promote_types_strings s6
|
z$TestTypes.test_promote_types_stringsÚdtype1Údtype2ZV6ÚV10)Úname1r\ )Zname2r\ c C s& t t¡ t ||¡ W 5 Q R X d S r )r> r? rr r rn ©r# r r r% r% r& Útest_invalid_void_promotion« s z%TestTypes.test_invalid_void_promotionr r\ zi8,i8zi8,>i8úi4,i4c C s t ||¡|kstd S r )r rn rK r r% r% r& Útest_valid_void_promotion´ s z#TestTypes.test_valid_void_promotionr, ÚAllzi,iZ10iÚS3ZS100ÚU3ZU100c C s ddi}t j||d}t ||¡}|j|jks2t| ¡ }|jrDd S t ||¡}|jdkrj|jd ksxtn|j|ksxt|jstd S )Nr ©Úmetadatarv )r r, rn r rK Ú newbyteorderÚisnativeÚchar)r# r, r Úresr% r% r& Ú%test_promote_identical_types_metadata¾ s
|
z/TestTypes.test_promote_identical_types_metadataz)ignore:Promotion of numbers:FutureWarningr )r c C s ddi}ddi}t j||d}t j||d}zt ||¡}W n tk
|
rR Y dS X |jdksr|jdk sr|jdkr|jdksØtnV||kr||ksØtn@||krÊt ||j ¡|kr¾|jdk qØ|j|k n|jdksØt|
|
¡ }|j|ksîtt ||¡}||kst|j|jkstdS )a¥ Metadata handling in promotion does not appear formalized
|
right now in NumPy. This test should thus be considered to
|
document behaviour, rather than test the correct definition of it.
|
|
This test is very ugly, it was useful for rewriting part of the
|
promotion, but probably should eventually be replaced/deleted
|
(i.e. when metadata handling in promotion is better defined).
|
r r r NZUSVr% ) r r, rn rr r Únamesr- r rK rL r )r# r r Z metadata1Z metadata2r Zres_bsr% r% r& Útest_promote_types_metadataÚ s, z%TestTypes.test_promote_types_metadatac C sö t t tjtj¡ t t tjt¡ t t tt¡ t t dd¡ t t dd¡ t t dd¡ t t ddd¡ t t ddd¡ t t ddd ¡ t t d
|
dd ¡ t t d
|
dd ¡ t t dd d ¡ t t dd d ¡ t t ddd ¡ t t ddd¡ t t dd¡ t t dd¡ t t dd¡ t t dd¡ t t dd¡ t t dd¡ t t dd¡ t t dd¡ t t dd¡ t t dd¡ t t dd ¡ t t dd ¡ t t d d¡ t t d d¡ t t d d ¡ t t d d¡ t t dd¡ t t dd¡ t t dd!¡ t t dd ¡ t t dd¡ t t dd¡ t t dd"¡ t t dd#¡ t t dd#¡ t t dd$¡ t t dd%¡ t t dd"¡ t t dd&¡ t t dd'¡ t t dd(¡ t t dd)¡ t t d d"¡ t t d d#¡ t t d d*¡ t t d d%¡ t t dd+¡ t t dd&¡ t t dd,¡ t t dd(¡ tt tjdd tt tjd d t tjtjtjd- d S ).Nr\ rc rd r ZS11Únorj rk Úequivz<i4Úsafez>i4Ú same_kindz>u4ÚunsaferÜ ZS5ZS4rH r rz ZS2r| r} ZS10ZS9r] ZS20ZS19r r ZS6ZS21ZU4r ZU2ZU5ZU10ZU9ZU20ZU19ZU6ZU11rl )Úfrom_Úto)
|
r r rq r r r ÚcomplexÚfloatr rr rá r% r% r& Ú test_can_cast st zTestTypes.test_can_castc C sæ t t dd¡ t t dd¡ t tjdddd t tjdddd t t ddg¡ t tjddgdd t tjddgdd t t dd g¡ t tjdd gdd t t dd
|
g¡ t tjdd
|
gdd d S ) Nr r zi4,i2r ©Úcastingr ©Úf1r r )r¨ r )r¨ z(2,3)i4©r r rq rá r% r% r& Ú"test_can_cast_simple_to_structured\ s z,TestTypes.test_can_cast_simple_to_structuredc C sÒ t t dgd¡ t tjdgddd t tjdgddd t tjdddd t t ddgfgd¡ t tjddgfgddd t t d gd¡ t tjd gddd t tjd
|
gddd d S ) Nr§ r r r¥ r r r¨ )rú r )Úf0z(3,)i4)r« r r÷ r© rá r% r% r& Ú"test_can_cast_structured_to_simplel s ÿz,TestTypes.test_can_cast_structured_to_simplec C sº t jd t jd D ]b}t |¡}tt |j|¡ tt |j|¡ tt |jd |¡ tt |jd |¡ qt jd D ]2}t |¡}tt |j|¡ tt |j|¡ qd S )Nr Zuintr r£ )r ÚsctypesÚiinfor rq ÚminrN rM )r# ÚdtÚiirQ r% r% r& Útest_can_cast_values s
|
|
z TestTypes.test_can_cast_valuesN) rC rD rE r` rg ri ro rs r> r¼ r½ r r r, r r r ré rU r r Zslowrt Ú itertoolsÚproductr r¤ rª r¬ r² r% r% r% r& rV Ó sZ J6 # þÿ
|
|
ÿüÿ
|
ÿÿ
|
|
ÿýÿ.MrV c @ s e Zd ZdS )Ú
|
NIterErrorN©rC rD rE r% r% r% r& rµ s rµ c @ s& e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d e e g¡ej
|
d d gddg¡dd Z ej
|
d ddddg¡dd Zej
|
d dgdddd d d d d d!d"g fd#ddd d d d d d!d"g fd$d%d&d'd(d)gfd*d%d&d'd(d)gfe d+¡d,d-gfg¡ej
|
d.d/dg¡d0d1 Zd2d3 Zd4d5 Zd6d7 Zd8S )9Ú TestFromiterc C s dd t dD S )Nc s s | ]}|d V qdS )r Nr% rø r% r% r& Ú <genexpr> s z'TestFromiter.makegen.<locals>.<genexpr>rd )rè rá r% r% r& Úmakegen s zTestFromiter.makegenc C sx t | ¡ t j¡}t | ¡ t j¡}t | ¡ t¡}t|jt t j¡k t|jt t j¡k t|jt t¡k d S r )r Úfromiterr¹ r r r£ r r, )r# Zai32Zai64Úafr% r% r& Ú
|
test_types s zTestFromiter.test_typesc C sz t t| ¡ ¡}t | ¡ t¡}t | ¡ td¡}tt|t|k tt|dk tt t j| ¡ tt|d d S )Né r6 )
|
r r ré r¹ rº r r r, r r@ ©r# Úexpectedr0 Za20r% r% r& Ú test_lengths s
|
ÿzTestFromiter.test_lengthsc C sl t t| ¡ ¡}t | ¡ t¡}t | ¡ td¡}tt j||kdd tt j||d d
kdd d S )Nr½ r r\ )r r ré r¹ rº r r Zalltruer¾ r% r% r& Ú test_values¨ s
|
zTestFromiter.test_valuesc c s, t |D ] }||kr td| |V qd S )Nzerror at index %s)rè rµ )r# ÚnZeindexrë r% r% r& Ú load_data¯ s zTestFromiter.load_datar, ÚcountÚ error_index)r6 rG )r6 r[ c C s6 | ||¡}t t¡ tj|||d W 5 Q R X d S )N)r, rÄ )rà r> r? rµ r rº )r# rÄ rÅ r, Úiterabler% r% r& Ú test_2592· s zTestFromiter.test_2592rw ZS0ÚV0ZU0c C s, t jtdd tjg |d W 5 Q R X d S )NzMust specify lengthr: r1 )r> r? r@ r rº )r# r, r% r% r& Útest_empty_not_structuredÀ s z&TestFromiter.test_empty_not_structuredÚdatarê r r r r rG rN rO rZ r[ r
zi,O©r r )rG r ©r r )r[ rZ ©rN rO Z2i)r
rÌ )©r r r )r r rG ))r r r )rG r r Ú length_hintr c sR t |¡} d G fddd}t j| |d}t j |d}t|| d S )Néd c s$ e Zd ZfddZ fddZdS )z?TestFromiter.test_growth_and_complicated_dtypes.<locals>.MyIterc s S r r% rá )rÏ r% r& Ú__length_hint__Ø s zOTestFromiter.test_growth_and_complicated_dtypes.<locals>.MyIter.__length_hint__c s t S r ©Úiterrá )rÊ r% r& Ú__iter__Ü s zHTestFromiter.test_growth_and_complicated_dtypes.<locals>.MyIter.__iter__N©rC rD rE rÑ rÔ r% ©rÊ rÏ r% r& ÚMyIterÖ s r× r1 )r r, rº r r )r# r, rÊ rÏ r× r r¿ r% rÖ r& Ú"test_growth_and_complicated_dtypesÇ s
|
z/TestFromiter.test_growth_and_complicated_dtypesc C s> G dd d}t j| dd}|jdks,t|jdks:td S )Nc @ s e Zd Zdd Zdd ZdS )z.TestFromiter.test_empty_result.<locals>.MyIterc S s dS ©Nr6 r% rá r% r% r& rÑ æ s z>TestFromiter.test_empty_result.<locals>.MyIter.__length_hint__c S s t g S r rÒ rá r% r% r& rÔ é s z7TestFromiter.test_empty_result.<locals>.MyIter.__iter__NrÕ r% r% r% r& r× å s r× rê r1 r) )r rº r- rK r, )r# r× r r% r% r& Útest_empty_resultä s z TestFromiter.test_empty_resultc C s8 d}t jt|d tjdddgdtd W 5 Q R X d S )Nz>iterator too short: Expected 10 but iterator had only 3 items.r: r r r r6 )rÄ r, )r> r? r@ r rº r )r# ræ r% r% r& Útest_too_few_itemsð s z TestFromiter.test_too_few_itemsc C sn t t¡ tjdd dgtd W 5 Q R X dd tdD }t t¡ tj|t tdf¡d W 5 Q R X d S )Nr r r1 c s s | ]
|
}d V qdS )©r r r Nr% )rù r r% r% r& r¸ ú s z7TestFromiter.test_failed_itemsetting.<locals>.<genexpr>rG r ) r> r? rr r rº r rè r@ r, )r# rÆ r% r% r& Útest_failed_itemsettingõ s
|
z$TestFromiter.test_failed_itemsettingN)rC rD rE r¹ r¼ rÀ rÁ rà r> r¼ r½ r ÚobjectrÇ rÉ r r, rØ rÚ rÛ rÝ r% r% r% r& r· s2
|
ÿúÿ r· c @ s¤ e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dd Z
|
dd Z dd Z dd Z dd Zdd Zdd Zd d Zd d Zejjed!d"d#d$ Zd%S )&Ú TestNonzeroc C sê t t t g ¡¡d t t tjg dd¡d t t t g ¡¡g f t t t dg¡¡d t t tjdgdd¡d t t t dg¡¡g f t t t dg¡¡d t t tjdgdd¡d t t t dg¡¡dgf d S ©Nr ú?r1 r )r r rb r Únonzerorá r% r% r& Útest_nonzero_trivialÿ s z TestNonzero.test_nonzero_trivialc C s¾ t t t d¡¡d t t tjddd¡d tt t t t d¡¡g f W 5 Q R X t t t d¡¡d t t tjddd¡d tt t t t d¡¡dgf W 5 Q R X d S rà )r r rb r r ÚDeprecationWarningrâ rá r% r% r& Útest_nonzero_zerod s
|
"
|
z TestNonzero.test_nonzero_zerodc C s t dddddddg¡}tt |¡d tt |¡d tt |¡ddddgf t jd d
|
d d d gddddgd}tt |d ¡d tt |d ¡d tt |d ¡d tt |d ¡d tt |d ¡dddgf tt |d ¡ddddgf d S )Nr r r r8 rZ r r rN )r r r| r{ )r r r rO )r r r r )r8 r r r )r rO r r )r0 r )rH r )r´ r )rê r\ r1 r0 rH r´ rê )r r r rb râ ©r# rú r% r% r& Útest_nonzero_onedim s
|
ÿ z TestNonzero.test_nonzero_onedimc C sV t dddgdddgg¡}tt | d¡¡d tt | d¡¡d tt | d¡¡d tt | d¡¡d tt |¡dddgdddgf t d¡}tt | d¡¡d tt | d¡¡d tt | d¡¡d tt | d¡¡d tt |¡dddgdddgf t jd d
|
d gd d d
|
gd
|
dd ggddgd}tt |d ¡d tt |d ¡d tt |d ¡ddddgddddgf tt |d ¡dddddgdddddgf t|d jj j
|
tt |d j¡d tt |d j¡d tt |d j¡ddddgddddgf tt |d j¡dddddgdddddgf d S )Nr r r r r r r r\ ©r r ©r r )r ri ©r r r¶ )r rG )r0 rd )rH rz r1 r0 r rH rG ) r r r rb r râ Zeyer ÚTÚflagsÚalignedræ r% r% r& Útest_nonzero_twodim( s6
|
þþ(,*z TestNonzero.test_nonzero_twodimc
|
C sº t dD ]¬}tjdtd}d||d d
< tt |¡d t |d| d¡ tjdtd}d|d| d|
< d|d|d < tt |¡d t t d| d| ¡d|d gf¡ qd S ) Nr½ éÈ r1 Tr i r6 r )rè r r3 rÜ r râ r= Z concatenate)r# r r´ r% r% r& Ú test_sparseE s "&ÿzTestNonzero.test_sparsec C s¤ G dd dt j}|t jfD ]}tddD ]r}ttdd| }t t |¡¡ |¡ |¡}t |¡| ¡ fD ],}|D ]"}t t
|
|t jk t |j j qvqnq*q d S )Nc @ s e Zd ZdS )z'TestNonzero.test_return_type.<locals>.CNr¶ r% r% r% r& ÚCS s rñ r r r ) r rH rè Útupler= r r rI râ r rJ rì Ú writeable)r# rñ rI r r- rú ZnzxZnzx_ir% r% r& Útest_return_typeR s z TestNonzero.test_return_typec C sÈ t dddddgdddddgg¡}t dddddg¡}tt j|dd| t ddg¡}tt j|dd| ttt j|dd ttt j|d d tt jt j|dd ttt j|t dgdgg¡d d S )
|
Nr r rO r r ra r\ rê Zfoo)r r r rb r r@ rr Ú AxisError)r# r¯ r¿ r% r% r& Útest_count_nonzero_axis_ s " ÿz#TestNonzero.test_count_nonzero_axisc C s& d}dd }t jd D ]}|t |¡jf }|dkrT|dkrvt jd|d}t jd |d}|d
|
|d < |d
|
|d < n4t d gd ¡}| d¡}d|d < d|d < | |¡}t jdd
|
d
|
gt j d}|t j
|
|d
|
d||d t jd d d
|
gt j d}|t j
|
|d d||d t d¡}t t j
|
|dd||d t t j
|
|d d||d t t
|
|¡||d |dkrt t d ¡gd ¡ d¡}t jd
|
d
|
d
|
gt j d}|t j
|
|d
|
d||d t jd
|
d
|
gt j d}|t j
|
|d d||d t d
|
¡}t t j
|
|dd||d t t j
|
|d d||d t t
|
|¡||d qd S )NzMismatch for dtype: %sc S s$ t | j|j|d t | ||d d S )N©Úerr_msg)r r, )r0 rH rø r% r% r& Úassert_equal_w_dtu s zITestNonzero.test_count_nonzero_axis_all_dtypes.<locals>.assert_equal_w_dtr ÚVÚMr£ r1 r r ré r¶ z
|
1970-01-01r[ z
|
1970-01-12r r\ r÷ rè rN rÌ ) r rU r, Únamer3 rÿ r r r Úintprb r Zvoid)r# ræ rù r° rø r¯ r r¿ r% r% r& Ú"test_count_nonzero_axis_all_dtypeso s~
|
|
|
ÿ ÿ
|
ÿ ÿ
|
ÿ ÿ ÿ
|
ÿ ÿ
|
ÿz.TestNonzero.test_count_nonzero_axis_all_dtypesc C s ddl m}m} d}d}d}tj d¡}|jdd|d }| t¡}t t
|
|D ]J} ||| D ]:}
|
||
|
D ],} t tj || d
|
tj || d
|
|| f d qhq\qNd S ) Nr )Ú combinationsÚ permutations©r r r r )rG rG rG rG zMismatch for axis: %siÒ iÿÿÿrÐ ©r¡ r\ r÷ ) r³ rÿ r r ÚrandomZ RandomStater r rÞ rè r, r rb ) r# rÿ r r] r¡ ræ Úrngr¯ r ÚlengthZcomboÚpermr% r% r& Ú"test_count_nonzero_axis_consistent± s
|
ýz.TestNonzero.test_count_nonzero_axis_consistentc C s8 t dddgdddgg¡}tt j|dd| t¡ d S )Nr r r% r\ )r r r rb r rÜ r} r% r% r& Ú test_countnonzero_axis_emptyÈ s z(TestNonzero.test_countnonzero_axis_emptyc C s t ddddgddddgddddgg¡}tt j|ddd ddddgg tt j|ddd dgdgdgg tt j|dd
|
d gg d S ) Nr r r rG rO r[ r Tr¬ ©r rN )r r r rb r} r% r% r& Útest_countnonzero_keepdimsÌ s
|
|
þ ÿÿÿz&TestNonzero.test_countnonzero_keepdimsc C s@ t dddgdddgg¡}dddgdddgg}t| ¡ | d S )Nr r r rN r )r r r râ )r# r¯ rR r% r% r& Útest_array_method× s z TestNonzero.test_array_methodc C sT t jt ddg¡dgtd}ttt j| G dd d}ttt jt | g¡ d S )Nr r r r1 c @ s e Zd Zdd ZdS )z;TestNonzero.test_nonzero_invalid_object.<locals>.BoolErrorsc S s t dd S )Nz Not allowed)r@ rá r% r% r& Ú__bool__å s zDTestNonzero.test_nonzero_invalid_object.<locals>.BoolErrors.__bool__N)rC rD rE r r% r% r% r& Ú
|
BoolErrorsä s r )r r rÞ r r@ râ )r# r0 r r% r% r& Útest_nonzero_invalid_objectß s z'TestNonzero.test_nonzero_invalid_objectc C s G dd d}G dd d}t d| g¡}ttt j| t dg| gg¡}ttt j| t d| g¡}ttt j| t dg| gg¡}ttt j| d S )Nc @ s e Zd ZdZdd ZdS )zATestNonzero.test_nonzero_sideeffect_safety.<locals>.FalseThenTrueFc S s z
|
| j W ¢S d| _ X d S )NT©Ú_valrá r% r% r& r î s
|
zJTestNonzero.test_nonzero_sideeffect_safety.<locals>.FalseThenTrue.__bool__N©rC rD rE r r r% r% r% r& Ú FalseThenTrueì s r c @ s e Zd ZdZdd ZdS )zATestNonzero.test_nonzero_sideeffect_safety.<locals>.TrueThenFalseTc S s z
|
| j W ¢S d| _ X d S )NFr rá r% r% r& r ö s
|
zJTestNonzero.test_nonzero_sideeffect_safety.<locals>.TrueThenFalse.__bool__Nr r% r% r% r& Ú TrueThenFalseô s r TF)r r r Ú RuntimeErrorrâ )r# r r r0 r% r% r& Ú test_nonzero_sideeffect_safetyê s z*TestNonzero.test_nonzero_sideeffect_safetyc C sZ t jddd}|jjst|d jjr*tt |¡ |jjs@tt |¡ |jjsVtd S )NrG zi1,i8,i8r1 re )r r3 rì rí rK râ rb ©r# rX r% r% r& Ú'test_nonzero_sideffects_structured_void
|
s
|
|
z3TestNonzero.test_nonzero_sideffects_structured_voidc C sz G dd d}t |dgd ¡}ttt j| t |dgd ¡}ttt j| t |dggd ¡}ttt j| d S )Nc @ s e Zd Zdd Zdd ZdS )z<TestNonzero.test_nonzero_exception_safe.<locals>.ThrowsAfterc S s
|
|| _ d S r )Ú
|
iters_left)r# Úitersr% r% r& Ú__init__ s zETestNonzero.test_nonzero_exception_safe.<locals>.ThrowsAfter.__init__c S s$ | j dkrtd| j d8 _ dS )Nr zcalled `iters` timesr T)r r@ rá r% r% r& r s
|
zETestNonzero.test_nonzero_exception_safe.<locals>.ThrowsAfter.__bool__N)rC rD rE r r r% r% r% r& Ú ThrowsAfter s r rG r6 r« )r r r r@ râ )r# r r0 r% r% r& Útest_nonzero_exception_safe s z'TestNonzero.test_nonzero_exception_safezwasm doesn't have threadsr c s¬ ddl m} t dg¡}t d|fg¡}t d|fgd ¡}tjjdd |¡d d
df dd |d
|
d fd d tdD }|D ] }| ¡ q j|ks¨t d S )Nr )ÚThreadPoolExecutor)rp rc rp r )i r r c S s | ¡ d S r )râ )rX r% r% r& ÚfuncC s z6TestNonzero.test_structured_threadsafety.<locals>.funcrZ )Ú max_workersc s g | ]} ¡qS r% )Zsubmit)rù Ú_©rX r Ztper% r& rü G s z<TestNonzero.test_structured_threadsafety.<locals>.<listcomp>r6 )
|
Úconcurrent.futuresr r r, r ÚuniformrI rè ÚresultrK )r# r r° Zfuturesr r% r! r& Ú test_structured_threadsafety7 s
|
|
z(TestNonzero.test_structured_threadsafetyN)rC rD rE rã rå rç rî rð rô rö rþ r r r
|
r r r r r r> r¼ r5 r r% r% r% r% r& rß þ s$ B !rß c @ s e Zd Zdd Zdd ZdS )Ú TestIndexc C s t ddd}t dd}tdddd}tdddd}|||f |||f< tt |d |dk |d |dk |d |dk g¡|d d
|dkf k ¡ d S ) Nr rG rZ r r« r r r )r r r r r rf )r# r0 rú Zg1Zg2r% r% r& Ú test_booleanO s
|
zTestIndex.test_booleanc C sD t jg dd}t jg dd}|| }t|g t|jt d¡ d S )Nr r1 rÜ )r r r r, )r# r0 rH r´ r% r% r& Útest_boolean_edgecaseW s
|
|
z TestIndex.test_boolean_edgecaseN)rC rD rE r' r( r% r% r% r& r& N s r& c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚTestBinaryReprc C s t t d¡d d S )Nr Ú0©r r Ú binary_reprrá r% r% r& Ú test_zero` s zTestBinaryRepr.test_zeroc C s4 t t d¡d t t d¡d t t d¡d d S )Nr6 Z1010iê0 Z11000011101010iÐÔ£ Z101000111101010011010000r+ rá r% r% r& Ú test_positivec s
|
ÿ
|
ÿz TestBinaryRepr.test_positivec C sD t t d¡d t t d¡d t t d¡d t t d¡d d S ) Nr8 z-1r7 z-1010iÏÿÿz-11000011101010i0+\ÿz-101000111101010011010000r+ rá r% r% r& Ú test_negativej s
|
ÿ
|
ÿz TestBinaryRepr.test_negativec C s@ t tjdddd t tjdddd t tjdddd d S )
|
Nr rG ©ÚwidthZ00000r6 rO Z0001010r| Z1111011r+ rá r% r% r& Útest_sufficient_widthr s z$TestBinaryRepr.test_sufficient_widthc C sZ t tjdddd tddD ]6}d|d }d|d d }t tj||d| q d S )
|
NiÿÿÿrZ r0 Z10000000r ri r r r* )r r r, rè )r# r1 ÚnumÚexpr% r% r& Útest_neg_width_boundariesw s
|
z(TestBinaryRepr.test_neg_width_boundariesc C s t tjt d¡ddd d S )Nlûÿÿÿ é@ r0 Z@1100000000000000000000000000000000000000000000000000000000000000)r r r, r rá r% r% r& Útest_large_neg_int64 s ÿz#TestBinaryRepr.test_large_neg_int64N) rC rD rE r- r. r/ r2 r5 r7 r% r% r% r& r) _ s r) c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
|
Ú TestBaseReprc C s t t dd¡d d S )Néó r Z100000©r r Ú base_reprrá r% r% r& Ú
|
test_base3 s zTestBaseRepr.test_base3c C sN t t dd¡d t t ddd¡d t t dd¡d t t dd¡d d S )
|
Nr r6 Z12r Z000012r l ¸5j é$ Z 10QR0ROFCEWr: rá r% r% r& r. s zTestBaseRepr.test_positivec C s< t t dd¡d t t ddd¡d t t dd¡d d S )Néôÿÿÿr6 z-12r z-000012z-30r: rá r% r% r& r/ s zTestBaseRepr.test_negativec C sD t t t dd¡ W 5 Q R X t t t dd¡ W 5 Q R X d S )Nr é% )r r@ r r; rá r% r% r& Útest_base_range s
|
|
z TestBaseRepr.test_base_rangeN)rC rD rE r< r. r/ r@ r% r% r% r& r8 s r8 c @ sH e Zd Zdd Zdd Zdd Zdd Zej d d
|
d d g¡d d Z
|
dS )ÚTestArrayComparisonsc C sd t t ddg¡t ddg¡¡}t| tt|tk t t ddg¡t dddg¡¡}t| tt|tk t t ddg¡t ddg¡¡}t| tt|tk t t ddg¡t ddg¡¡}t| tt|tk t t jdgddt jdgdd¡}t| tt|tk t t jdgd dt jdgd d¡}t| tt|tk d S )
|
Nr r r r r0 ZS1r1 )r0 r zS1,u4)r Ú array_equalr r rJ rÜ ©r# r r% r% r& Útest_array_equal s( "
|
|
|
$ÿz%TestArrayComparisons.test_array_equalc C s t ddt jg¡}t dt jdg¡}t ddt jg¡}tt ||¡ tt j||dd tt j||dd tt j||dd t t j¡}tt ||¡ tt j||dd t jdddgtd}tt ||¡ tt j||dd t ddgt jdgg¡}tt ||¡ tt j||dd t dg¡gd \}}t jt j |_|_tt j||d d tt j||dd d S )
|
Nr r T©Ú equal_nanr r1 r y ð? ð?F) r r r r r rB r rO Úimag)r# Za1Úa2Úa3r0 rH r% r% r& Útest_array_equal_equal_nanµ s( z/TestArrayComparisons.test_array_equal_equal_nanc C sr t jd dd gtd}t|d kdddg t|d kdddg t d¡}t|d kdddg t|d kdddg d S )Nr r1 TFr )r r rÞ r rÿ r} r% r% r& Ú test_none_compares_elementwiseÓ s
|
z3TestArrayComparisons.test_none_compares_elementwisec
|
C s" t t ddg¡t ddg¡¡}t| tt|tk t t ddg¡t dddg¡¡}t| tt|tk t t ddg¡t ddg¡¡}t| tt|tk t t ddg¡t ddg¡¡}t| tt|tk t t ddg¡t dg¡¡}t| tt|tk t t ddg¡t dgdgg¡¡}t| tt|tk t t ddg¡t dg¡¡}t| tt|tk t t ddg¡t dgdgg¡¡}t| tt|tk t t ddg¡t dddgdddgddd gg¡¡}t| tt|tk d S )
|
Nr r r r rG rN rO rZ r[ )r Z array_equivr r rJ rÜ rC r% r% r& Útest_array_equivÜ s6 "
|
|
|
$
|
$
|
4
|
z%TestArrayComparisons.test_array_equivr, rÈ ZV3r c C sl t jd|d}t|| ||k ¡ r(t|dkr4d S t jdddg|d}||k ¡ rXt||k ¡ shtd S )Nr r1 rÈ ó 1ó 2ó 3)r r3 r rç rK r rf )r# r, r3 Znonzerosr% r% r& Ú test_compare_unstructured_voidsú s
|
z4TestArrayComparisons.test_compare_unstructured_voidsN) rC rD rE rD rJ rK rL r> r¼ r½ rP r% r% r% r& rA s rA c C s¾ t | | | jjdks&t ¡ jjdkrBtjdkrBt| j |j k ndt| j j
|
|j j
|
k t| j j |j j k t| j j |j j k t| j j |j j k t| j j|j jk t| jj|jjk d S )NrZ r Úwin32)r r, Ú alignmentr rý ÚitemsizeÚsysr6 r rì Zowndataró Ú c_contiguousÚ f_contiguousZwritebackifcopyr )rú r= r% r% r& Úassert_array_strict_equal s
|
ÿþrW c @ sH e Zd Zdd ZdddZdddZdd Zd
|
d Zd d Zdd Z dd Z
|
dd Z dd Z e j dd¡dd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Ze j d*dd+g¡d,d- Zd.d/ Zd0d1 Zd2d3 Z d4d5 Z d6d7 Z d8d9 Z d:d; Z d<d= Z!d>d? Z"d@dA Z#dBdC Z$dDdE Z%dFdG Z&dHdI Z'dJdK Z(dLdM Z)dNdO Z*dPdQ Z+dRdS Z,dTdU Z-dVdW Z.dXdY Z/dZd[ Z0d\d] Z1d^d_ Z2d`da Z3dbdc Z4ddde Z5dfdg Z6dhdi Z7djdk Z8dldm Z9e j dndodpe: ;dq¡fe: <dq¡e: ;dq¡fg¡drds Z=e j dte:j;dqe:j>dudvdwe:j?dqdwe@dufe:j;dqdxdudp dvdve:j;dqdxdufg¡dydz ZAe jjBd{eCd|d}e j d~e:j;dqe:j>due: De:jE¡e:j;dqe:jFdufg¡e j Gd¡dd ZHe jjBdde j d~e:jDdpgdq dxdue: Id¡e:j;dqe:jFdufg¡e j Gd¡d
d ZJeKeL M¡ eNjOeN P¡ eN Q¡ B eN R¡ dddd ZSdS )ÚTestClipc C s d| _ d| _d S )NrG r )ÚnrÚncrá r% r% r& râ s zTestClip.setup_methodNc C sZ |d kr.|d kr | ||¡S |j |||dS n(|d krD| |||¡S |j ||||dS d S )Nr¥ )rW )r# r0 r¯ rû rS r¦ r% r% r& Úfastclip! s zTestClip.fastclipc C s0 t ||¡dt ||¡ }|j|||f|dS )Nr rí )r ZlessZgreaterrP )r# r0 r¯ rû rS Úselectorr% r% r& rW - s z TestClip.clipc C s
|
t ||S r )r ©r# r r¯ r% r% r& Ú_generate_data3 s zTestClip._generate_datac C s t ||dt|| S )Nr¸ )r r r] r% r% r& Ú_generate_data_complex6 s z TestClip._generate_data_complexc C s t || tj¡S r )r r r r4 r] r% r% r& Ú_generate_flt_data9 s zTestClip._generate_flt_datac C s> t |¡}tjdkr(| |j d¡¡}n| |j d¡¡}|S )NÚlittleú>ú<)r ZasarrayrT Ú byteorderr r, r r} r% r% r& Ú_neg_byteorder< s
|
|
|
zTestClip._neg_byteorderc C s& t ||}| |¡}t|jj |S r )r re r r, r )r# r r¯ rÊ r% r% r& Ú_generate_non_native_dataD s
|
|
z"TestClip._generate_non_native_datac C s dt || tj¡S rÙ )r r r r r] r% r% r& Ú_generate_int_dataJ s zTestClip._generate_int_datac C s dt || tj¡S rÙ )r r r r r] r% r% r& Ú_generate_int32_dataM s z TestClip._generate_int32_datar, z?bhilqpBHILQPefdgFDGOc C sV t jd|d}t jd|d}t |dd¡}|dkrH| ¡ | ¡ ksRtn
|
t|| d S )Nr6 r1 r r r
)r rÿ r3 rW ÚtolistrK r )r# r, rX r¿ Úactualr% r% r& Útest_ones_pathologicalR s z TestClip.test_ones_pathologicalc C sB | | j| j¡}d}d}| |||¡}| |||¡}t|| d S )Ng¹?ç333333ã?©r^ rY rZ r[ rW rW ©r# r0 r¯ rû ÚacÚactr% r% r& Útest_simple_double_ s zTestClip.test_simple_doublec C sL | | j| j¡}| t¡}d}d}| |||¡}| |||¡}t|| d S ©Nr r )rg rY rZ r r r[ rW rW rn r% r% r& Útest_simple_inth s
|
zTestClip.test_simple_intc C sN | | j| j¡}t |j¡}|d }| |||¡}| |||¡}t|| d S )NrK © r^ rY rZ r r3 r- r[ rW rW rn r% r% r& Útest_array_doubler s zTestClip.test_array_doublec C s | | j| j¡}d}d}| |||¡}| |||¡}t|| | | j| j¡}d}| d¡}t|j j
|
| |||¡}| |||¡}t|| d S ©Nç à¿rl ) rf rY rZ r[ rW r r^ re r r, r rn r% r% r& Útest_simple_nonnative{ s
|
|
z TestClip.test_simple_nonnativec C s d| | j| j¡ }d}d}| |||¡}| |||¡}t|| d| | j| j¡ }d}d}| |||¡}| |||¡}t|| d S )Nr rw rB y à¿ ð?y ð? @)r_ rY rZ r[ rW rW r^ rn r% r% r& Útest_simple_complex s
|
z TestClip.test_simple_complexc C sR t jdtd}| ¡ }| ¡ }| ||d ¡}| |d |¡}t|| t|| d S ©Nr6 r1 )r rÿ r¢ r¯ rN r[ rW )r# r0 r¯ rû ÚamZaMr% r% r& Útest_clip_complex s
|
zTestClip.test_clip_complexc C sz | | jd | jd ¡}|d d d
d d d
f }t|jd t|jd | |dd¡}| |dd¡}t|| d S )Nr r Ú F_CONTIGUOUSÚ C_CONTIGUOUSgù¿g333333û?)r^ rY rZ r rì r[ rW rW )r# r0 ro rp r% r% r& Útest_clip_non_contig¬ s z TestClip.test_clip_non_contigc C s^ | | j| j¡}d}d}t |j¡}t |j¡}| ||||¡ | ||||¡ t|| d S rv rt rn r% r% r& Útest_simple_out¶ s zTestClip.test_simple_outr¦ r c C s¢ | | j| j¡}t d¡}t d¡}tj|jtjd}| ¡ }|d krpt t
|
| j |||||d W 5 Q R X n| j |||||d | ||||¡ t || d S )Nr r r1 r¥ ©rh rY rZ r r r3 r- r rÞ r rä r[ rW rW )r# r¦ r0 r¯ rû ro rp r% r% r& Útest_simple_int32_inoutÁ s
|
|
|
z TestClip.test_simple_int32_inoutc C sl | | j| j¡}t d¡}t d¡}tj|jtjd}| ¡ }| ||||¡ |
|
||||¡ t || d S ©Nr8 r r1 © rh rY rZ r r r3 r- r rÞ r[ rW rW rn r% r% r& Útest_simple_int64_outÓ s
|
|
z TestClip.test_simple_int64_outc C s | | j| j¡}t |jtj¡}t d¡}tj|jtjd}| ¡ }t t
|
| ||||¡ W 5 Q R X | ||||¡ t || d S ©Nr r1 ©rh rY rZ r r3 r- r r rÞ r rä r[ rW rW rn r% r% r& Útest_simple_int64_inoutÞ s
|
|
z TestClip.test_simple_int64_inoutc C st | | j| j¡}d}d}tj|jtjd}| ¡ }tt |
|
||||¡ W 5 Q R X | ||||¡ t || d S ©Nrþ rk r1 © r^ rY rZ r r3 r- r rÞ r rä r[ rW rW rn r% r% r& Útest_simple_int32_outë s
|
z TestClip.test_simple_int32_outc C sV | | j| j¡}| ¡ }t |j¡}d}| ||||¡ | ||||¡ t || d S ©NrB ©
|
r^ rY rZ rÞ r r3 r- r[ rW rW ©r# r0 ro r¯ rû r% r% r& Útest_simple_inplace_01ø s z TestClip.test_simple_inplace_01c C sN | | j| j¡}| ¡ }d}d}| ||||¡ | ||||¡ t|| d S rv ©r^ rY rZ rÞ r[ rW rW r r% r% r& Útest_simple_inplace_02 s z TestClip.test_simple_inplace_02c C s | | jd | jd ¡}|d d d
d d d
f }t|jd t|jd | ¡ }d}d}| ||||¡ | ||||¡ t|| d S )Nr r r} r~ rw rl ) r^ rY rZ r rì rÞ r[ rW r r r% r% r& Útest_noncontig_inplace s z TestClip.test_noncontig_inplacec C sB | | j| j¡}d}d}| |||¡}| |||¡}t|| d S rv rm rn r% r% r& Útest_type_cast_01 s zTestClip.test_type_cast_01c C sN | | j| j¡}| tj¡}d}d}| |||¡}| |||¡}t|| d S rr ) rg rY rZ r r r r[ rW rW rn r% r% r& Útest_type_cast_02" s zTestClip.test_type_cast_02c C sZ | | j| j¡}d}d}| |t |¡t |¡¡}| |t |¡t |¡¡}t|| d S rr )rh rY rZ r[ r r rW rW rn r% r% r& Útest_type_cast_03, s zTestClip.test_type_cast_03c C sN | | j| j¡}t d¡}t d¡}| |||¡}| |||¡}t|| d S rr )rh rY rZ r r4 r[ rW rW )r# r0 r¯ rû rp ro r% r% r& Útest_type_cast_045 s
|
|
zTestClip.test_type_cast_04c C sZ | | j| j¡}d}d}| ||t |j¡ |¡}| ||t |j¡ |¡}t|| d S ©Nrw rB ) rg rY rZ r[ r r3 r- rW rW rn r% r% r& Útest_type_cast_05> s zTestClip.test_type_cast_05c C sL | | j| j¡}d}| |¡}d}| |||¡}| |||¡}t|| d S )NrK rB )r^ rY rZ re rW r[ rW )r# r0 r¯ Úm_srû rp ro r% r% r& Útest_type_cast_06G s
|
zTestClip.test_type_cast_06c C sd | | j| j¡}dt |j¡ }d}| |¡}t|jj |
|
||¡}| |||¡}t || d S r ) r^ rY rZ r rÿ r- re r r, r rW r[ rW )r# r0 r¯ rû Úa_srp ro r% r% r& Útest_type_cast_07Q s
|
zTestClip.test_type_cast_07c C sX | | j| j¡}d}d}| |¡}t|jj | |||¡}| ||¡}t || d S r )
|
r^ rY rZ re r r, r r[ rW rW )r# r0 r¯ rû r ro rp r% r% r& Útest_type_cast_08\ s
|
zTestClip.test_type_cast_08c C sf | | j| j¡}dt |j¡ }d}| |¡}t|jj |
|
|||¡}| |||¡}t || d S r ) r^ rY rZ r rÿ r- re r r, r r[ rW rW )r# r0 r¯ rû r ro rp r% r% r& Útest_type_cast_09g s
|
zTestClip.test_type_cast_09c C sh | | j| j¡}tj|jtjd}t d¡}t d¡}| j||||d}| j||||d}t || d S )Nr1 rw r rí )
|
rg rY rZ r r3 r- r4 rW r[ rW ©r# r0 rH r¯ rû rp ro r% r% r& Útest_type_cast_10r s
|
|
zTestClip.test_type_cast_10c C sl | | j| j¡}| ¡ }| |j d¡¡}| ¡ }d}d}| j||||d | j||||d t || d S )Nrb rw rB rí )
|
rf rY rZ rÞ r r, r r[ rW rW )r# r0 rH Zbtr¯ rû r% r% r& Útest_type_cast_11| s zTestClip.test_type_cast_11c C sh | | j| j¡}tj|jtjd}t d¡}t d¡}| j||||d}| j ||||d}t
|
|| d S )Nr1 r r rí ) rg rY rZ r r3 r- r4 r rW r[ rW r r% r% r& Útest_type_cast_12 s
|
|
zTestClip.test_type_cast_12c C s^ | | j| j¡}d}d}t |j¡}t |j¡}| ||||¡ | ||||¡ t|| d S rv rt rn r% r% r& Útest_clip_with_out_simple s z"TestClip.test_clip_with_out_simplec C s | | j| j¡}t d¡}t d¡}tj|jtjd}| ¡ }t t
|
| ||||¡ W 5 Q R X | ||||¡ t || d S )Nr r r1 r rn r% r% r& Útest_clip_with_out_simple2 s
|
|
|
z#TestClip.test_clip_with_out_simple2c C sl | | j| j¡}t d¡}t d¡}tj|jtjd}| ¡ }| ||||¡ |
|
||||¡ t || d S r r rn r% r% r& Ú test_clip_with_out_simple_int32ª s
|
|
z(TestClip.test_clip_with_out_simple_int32c C s | | j| j¡}t |jtj¡}t d¡}tj|jtjd}| ¡ }t t
|
| ||||¡ W 5 Q R X | ||||¡ t || d S r r rn r% r% r& Ú test_clip_with_out_array_int32µ s
|
|
z'TestClip.test_clip_with_out_array_int32c C st | | j| j¡}d}d}tj|jtjd}| ¡ }tt |
|
||||¡ W 5 Q R X | ||||¡ t || d S r r rn r% r% r& Ú!test_clip_with_out_array_outint32Â s
|
z*TestClip.test_clip_with_out_array_outint32c C sJ t d¡ dd¡}t |¡j}|jdd|d | |dd¡}t|| d S )Nrý r r6 rí )r r= r Ú
|
empty_likerë rW r )r# r0 rS r¿ r% r% r& Ú test_clip_with_out_transposedÏ s
|
z&TestClip.test_clip_with_out_transposedc C sf t d¡ dd¡}| ¡ }|d d
jdd|dd
d | |d d
dd¡}t|dd
| d S )Nrý r r8 r6 r rí )r r= r rÞ rW r )r# r0 ro r¿ r% r% r& Ú!test_clip_with_out_memory_overlap× s
|
z*TestClip.test_clip_with_out_memory_overlapc C sV | | j| j¡}| ¡ }t |j¡}d}| ||||¡ | ||||¡ t || d S r r r r% r% r& Útest_clip_inplace_arrayß s z TestClip.test_clip_inplace_arrayc C sN | | j| j¡}| ¡ }d}d}| ||||¡ | ||||¡ t|| d S rv r r r% r% r& Útest_clip_inplace_simpleé s z!TestClip.test_clip_inplace_simplec C s\ | | j| j¡}| ¡ }d}d}tj||||d}| ||||¡ t|| t||k d S )Nrw rl rí )r^ rY rZ rÞ r rW rW r )r# r0 ro r¯ rû rH r% r% r& Útest_clip_func_takes_outó s
|
z!TestClip.test_clip_func_takes_outc C sÞ t d¡}tt t|jt jd| W 5 Q R X tt t|jt jd| W 5 Q R X tt t|jt jt jd| W 5 Q R X tt t|jdt jd| W 5 Q R X tt t|jt jdd| W 5 Q R X d S )Ng @)r¯ )rN )r¯ rN r r6 )r r= r rä r rW r )r# rê r% r% r& Ú test_clip_nanþ s
|
|
|
|
"
|
|
zTestClip.test_clip_nanc C sR t jdtd}t |dd¡}t ddddddddddg
|
¡}| ¡ | ¡ ksNtd S )Nr6 r1 r rG r r r )r r= rÞ rW r ri rK )r# r0 rj r¿ r% r% r& Útest_object_clip s zTestClip.test_object_clipc C s6 t jdtd}ttd t |d d ¡ W 5 Q R X d S )Nr6 r1 z
|
max or min)r r= rÞ r
|
r@ rW r} r% r% r& Útest_clip_all_none s zTestClip.test_clip_all_nonec C s: t jdtd}ttd | j|dddd W 5 Q R X d S )Nr6 r1 zcasting must be one ofr rZ Úgarbager¥ )r r= rÞ r
|
r@ r[ r} r% r% r& Útest_clip_invalid_casting s
|
ÿz"TestClip.test_clip_invalid_castingz
|
amin, amaxr¶ r r6 c C s@ t jdt jd}t t ||¡|¡}t |||¡}t|| d S rz )r r= r ÚminimumÚmaximumrW r )r# ÚaminÚamaxr0 r¿ rj r% r% r& Ú test_clip_value_min_max_flip s z%TestClip.test_clip_value_min_max_flipzarr, amin, amax, expr1 r lûÿÿÿÿÿÿÿ rm c C s t |||¡}t|| d S r rV )r# rX rµ r¶ r4 rj r% r% r& Útest_clip_problem_cases+ s z TestClip.test_clip_problem_casesz no scalar nan propagation yetT)r r? Ústrictzarr, amin, amaxzignore::DeprecationWarningc C s0 t t ||¡|¡}t |||¡}t|| d S r ©r r³ r´ rW r ©r# rX rµ r¶ r¿ rj r% r% r& Ú test_clip_scalar_nan_propagation= s z)TestClip.test_clip_scalar_nan_propagationz propagation doesn't match specr ÚNaTc C s0 t t ||¡|¡}t |||¡}t|| d S r rº r» r% r% r& Útest_NaT_propagationN s
|
z TestClip.test_NaT_propagation)r, r- )rÊ rX c C sÐ t ¡ t ¡ B }| t jd|jd¡\}}| dd ¡}| |t j||d ddidB ¡}| |t j||d ddidB ¡}t |||¡} t
|
|||¡}
|
tj |tj |||
|
d
|
|
|
d
|
} | j |
|
ksÂtt| | d S ) a A property-based test using Hypothesis.
|
|
This aims for maximum generality: it could in principle generate *any*
|
valid inputs to np.clip, and in practice generates much more varied
|
inputs than human testers come up with.
|
|
Because many of the inputs have tricky dependencies - compatible dtypes
|
and mutually-broadcastable shapes - we use `st.data()` strategy draw
|
values *inside* the test function, from strategies we construct based
|
on previous values. An alternative would be to define a custom strategy
|
with `@st.composite`, but until we have duplicated code inline is fine.
|
|
That accounts for most of the function; the actual test is just three
|
lines to calculate and compare actual vs expected results!
|
r )Z
|
num_shapesZ
|
base_shapec S s t j| ddS )NF)Ú allow_nan)ÚhynpZ
|
from_dtype©rú r% r% r& rI ~ rJ z-TestClip.test_clip_property.<locals>.<lambda>r r¿ F)r, r- Úelementsr r1 N)rÀ Úinteger_dtypesÚfloating_dtypesZdrawZ mutually_broadcastable_shapesr- ZflatmapÚarraysr rW rh r³ r´ r, rK r ) r# rÊ rX Znumeric_dtypesZ in_shapesZ result_shaper rµ r¶ r$ rÓ r¿ r% r% r& Útest_clip_property\ s. ÿÿÿ ÿ
|
ÿ
|
zTestClip.test_clip_property)NN)N)TrC rD rE râ r[ rW r^ r_ r` re rf rg rh r> r¼ r½ rk rq rs ru rx ry r| r r r r
r r r r r r r r r r r r r r r r¡ r¢ r£ r¤ r¥ r¦ r§ r© rª r« r¬ r r® r¯ r° r² r r3 rÿ r· r ÚfullrÞ r¸ rÆ rK r r r rt r¼ Ú timedelta64r¾ r ÚstrÊ rÀ rÅ rà rÄ Z array_shapesrÆ r% r% r% r& rX sÌ
|
|
|
|
|
|
|
|
|
|
|
|
|
|
ú
|
ý ý÷
|
þ
|
þþ
|
þÿ
|
þþrX c @ sd e Zd ZdZdZdd Zdd Zdd Zd d
|
Zd d Z d d Z
|
dd Z dd Z dd Z dd ZdS )Ú TestAllcloseçñhãµøä>ç:0âyE>c C s t jdd| _d S )Nr ©r ©r r# Zolderrrá r% r% r& râ s zTestAllclose.setup_methodc C s t jf | j d S r rÎ rá r% r% r& Úteardown_method s z TestAllclose.teardown_methodc C s t t ||¡d||f d S ©Nz%s and %s not close©r r Úallclose©r# rú r= r% r% r& Ú tst_allclose s zTestAllclose.tst_allclosec C s t t ||¡ d||f d S ©Nz %s and %s shouldn't be closerÑ rÓ r% r% r& Útst_not_allclose s z TestAllclose.tst_not_allclosec C s¾ t ddg¡}t d¡ d¡}| j}| j}ddgddgf|gdgfdgd| | gf|||| f|||| |d f|||| ft jt jft jt jgfg}|D ]\}}| ||¡ q¤d S )NrÐ éè é} ©rG rG rG r r r )r r r= r ÚatolÚrtolr rÔ )r# rX ÚaranrÚ rÛ rÊ rú r= r% r% r& Útest_ip_allclose s
|
|
ù z TestAllclose.test_ip_allclosec C s t d¡ d¡}| j}| j}t jdgdt jgft jdgddgft jt jgdt jgft jt jgddgft j dgt jdgft jdgt jdgf|d gdgfdgd| |d gf|||| |d ft t jdg¡t dt jg¡fg
|
}|D ]\}}| ||¡ qêd S )NrØ rÙ r r r ) r r= r rÚ rÛ r r r rÖ )r# rÜ rÚ rÛ rÊ rú r= r% r% r& Útest_ip_not_allclose² s ÷ z!TestAllclose.test_ip_not_allclosec C s\ t t jdg¡}t dt jg¡}t ||¡ t|t t jdg¡ t|t dt jg¡ d S ©Nr r )r r r rÒ r rÓ r% r% r& Ú test_no_parameter_modificationÇ s
|
z+TestAllclose.test_no_parameter_modificationc C s4 t t j¡j}t j|gt jd}tt ||¡ d S )Nr1 )r r® Úint_r¯ r r rÒ )r# Zmin_intr0 r% r% r& Ú test_min_intÎ s zTestAllclose.test_min_intc C s( t dt jg¡}tt j||dd d S )NrB TrE )r r r r rÒ ræ r% r% r& Ú test_equalnanÔ s zTestAllclose.test_equalnanc C s8 G dd dt j}|dg}ttt ||¡tk d S )Nc @ s e Zd Zdd ZdS )z6TestAllclose.test_return_class_is_ndarray.<locals>.Fooc _ s t j|| | ¡S r )r r rI )Úclsr- Úkwargsr% r% r& Ú__new__Ü s z>TestAllclose.test_return_class_is_ndarray.<locals>.Foo.__new__N)rC rD rE ræ r% r% r% r& ÚFooÛ s rç r )r rH r rJ rÒ rÜ )r# rç r0 r% r% r& Ú test_return_class_is_ndarrayØ s
|
z)TestAllclose.test_return_class_is_ndarrayN)rC rD rE rÛ rÚ râ rÏ rÔ rÖ rÝ rÞ rà râ rã rè r% r% r% r& rÊ s rÊ c @ s e Zd ZdZdZdd Zdd Zdd Zd d
|
Zd d Z d d Z
|
dd Z dd Z dd Z dd Zdd Zdd Zdd Zd d Zd S ) Ú TestIscloserË rÌ c C sN | j }| j}t ddg¡}t d¡ d¡}ddgddgf|gdgfdgd| | gf|||| f|||| | f|||| ftjtjftjtjgftjtj gtjtj gfg | _tjdgdtjgftjtj gddgftjtjgdtj gftjtjgddgftjdgtjtj gf|d gdgfdgd| |d gf|||d | |d ft tjdg¡t dtjg¡fg | _ tjdgtj|d gf|dd dd| | gdtjd gft d
|
¡ddd gftjtjtjtjgfdg|tjtj tjgfd|tjtj tjgfg| _
|
d d gd d d gd d d gd d d gd d d d gd d d d gg| _ d S )NrÐ r× rØ rÙ r r r gñ?g .Ar gÍÌÌÌÌÌ @TF) rÚ rÛ r r r= r r Úall_close_testsr Únone_close_testsÚsome_close_testsÚsome_close_results)r# rÚ rÛ rX rÜ r% r% r& Ú_setupç sL
|
|
÷ ÷ $ú
|
|
úzTestIsclose._setupc C sB | ¡ | j}| j}t||D ] \\}}}tt ||¡| q d S r )rî rì rí Úzipr r Úisclose)r# ÚtestsÚresultsrú r= r$ r% r% r& Útest_ip_isclose
|
s
|
zTestIsclose.test_ip_isclosec C s$ t t t ||¡¡d||f d S rÐ )r r rf rð rÓ r% r% r& Útst_all_isclose
|
s zTestIsclose.tst_all_isclosec C s* d}t t t ||¡¡ |||f d S rÕ )r r rç rð )r# rú r= ræ r% r% r& Útst_none_isclose
|
s z TestIsclose.tst_none_isclosec C sr d}d}t |¡rFt |¡rFtt ||¡t ||¡k|||f d n(tt ||¡ ¡ t ||¡|||f d S )Nz4isclose.all() and allclose aren't same for %s and %sz.isclose and allclose aren't same for %s and %srå )r rÎ r rð rÒ r rf )r# rú r= ræ Zmsg2r% r% r& Útst_isclose_allclose"
|
s
|
*z TestIsclose.tst_isclose_allclosec C s( | ¡ | jD ]\}}| ||¡ qd S r )rî rê rô rÓ r% r% r& Útest_ip_all_isclose*
|
s z TestIsclose.test_ip_all_isclosec C s( | ¡ | jD ]\}}| ||¡ qd S r )rî rë rõ rÓ r% r% r& Útest_ip_none_isclose/
|
s z TestIsclose.test_ip_none_isclosec C s8 | ¡ | j| j | j }|D ]\}}| ||¡ q d S r )rî rê rë rì rö )r# rñ rú r= r% r% r& Útest_ip_isclose_allclose4
|
s
|
ÿ z$TestIsclose.test_ip_isclose_allclosec C sJ t tjtjtjdddg t dtjg¡}t tj||ddddg d S )NTrE rB )r r rð r r r r% r% r& Útest_equal_nan;
|
s zTestIsclose.test_equal_nanc C s t j dddgt d¡¡}tt|tt d|¡k tt|tt |d¡k t j dddgt jt jt jg¡}tt|tt t j|¡k tt|tt |t j¡k t j dddgt jt jt jg¡}t jt j|dd}tt|t|k t dddg|j
|
t j|t jdd}tt|t|k t dddg|j
|
t j dddgt jt jt jg¡}t j||dd}tt|t|k t dddg|j
|
d S )NTFr r rE ) r ÚmaZ masked_wherer= r rJ rð r r r ÚmaskrÓ r% r% r& Útest_masked_arrays@
|
s" z TestIsclose.test_masked_arraysc C s t t t dd¡¡ d S ©Nr )r r rÎ rð rá r% r% r& Útest_scalar_return[
|
s z TestIsclose.test_scalar_returnc C s\ t t jdg¡}t dt jg¡}t ||¡ t|t t jdg¡ t|t dt jg¡ d S rß )r r r rð r rÓ r% r% r& rà ^
|
s
|
z*TestIsclose.test_no_parameter_modificationc C sT t t tjtj ¡tjk t t dtj¡tjk t tt dtj¡tjk d S ©Nr )r r rð r rÒ rJ rW rá r% r% r& Útest_non_finite_scalare
|
s z"TestIsclose.test_non_finite_scalarc C s t jddddggdd}t j||ddd ¡ s2tt j||t dd
|
¡dd ¡ sTtt j||ddd sjtt j||t dd
|
¡dd std S ) Nr r r r½ zm8[ns]r1 r T)rÚ rF Úns)r r rð rf rK rÈ rÒ r} r% r% r& Útest_timedeltal
|
s
|
"zTestIsclose.test_timedeltaN)rC rD rE rÛ rÚ rî ró rô rõ rö r÷ rø rù rú rý rÿ rà r r r% r% r% r& ré ã s -ré c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )Ú
|
TestStdVarc C s t ddddg¡| _d| _d S )Nr r8 )r r r$ Úreal_varrá r% r% r& râ w
|
s zTestStdVar.setup_methodc C s0 t t | j¡| j t t | j¡d | j d S )Nr )r r r¹ r$ r r§ rá r% r% r& Ú
|
test_basic{
|
s zTestStdVar.test_basicc C s$ t t d¡d t t d¡d d S rß ©r r r¹ r§ rá r% r% r& Ú test_scalars
|
s zTestStdVar.test_scalarsc C sh t tj| jdd| jt| j t| jd t tj| jddd | jt| j t| jd d S )Nr ©Zddofr ©r r r¹ r$ r r, r§ rá r% r% r& Ú
|
test_ddof1
|
s ÿ ÿzTestStdVar.test_ddof1c C sh t tj| jdd| jt| j t| jd t tj| jddd | jt| j t| jd d S )Nr r r
|
rá r% r% r& Ú
|
test_ddof2
|
s ÿ ÿzTestStdVar.test_ddof2c C s t d¡}t d¡}t j||d}t||k t|| t j||d}t||k t|| t j||d}t||k t|| d S )Nr6 r( rí )r r= r r§ r r r¹ rq )r# rê rS r r% r% r& Útest_out_scalar
|
s
|
|
|
|
zTestStdVar.test_out_scalarN) rC rD rE râ r r r r r r% r% r% r& r v
|
s r c @ s e Zd Zdd Zdd ZdS )ÚTestStdVarComplexc C s> t ddddg¡}d}tt |¡| tt |¡d | d S )Nr r¸ r8 y ð¿r )r r r r¹ r§ )r# r$ r r% r% r& r
|
s z TestStdVarComplex.test_basicc C s$ t t d¡d t t d¡d d S )Nr¸ r r rá r% r% r& r ¤
|
s z TestStdVarComplex.test_scalarsN)rC rD rE r r r% r% r% r& r
|
s r c @ sX e Zd Zdd ZdddZdd Zdd Zd
|
d Zd d Ze j
|
j e dddd Z dS )ÚTestCreationFuncsc C sb dd t jtj ¡ D }dd |D }t|| dd |D B dd d| _dd d
|
| _d | _d S ) Nc S s h | ]}t |¡qS r% )r r, ©rù Útpr% r% r& Ú <setcomp>
|
s z1TestCreationFuncs.setup_method.<locals>.<setcomp>c S s h | ]}|j d ¡r|qS )r* )r9 Úendswithr r% r% r& r ¯
|
s c
|
S s4 h | ],}t d dD ] }t |j dt|¡¡qqS )r r6 r* )rè r r, r9 Úreplace)rù r r r% r% r& r ±
|
s ÿc S s | j S r )r9 r1 r% r% r& rI ³
|
rJ z0TestCreationFuncs.setup_method.<locals>.<lambda>)ÚkeyrU rV )rñ ÚFr6 ) r³ Úchainr r ÚvaluesÚsortedÚdtypesÚordersÚndims)r# r Zvariable_sizedr% r% r& râ ¬
|
s ÿÿý z TestCreationFuncs.setup_methodNc C sÌ dt | j| j| jf}i }|d k r*d|i}tj| D ]\}}}}||g } |r\|j d¡r\q4|| f||d|}
|
t|
|
j | t
|
t |
|
j | j| |d k r4|j d¡r²t|} n|} t|
|
| | ¡ q4d S )N)r r r Ú
|
fill_valuez|V©Úorderr, z|S)rè r r r r³ r´ r9 Ú
|
startswithr r, r Úgetattrrì rJ ) r# r r ÚparÚ
|
fill_kwargr¡ r r r, r- rX r r% r% r& Úcheck_function·
|
s, ý
|
ÿ
|
z TestCreationFuncs.check_functionc C s | tj¡ d S r )r$ r r3 rá r% r% r& Ú
|
test_zerosÔ
|
s z TestCreationFuncs.test_zerosc C s | tj¡ d S r )r$ r rÿ rá r% r% r& Ú test_ones×
|
s zTestCreationFuncs.test_onesc C s | tj¡ d S r )r$ r Úemptyrá r% r% r& Ú
|
test_emptyÚ
|
s z TestCreationFuncs.test_emptyc C s | tjd¡ | tjd¡ d S )Nr r )r$ r rÇ rá r% r% r& Ú test_fullÝ
|
s zTestCreationFuncs.test_fullzPython lacks refcountsr c C s d}t |¡}t |gd ¡ tt |¡|k t |gd ¡ tt |¡|k t |gd ¡ tt |¡|k t |gd d¡ tt |¡|k d S )Nr r6 r )rT Ú getrefcountr r3 r rÿ r' rÇ )r# ZdimÚbegr% r% r& Útest_for_reference_leaká
|
s
|
z)TestCreationFuncs.test_for_reference_leak)N)rC rD rE râ r$ r% r& r( r) r> r¼ r5 r r, r% r% r% r& r ©
|
s
|
r c @ s e Zd ZdZdd Zdd ZdddZd d
|
Zd d Zd d Z dd Z
|
e j dejejejejg¡e j deeg¡dd ZdS )Ú TestLikeFuncsz4Test ones_like, zeros_like, empty_like and full_likec C s4 t d¡d ft d¡dft jdddd ft d¡dft d¡ dd¡d ft d¡ dd¡d ft d¡jd
|
d d d ft d¡jd d d d ft d¡ ddd¡d ft d¡ ddd¡dft d¡jdd d d ft d¡jdd d dft d¡ ddd¡ dd¡d ft d¡ ddd¡ dd¡dfg| _ddddg| _d S )Ng @r rc rN rd r1 rf r r rÌ r )r ©r r rd r rÜ )r r r r r rá r% ©rG )rG rN ©rG rN rO )r r r= r r© rÊ Úshapesrá r% r% r& râ ó
|
s ëzTestLikeFuncs.setup_methodc C sH |d k rD|r2t |¡ |j¡}tt ||k¡ ntt ||k¡ d S r )r r r r, r rf )r# ÚdzÚvaluer Úzr% r% r& Úcompare_array_value s
|
z!TestLikeFuncs.compare_array_valueFc C sÞ |rd|i}ni }| j D ]P\}}||fd|i|}t|j|j tt |j¡|jj t |j¡|jj t|jj |jj t|jj
|
|jj
|
|d krªt|j|j nt|jt |¡ | |||¡ ||fd|d|}t|j|j t |jj |d krt|j|j nt|jt |¡ | |||¡ ||fd|d|}t|j|j t |jj
|
|d kr~t|j|j nt|jt |¡ | |||¡ ||fd|d|}t|j|j |jj
|
rÚt |jj
|
n t |jj |d kr t|j|j nt|jt |¡ | |||¡ | j D ]@}dD ]º} ||f||| d|}
|
t|
|
j| |d krrt|
|
j|j nt|
|
jt |¡ | dks¢| dkr°|jj r°t |
|
jj n*| dksÎ| dkrÚ|jj
|
rÚt |
|
jj
|
| |
|
||¡ q0|jt|kr8tt ||f||d d|j¡t tj||dd
|
j¡ n.tt ||f||d d|j¡t |j¡ q&qG d d d tj} t d dgddgg¡ | ¡} || f|} t t| | k || fddi|} t t| | k d S )Nr r, rñ r r r$ ZCFA)r, r- r ÚK)r, r c @ s e Zd ZdS )z4TestLikeFuncs.check_like_function.<locals>.MyNDArrayNr¶ r% r% r% r& Ú MyNDArrayi s r7 r r r r ZsubokF)rÊ r r- r r Ústridesr, rS rì rU rV r5 r r1 Úndimr, Zargsortr' rH rI rJ )r# Z like_functionr3 r r# rê r, r2 r rÝ Úszr7 r0 rH r% r% r& Úcheck_like_function s
|
ÿ
|
|
|
|
ÿ
|
ÿþ ÿý ÿþ
|
ý
|
z!TestLikeFuncs.check_like_functionc C s | tjd¡ d S rþ )r; r Ú ones_likerá r% r% r& Útest_ones_liket s z TestLikeFuncs.test_ones_likec C s | tjd¡ d S r )r; r r rá r% r% r& Útest_zeros_likew s z TestLikeFuncs.test_zeros_likec C s | tjd ¡ d S r )r; r r¨ rá r% r% r& Útest_empty_likez s z TestLikeFuncs.test_empty_likec C sn | tjdd¡ | tjdd¡ | tjdd¡ | tjdd¡ tjdd | tjtjd¡ W 5 Q R X d S )Nr Tr r× gw¾/Ý^@r rÍ )r; r Ú full_liker" r rá r% r% r& Útest_filled_like} s z TestLikeFuncs.test_filled_likeÚlikefuncr, c C s| t d¡ dd¡}|d d
d d d
f }|t jkr:ddini }||fd|i|}|tkrj|jdksxtn|jdksxtd S ) Nrý r rZ r rp r, )rý r )r r )r r= r r@ r9 r8 rK )r# rB r, r0 rH rå r$ r% r% r& Útest_dtype_str_bytes s z"TestLikeFuncs.test_dtype_str_bytesN)F)rC rD rE Ú__doc__râ r5 r; r= r> r? rA r> r¼ r½ r r¨ r@ r r< r9 ÚbytesrC r% r% r% r& r- ð
|
s
|
\ ÿr- c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd d Z dS )Ú TestCorrelatec
|
C s t jdddddg|d| _t dd¡d d d
| _t jdd d
|
g|d| _t jd d d ddd dg|d| _t jddd dd dg|d| _t jdddddddg|d| _t jdd ddd d d g|d| _ t jdddddddg|d| _
|
t jd d dddd d d d g |d| _ d S ) Nr r r r rG r1 r½ r8 r r{ g Àg Àg ,Àg 4Àg :Àg Àg Àg &Àg .Àg 6Àg 0Àg $Àg Àrþ g >Àg HÀg PÀg UÀg YÀg KÀg 3À) r r rú r= Úxsr= Úz1Úz1_4Úz1rÚz2Úz2rÚzs)r# r° r% r% r& rî s ÿÿzTestCorrelate._setupc C sÞ | t¡ t | j| jd¡}t|| j t | j| jd d
d¡}t|| j t | j| jd¡}t|| j t | jd d d
| jd¡}t|| j
|
t | j| jd d d
d¡}t|| j t | j | jd¡}t|| j d S )NrÇ r8 )rî r£ r Ú correlaterú r= r rH rI rK rJ rL rG rM ©r# r4 r% r% r& r £ s
|
zTestCorrelate.test_floatc C sJ | t¡ t | j| jd¡}t|| j t | j| jd¡}t|| j d S )NrÇ ) rî r r rN rú r= r rH rK rO r% r% r& Ú test_object² s
|
|
zTestCorrelate.test_objectc C sD t d¡}t d¡}t ||¡ t|t d¡ t|t d¡ d S ©NrÐ r )r rÿ rN r ©r# rê Úkr% r% r& Útest_no_overwrite¹ s
|
|
|
z TestCorrelate.test_no_overwritec C st t jddddgtd}t jdddgtd}t jd d
|
d d d dgtd}|d d d
¡ }t j||dd}t|| d S )Nr r r y @ ð?r1 r8 y Ày @ ð?y @ ð¿rN y @ ð?y &@ @y À @y À ð¿rÇ ©Úmode)r r r¢ Ú conjugaterN r )r# rú r= Zr_zr4 r% r% r& Ú test_complexÀ s zTestCorrelate.test_complexc C sh t t¡" tjt g ¡t d¡dd W 5 Q R X t t¡" tjt d¡t g ¡dd W 5 Q R X d S )Nr× rÇ rU )r> r? r@ r rN r rÿ rá r% r% r& Útest_zero_sizeÈ s & z TestCorrelate.test_zero_sizec C s´ t d¡}t d¡}t j||dd}tt t j||dd}W 5 Q R X t|| tt t j||dd W 5 Q R X tt j||dd| tt t j||d d W 5 Q R X d S )NrÐ r ZvalidrU Úvr8 r ) r rÿ rN r rä r r r@ rr )r# rê rS Ú default_modeZ
|
valid_moder% r% r& Ú test_modeÎ s
|
|
|
|
|
|
zTestCorrelate.test_modeN)
|
rC rD rE rî r rP rT rX rY r\ r% r% r% r& rF s rF c @ s$ e Zd Zdd Zdd Zdd ZdS )Ú TestConvolvec C s: dgd }dgd }t t ||¡dd
t dd¡ d S )NrB rÐ r r r éb )r r ÚconvolverÇ rR r% r% r& rP ß s
|
|
zTestConvolve.test_objectc C sD t d¡}t d¡}t ||¡ t|t d¡ t|t d¡ d S rQ )r rÿ r_ r rR r% r% r& rT ä s
|
|
|
z TestConvolve.test_no_overwritec C s´ t d¡}t d¡}t j||dd}tt t j||dd}W 5 Q R X t|| tt t j||dd W 5 Q R X tt j||dd| tt t j||d d W 5 Q R X d S )NrÐ r rÇ rU r r8 r ) r rÿ r_ r rä r r r@ rr )r# rê rS r[ Z full_moder% r% r& r\ ë s
|
|
|
|
|
|
zTestConvolve.test_modeN)rC rD rE rP rT r\ r% r% r% r& r] Þ s r] c @ s8 e Zd Zej ddddg¡dd Zdd Zd d
|
Zd S ) Ú TestArgwherer r r r c C s¨ t d| t¡}d|d< tt |¡jd|f d|d< d|jd< tt |¡jd|f d|d< d|jd< tt |¡j|jd |f d|d< tt |¡j|j|f d S )Nr÷ F.r Tr )r r' rÜ r Úargwherer- Zflatr¡ )r# r rú r% r% r& Útest_ndý s
|
|
zTestArgwhere.test_ndc C s@ t d¡ d¡}tt |dk¡ddgddgddgddgg d S )NrN rÌ r r r )r r= r r ra ræ r% r% r& Útest_2D s ýÿzTestArgwhere.test_2Dc C s. t t dddddg¡dgdgdgdgg d S )Nr r r r r )r r ra rá r% r% r& Ú test_list s zTestArgwhere.test_listN) rC rD rE r> r¼ r½ rb rc rd r% r% r% r& r` û s
|
r` c @ s e Zd Zdd ZdS )ÚTestStringFunctionc C s t dg¡}t jdd dd tt|d t jd dd tt|d t jdd d d tt|d t jd d d tt|d
|
d S ) Nr c S s dS ©NÚFOOr% rÁ r% r% r& rI $ rJ z=TestStringFunction.test_set_string_function.<locals>.<lambda>T)Úreprrg z
|
array([1])c S s dS rf r% rÁ r% r% r& rI ) rJ Fz[1])r r Zset_string_functionr rh r9 r} r% r% r& Útest_set_string_function" s z+TestStringFunction.test_set_string_functionN)rC rD rE ri r% r% r% r& re s re c @ s$ e Zd Zdd Zdd Zdd ZdS )ÚTestRollc C s> t d¡}t |d¡}t|t dddddddd d
|
d g
|
¡ d S ) Nr6 r rZ r[ r r r r rG rN rO )r r= Úrollr r )r# rú Zxrr% r% r& Ú test_roll1d0 s
|
zTestRoll.test_roll1dc
|
C s" t t d¡d¡}t |d¡}t|t dddddgdd d
|
d d gg¡ t j|ddd }t|t d d
|
d d dgdddddgg¡ t j|ddd }t|t dddddgdd d
|
d d gg¡ t j|ddd }t|t dd d
|
d d gdddddgg¡ t j|ddd }t|t d d
|
d d dgdddddgg¡ t j|ddd }t|t d d
|
d d dgdddddgg¡ t j|ddd }t|t dddddgdd d
|
d d gg¡ t j|ddd }t|t dddddgd
|
d d dd gg¡ t j|ddd }t|t dd d
|
d d gdddddgg¡ t j|ddd }t|t d
|
d d dd gdddddgg¡ t j|ddd }t|t dddddgd d
|
d d dgg¡ t j|ddd }t|t dddddgd dd d
|
d gg¡ t j|d
|
dd }t|t dddddgdd d
|
d d gg¡ t j|ddd }t|t dddddgdd d
|
d d gg¡ d S )Nr6 ©r rG r r[ r r r r rG rN rO rZ r\ rè r¶ )r8 r )r r8 rê )r8 r8 ré rU )r r r= rk r r )r# Zx2Zx2rr% r% r& Ú test_roll2d5 s: (((((((((((((zTestRoll.test_roll2dc C s& t g ¡}tt |d¡t g ¡ d S rþ )r r r rk ræ r% r% r& Útest_roll_emptyd s
|
zTestRoll.test_roll_emptyN)rC rD rE rl rn ro r% r% r% r& rj / s /rj c @ sJ e Zd Zdddddddddddddddd d
|
dddd Zd d Zdd ZdS )Ú TestRollaxis©r r r r ©r r r r )r r r r )r r r r ©r r r r ©r r r r )r r r r )r r r r )r r r r )r r r r )ré rè r* )r r ©r r r¶ rê rË r¥ )r r r+ r2 ©r r rÌ r ©r r r¤ r. r£ ©r r c C sj t d¡ dddd¡}tt jt j|dd tt jt j|dd tt jt j|dd tt jt j|dd d S ) Nrd r r r r r| r rG )r r= r r rõ Úrollaxisr} r% r% r& Útest_exceptionsz s
|
z TestRollaxis.test_exceptionsc C sp t d¡ dddd¡ ¡ }t |j¡}t|jd | jD ].\}}t j |||d}|t
|
|j¡d \}}}} tt ||||| f |k¡ t|j| j||f kt ||f t|jd |d }
|
t j ||
|
|d}|t
|
|j¡d \}}}} tt ||||| f |k¡ t|j| jd|
|
|f k t|jd |dk rR|d n|} t j ||| d}|t
|
|j¡d \}}}} tt ||||| f |k¡ t|j| j|d| f k t|jd |d }
|
|dk rè|d n|} t j ||
|
| d}|t
|
|j¡d \}}}} tt ||||| f |k¡ t|j| jd|
|
d| f k t|jd q:d S )Nrd r r r r ÚOWNDATA)r] Ústart) r r= r rÞ r± r- r rì Útgtshapery r rf r9 ) r# r0 Zaindr Újr Zi0r r Zi3ÚipZjpr% r% r& Ú test_results s8 " zTestRollaxis.test_resultsN)rC rD rE r} rz r r% r% r% r& rp i s, õ rp c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )Ú TestMoveaxisc C s< t j ddd¡}dD ]"\}}t ||d¡j}t|| qd S )NrG rN rO ))r )rN rO rG )r )rG rO rN )r r0 )r8 r0 r8 ©r r r Úmoveaxisr- r )r# rú Úsourcer¿ rj r% r% r& Útest_move_to_end¨ s z TestMoveaxis.test_move_to_endc C s@ t j dddd¡}dD ]$\}}}t |||¡j}t|| qd S )Nr r r r ))r r rr )r r rs )r r8 rt r ©r# rú r Ú destinationr¿ rj r% r% r& Útest_move_new_position± s z#TestMoveaxis.test_move_new_positionc C sf t d¡}dddddgddgfddgddgftdtdffD ]"\}}t |||¡j}t|d q>d S ) Nrq ré )r r8 )r8 r r r8 r r )r r3 rè r r- r )r# rú r r rj r% r% r& Útest_preserve_order» s
|
ú z TestMoveaxis.test_preserve_orderc C s t d¡}ddgddgdfddgddgdfdddgdddgdfddgddgdfddgddgdffD ]$\}}}t |||¡j}t|| qbd S )Nr r r r r )r r r r )r r r r )r r3 r r- r r r% r% r& Útest_move_multiplesÈ s
|
ûz TestMoveaxis.test_move_multiplesc C sÀ t j ddd¡}tt jdt j|dd tt jdt j|dd tt jdt j|dd ttd t j|ddgddg ttd
|
t j|ddgddg ttd t j|dddg ttd t j|ddgdg d S ) Nr r r zsource.*out of boundsr rU zdestination.*out of boundsrG zrepeated axis in `source`z repeated axis in `destination`zmust have the same number)r r r r
|
rõ r r@ ræ r% r% r& Ú test_errorsÔ sV ÿ ÿ ÿ ÿ ÿ ÿ ÿzTestMoveaxis.test_errorsc C st t j d¡}t |dd¡}t|j|j tt|t jj dddg}t |dd¡}t|t| tt|t j d S )NrÎ r r r r )
|
r rû r3 r r r- r Z MaskedArrayré rH )r# rú r$ r% r% r& Útest_array_likeså s
|
z TestMoveaxis.test_array_likesN) rC rD rE r
r r r r r r% r% r% r& r § s
|
r c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )Ú TestCrossc C sF ddg}ddg}d}t ||¡}t|| t ||¡}t|| d S )Nr r r r r )r Úcrossr ©r# ÚurZ r4 Úcpr% r% r& Útest_2x2ò s
|
zTestCross.test_2x2c C sT ddg}dddg}t dddg¡}t ||¡}t|| t ||¡}t|| d S ) Nr r r r rG r6 r| r ©r r r r r r% r% r& Útest_2x3û s
|
|
zTestCross.test_2x3c C sV dddg}dddg}t dddg¡}t ||¡}t|| t ||¡}t|| d S )Nr r r r rG rN r{ r r r% r% r& Útest_3x3 s
|
|
|
zTestCross.test_3x3c C sÈ t ddgd¡}t ddgd¡}d}tt ||¡| tt ||¡| tt ||¡d t ddgd¡j}t dddgd¡}t d d
|
dgd¡}tt j||dd | tt ||j¡| tt ||¡d t dddgd¡j}t ddgd¡j}t d d dgd¡}tt j||ddd| tt |j|j¡| tt |j|j¡d t dddgd¡}t dddgd¡j}t dddgd¡}tt j||dd| tt |j|¡| tt ||¡d d S )Nr r )ri r r r r r rG r6 r| )Úaxisar> r[ ©r Úaxisb)rG r rN r{ )r )r Ztiler r rë ©r# r rZ r4 r% r% r& Útest_broadcasting s0 zTestCross.test_broadcastingc C s t d¡}t d¡}tt ||¡jd t d¡}t d¡}tt j||dddjd tt jt j||dd
|
d tt jt j||d dd t d ¡}t d ¡}tt j||dd
|
djd tt jt j||dd
|
d tt jt j||ddd t d¡}tdd
|
D ] }tt j|||djd qød S )N)r r r )rG r )r rG r )r6 r rG rm r r r )r6 rG r r r )r6 r rG rO )rG rO r )r Úaxisc)r6 rG r rO r| rU )r r r r )r rx )r rÿ r r r- r rõ rè )r# r rZ r r% r% r& Útest_broadcasting_shapes+ s
|
|
|
|
|
|
|
z"TestCross.test_broadcasting_shapesc C sl t dddggt j¡}t dddgt j¡}t jddd ggt jd
|
}tt ||¡| tt ||¡| d S ) Néà rZ r[ éú é¦ éD i¶ i+ i^ÿÿr1 )r r rÀ r r r r r% r% r& Ú test_uint8_int32_mixed_dtypes> s
|
z'TestCross.test_uint8_int32_mixed_dtypesN) rC rD rE r r r r r r¡ r% r% r% r& r ñ s r c C sj t d¡} t d¡}t ddd¡}t jdd}t jdd}t | ||¡}t|| tt |||¡| d S ) Nr/ r÷ r r rG )rG rG )r- rm )r rÿ ZlinspacerH Úouterr )Zarr1Zarr2Zarr3Zout1Zout2Zres1r% r% r& Útest_outer_out_paramG s
|
|
|
r£ c @ sf e Zd Zdd Zdd Zdd Zdd Zej d e
|
j e
|
j e
|
j e
|
jg¡ej d
|
d d d g¡dd ZdS )Ú TestIndicesc
|
C sr t d¡\}}t|t dddgdddgdddgdddgg¡ t|t dddgdddgdddgdddgg¡ d S )Nr r r r r ©r r± r r rÓ r% r% r& Ú test_simpleT s ýýzTestIndices.test_simplec C sP t d¡\}t|t ddddg¡ t jddd\}t|t ddddg¡ d S )N)r r r r r T©Úsparser¥ ræ r% r% r& Útest_single_input_ s z TestIndices.test_single_inputc C sP t g t d¡ t g tjddd t g gt d¡ t g gtjddd d S )Nr% Tr§ r) )r r r± rá r% r% r& Útest_scalar_inputf s z TestIndices.test_scalar_inputc C sN t jddd\}}t|t dgdgdgdgg¡ t|t dddgg¡ d S )Nr Tr§ r r r r r¥ rÓ r% r% r& rð l s zTestIndices.test_sparser, Údimsr% r) r c C sD t j||d}t|j|k t j||ddD ]}t|j|k q,d S )Nr1 T)r, r¨ )r r± r r, )r# r, r« ZindsrX r% r% r& rô q s z TestIndices.test_return_typeN)rC rD rE r¦ r© rª rð r> r¼ r½ r r r r4 r rô r% r% r% r& r¤ R s r¤ c @ sh e Zd Zddddddddd d
|
d d g Zd d Zdd Zdd Zdd Zdd Zdd Z dd Z
|
dd Z d S ) Ú TestRequirerñ r~ Z
|
CONTIGUOUSr r} ZFORTRANr$ ZALIGNEDÚWZ WRITEABLEr
r{ c C s| t ddd|fg¡}|jdd |d }t|jd t|jd t|jd t|jd t|jd
|
|S ) Nrv )Zjunkr r0 F)Úwriterñ r r
r r$ )r r3 Zsetflagsr rì )r# r, rX r0 r% r% r& Úgenerate_all_false s z TestRequire.generate_all_falsec C sv |d kr|j }t |||g¡}t|j| t|j |k t |d |g¡}|d dkrdt||k nt|j| d S )Nr r
)r, r Úrequirer rì )r# Úflagr, rX rH r´ r% r% r& Úset_and_check_flag s z TestRequire.set_and_check_flagc C sJ ddg}d ddg}t ||| j¡D ]"\}}}| |¡}| |||¡ q"d S )Nrc r rf )r³ r´ Ú
|
flag_namesr¯ r² )r# ÚidÚfdZidtypeZfdtyper± r0 r% r% r& Útest_require_each s
|
|
|
z TestRequire.test_require_eachc C s | d¡}tttj|d d d S )Nrc ÚQ)r¯ r ÚKeyErrorr r° r} r% r% r& Útest_unknown_requirement¤ s
|
z$TestRequire.test_unknown_requirementc C sj t ddddgddddg¡}t|jd t|jd t|jd t|jdk t|ddddg d S ) Nr r r r r rñ r$ r
)r r° r rì r, r r} r% r% r& Útest_non_array_input¨ s z TestRequire.test_non_array_inputc C s$ | d¡}tttj|d ddg d S )Nrc rñ r )r¯ r r@ r r° r} r% r% r& Útest_C_and_F_simul° s
|
z TestRequire.test_C_and_F_simulc C s@ G dd dt j}|d}t |d dg¡}tt|t jk d S )Nc @ s e Zd ZdS )z4TestRequire.test_ensure_array.<locals>.ArraySubclassNr¶ r% r% r% r& Ú ArraySubclassµ s r¼ rv ÚE)r rH r° r rJ )r# r¼ r0 rH r% r% r& Útest_ensure_array´ s z TestRequire.test_ensure_arrayc C s8 G dd dt j}| jD ]}|d}| |d |¡ qd S )Nc @ s e Zd ZdS )z8TestRequire.test_preserve_subtype.<locals>.ArraySubclassNr¶ r% r% r% r& r¼ ½ s r¼ rv )r rH r³ r² )r# r¼ r± r0 r% r% r& Útest_preserve_subtype¼ s
|
z!TestRequire.test_preserve_subtypeN) rC rD rE r³ r¯ r² r¶ r¹ rº r» r¾ r¿ r% r% r% r& r¬ { s& ü r¬ c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd S ) Ú TestBroadcastc C s( t d¡t d¡t d¡t d¡g}t j| t t j|d d
t j|dd
¡t t j|d d
t j|dd
¡t t j|d d
t j|dd
¡t |d t j|dd
|d ¡g}|D ]Z}t|jd t|jd
|
t|jd
|
t|jd t||j D ]\}}t
|
||j k qqÈd S ) NrÍ )rG rN r )rO )rG r rO r r r r8 r0 r r ) r r' Ú broadcastr r- r9 r Únumiterrï r r Úbase)r# ÚarrsZmitsÚmitr0 Úiar% r% r& Útest_broadcast_in_argsÆ s ÿ&&&"ü z$TestBroadcast.test_broadcast_in_argsc C sb t d¡g}t j| }t|jd t|jd t|jd t|jd t|d |j d j
|
k d S )Nr0 r r r ) r r' rÁ r r- r9 r r r r rà )r# rÄ rÅ r% r% r& Útest_broadcast_single_arg× s
|
z'TestBroadcast.test_broadcast_single_argc C sX t d¡}tdD ]@}|g| }|dkr<ttt jf| qt j| }t|j| qd S )Nr/ é# rö )r r' rè r r@ rÁ r r )r# rX r~ rÄ rÅ r% r% r& Útest_number_of_argumentsá s
|
|
|
z&TestBroadcast.test_number_of_argumentsc C s t d¡g}t j| }t j|i }t|j|j t|j|j t|j|j t|j|j t|j d j
|
|j d j
|
k t t t jdfddi d S )Nr0 r r rú ) r r' rÁ r r- r9 r r r r rà r r@ )r# rÄ rÅ Zmit2r% r% r& Útest_broadcast_error_kwargsë s
|
z)TestBroadcast.test_broadcast_error_kwargsc C s@ t jtdd( t dddggdgdggdd g¡ W 5 Q R X d S )
|
Nz5arg 0 with shape \(1, 3\) and arg 2 with shape \(2,\)r: r r r r rG rN rO )r> r? r@ r rÁ rá r% r% r& Ú!test_shape_mismatch_error_messageø s z/TestBroadcast.test_shape_mismatch_error_messageN)rC rD rE rÇ rÈ rÊ rË rÌ r% r% r% r& rÀ Å s
|
|
|
rÀ c @ s& e Zd ZG dd dejZdd ZdS )Ú TestKeepdimsc @ s e Zd ZdddZdS )zTestKeepdims.sub_arrayNc C s t jj| |||ddS )NTr )r rH r® )r# r] r, rS r% r% r& r® s zTestKeepdims.sub_array.sum)NNN)rC rD rE r® r% r% r% r& Ú sub_array s rÎ c C s, | j }t d¡ |¡}tttj|dd d S )Né Tr )rÎ r r= rI r rr r® )r# Z sub_classrú r% r% r& Ú
|
test_raise s zTestKeepdims.test_raiseN)rC rD rE r rH rÎ rÐ r% r% r% r& rÍ þ s rÍ c @ s e Zd Zdd Zdd ZdS )Ú TestTensordotc C sL t d¡}t d¡}t ||d¡}t|t ||¡ t|t d||¡ d S )Nrw ru r¶ zij,jk)r rH Ú tensordotr ÚdotZeinsum)r# r0 rH Útdr% r% r& Útest_zero_dimension s
|
|
|
z!TestTensordot.test_zero_dimensionc C s* t d¡}t ||g g f¡}t|| d S rþ )r r rÒ r )r# Zarr_0dÚretr% r% r& Útest_zero_dimensional s
|
z#TestTensordot.test_zero_dimensionalN)rC rD rE rÕ r× r% r% r% r& rÑ
|
s rÑ )KrT rr r³ r6 r> rÇ Údecimalr r r Z
|
numpy.corer Z numpy.randomr r r Z numpy.testingr r r r
|
r r r r r r r Znumpy.core._rational_testsr Z
|
hypothesisr r rÉ Zhypothesis.extrarÀ r rM rÊ rÐ rÙ rô r r8 rV Ú Exceptionrµ r· rß r& r) r8 rA rW rX rÊ ré r r r r- rF r] r` re rj rp r r r£ r¤ r¬ rÀ rÍ rÑ r% r% r% r& Ú<module> s~ 4 < Z!WdSh ?k R*l vU ' G 'H %:>JV )J9
|
