w5
d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlmZ d dlmZ d dlZd dlZd dlmZ d dlZd dlZd dl m c m Z d dl m Z d dl!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3 d dl4m5Z5m6Z6 d d l7m8Z8 d d
l9m:Z: d d l;mZ>m=Z= d
Z?e@ddfdZA G d d ZB G d d ZC G d d ZD G d d ZE G d d ZF G d d ZG G d d ZH G d d ZI G d d! ZJ G d" d# ZK G d$ d% ZL G d& d' ZM G d( d) ZN G d* d+ ZO G d, d- ZP G d. d/ ZQ G d0 d1 ZR G d2 d3 ZS G d4 d5 ZT G d6 d7 ZU G d8 d9 ZV G d: d; ZW G d< d= ZX G d> d? ZY G d@ dA ZZ G dB dC Z[ G dD dE Z\ G dF dG Z] G dH dI Z^ G dJ dK Z_ G dL dM Z` G dN dO Za G dP dQ Zb G dR dS Zc G dT dU Zd G dV dW Ze G dX dY ZfdZ Zgd[ Zhd\ Zi G d] d^ Zj G d_ d` Zk G da db Zl G dc dd Zm G de dfem Zn G dg dhem Zo G di dj Zpdk Zq G dl dm Zr G dn do Zs G dp dq Ztd drdsdtdudvZuejv w dwe@egdxdygz G d{ d| Zx G d} d~ Zy G d d Zz G d d Z{d dl|m}Z} G d d Z~ G d d Z G d de Z G d d Z G d d Zd Zd Zd Zd Zd Zd Zd Z G d d Z G d d Z G d d Z G d d Ze+s
G d d Z G d d Z G d d Z G d d Z G d d Z G d d Z G d d Zd dl!m+Z+ G d d Z G d d Z G d d Z G d d Zd Zd Zejv w dej ej g ejv w ddgdgdgfddg d Zejv w ddgg d d Zejv w dej ej ej ej ej ej g d Zd Zejv w dej ej ej ej ej ej g d Zejv w ddge j ej fe j ej fg ejv w ddrej ej ej g ejv w ddrej ej ej g ejv w dej dDŽ ZdȄ ZdɄ Z G dʄ d˦ Zd̄ Z G d̈́ dΦ Zejv e3dϬЦ ejv w d ej drdҦ ejv w dg dԢ dՄ Zdք Zejv w d ej drdҦ ejv w dg dע d Zdل Zdڄ Zejv w d ej drdҦ ejv w dej ej g dۄ Zejv w d ej dsdҦ ejv w dej ej ej ej g d܄ ZdS ) ) annotationsN)contextmanager)pickle)Decimal)rational)assert_
assert_raises assert_warns assert_equalassert_almost_equalassert_array_equalassert_raises_regexassert_array_almost_equalassert_allcloseIS_PYPY IS_PYSTON HAS_REFCOUNTassert_array_less runstringtemppathsuppress_warnings break_cycles
_SUPPORTS_SVE)requires_memory _no_tracing)CommaDecimalPointLocale)
repack_fields)_get_ndarray_c_version) timedeltadatetimec t | | t j | t t j | t j t | d S N)r npsortarangelen)arrargs t/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/core/tests/test_multiarray.pyassert_arg_sortedr* , sL S273<<((( ryS 2 233333 Cc t j | }|t j t k r(|t d t j | || S ||j }t
| d s| f} t j t j
| |j z }t j |d|z z dz t j
}|j d d }||z }|dk r||z
}|d|z z dk r||z
}||||z dz dd }| d t j | |||
}|S ) z
Allocate a new ndarray with aligned memory.
The ndarray is guaranteed *not* aligned to twice the requested alignment.
Eg, if align=4, guarantees it is not aligned to 8. If align=None uses
dtype.alignment.Nz$object array alignment not supporteddtypeorder__len__ datar r0 )r# r/ object
ValueErrorzeros alignmenthasattr functoolsreduceoperatormulitemsizeemptyuint8__array_interface__fillndarray) shaper/ r0 alignsizebufptroffsetr4 s r) _aligned_zerosrL 2 sK
HUOOE CDD DxU%8888 } 5) $ $ HL% 0 05> AD
(4!E'>A%rx
0
0C
!&
)!
,C
5[F
{{ qwA % fVD[]"
#CRC
(CHHQKKK
:eUCu 5 5 5D Kr+ c e Zd Zd Zd Zd Zd Zd Zej
e d d Z d Z
d
Zej g d g d d
Zd Zd Zd ZdS ) TestFlagsc 8 t j d | _ d S N
)r# r% aselfs r) setup_methodzTestFlags.setup_methodW s 2 r+ c t }d| j j _ t t
t d| t t
t d| d| j j _ d| j d< d| j d< d S )NFz
self.a[0] = 3zself.a[0:1].itemset(3)T r )localsrR flags writeabler r8 r )rT mydicts r) test_writeablezTestFlags.test_writeableZ sf !& j)_fEEEj)-EvNNN!% q q r+ c , t j d } G d dt j }| | }|d }d|j _ d|j _ d|j _ t j d } G d d }t j | }|d }d|j _ d|j _ d|j _ d|j _ t t 5 d|j _ d d d d S # 1 swxY w Y d S ) NrQ c e Zd ZdS )3TestFlags.test_writeable_any_base..subclassN__name__
__module__ __qualname__ r+ r) subclassr_ h Dr+ re .FTc e Zd Zd ZdS )8TestFlags.test_writeable_any_base..frominterfacec , || _ |j | _ d S r" )r' rC rT r' s r) __init__zATestFlags.test_writeable_any_base..frominterface.__init__u s +.+B(((r+ N)ra rb rc rk rd r+ r)
frominterfacerh t s(
C
C
C
C
Cr+ rl ) r# r% rE viewrY rZ asarrayr r8 )rT r' re view1view2rl s r) test_writeable_any_basez!TestFlags.test_writeable_any_basec sr imm rz ""c
# % $ imm C C C C C C C C
=))c
% $ % %
:
&
& ) )$(EK ! ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s /
D D
D
c * d}t j |d }t t |j d t j ddg }t j j || }|d }t t |j d d S )Nd BTwritevalsu1res3S4rx ) r#
frombufferr r8 setflagsr/ corerecords
fromstringrT r4 rx typesvaluess r) test_writeable_from_readonlyz&TestFlags.test_writeable_from_readonly s }T3''j$-t<<<<>>:<< ++D%88f~j$-t<<<<<>:<< ++D%88f~
$%%%
E
"""
$-...
D
!!!
$%%%%%r+ zPyPy always copiesreasonc " dd l }t j d }t |j D ]a} |j |j || }t |j j t t |j t bd S )Nr )
r r# r% rangeHIGHEST_PROTOCOLloadsdumpsr rY rZ
isinstancebasebytes)rT r rR vrx s r) test_writeable_picklez TestFlags.test_writeable_pickle s
IdOOv.// 2 2A 6< Q 2 233D DJ( ) ) ) Jty%00 1 1 1 1 2 2r+ c ddl m} |d }|j j rJ |j j sJ |d }d|j _ |j j rJ d|j _ |j j sJ d|j _ |d }|j j rJ |j j rJ ||fD ] }|d }d|j _ d|j _ |j j rJ t t 5 d|j _ d d d n # 1 swxY w Y t j 5 t j dt t t 5 d|j _ d d d n # 1 swxY w Y d d d n # 1 swxY w Y t t 5 d|j _ d d d n # 1 swxY w Y d S )Nr )get_c_wrapping_arrayT.Ferror) numpy.core._multiarray_testsr rY owndatarZ r r8 warningscatch_warnings simplefilterDeprecationWarningr
)rT r
arr_writeablerm arr_readonlyr' s r) test_writeable_from_c_dataz$TestFlags.test_writeable_from_c_data s FEEEEE,,T22
&.... ",,,, S! %
:'''' #
z####(-
%++E22 %---- %////!<0 + +Cs8D#(DJ "'CI y* * ** z**
,
,'+
$
,
,
,
,
,
,
,
,
,
,
,
,
,
,
, (**
/
/%g/ABBB"#566 / /*.CI'/ / / / / / / / / / / / / / /
/
/
/
/
/
/
/
/
/
/
/
/
/
/
/
011
+
+&* #
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+ + +sZ &
C??D D /E<
E% E<%E) )E<,E) -E<<F F
F44F8 ;F8 c t j d }d|j _ t j d 5 }t j d d|d< d|d< t t | dk d d d d S # 1 swxY w Y d S )NrQ Trecordalwaysr3 r2 ) r# r% rY _warn_on_writer r filterwarningsr r& )rT rR ws r) test_warnonwritezTestFlags.test_warnonwrite s IbMM!%
$D
1
1
1 !Q #H - - -AaDAaD CFFaK ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !s ?BB B)flag
flag_valuerZ ))rZ TT)r TF)rZ FFc , t j d t j || G fdd }t j |usJ j d d |usJ t j | j j |u sJ d S )NrQ c e Zd Z j ZdS )5TestFlags.test_readonly_flag_protocols..MyArrN)ra rb rc __array_struct__rR s r) MyArrr s 1 r+ r r4 r3 ) r# r% setattrrY
memoryviewreadonlyrC rn rZ )rT r r rZ r rR s @r) test_readonly_flag_protocolsz&TestFlags.test_readonly_flag_protocols s
IbMMz*** 2 2 2 2 2 2 2 2 2 2 !}}%Y6666$V,Q/y@@@@z%%''""(2i??????r+ c t | j j j d t | j j d d t | j j j d t | j j j d t | j j j d t | j j j d t | j j j d t | j j j d t | j j j
d t | j j j d t | j j d d t | j j d d d S )NTr, FXWRITEBACKIFCOPY) r rR rY carrayfarraybehavedfncforcr rZ alignedwritebackifcopyrS s r) test_otherflagszTestFlags.test_otherflags s# TV\($///TV\#&---TV\(%000TV\)4000TV\%u---TV\&---TV\)4000TV\+T222TV\)4000TV\15999TV\#&...TV\"34e<<<<= 2&+ &+ &+P! ! ! [ @ @ @
*
*
*+ + @ @
+ + @ = = = ! ! !! ! ! ! !r+ rN c e Zd Zd ZdS )TestHashc t j t j dft j t j dft j t j dft j t j dffD ]\ }}}t d| D ]v}t t |d|z t d|z d||fz t t |d|dz
z t d|dz
z d ||dz
fz t t |d|z dz
t d|z dz
d
||fz t |dz
d }t t |d|dz
z t d|dz
z d ||dz
fz t t |d|z dz
t d|z dz
d
||fz xd S ) N @ r3 r2 z
%r: -2**%derr_msgz %r: 2**%dz
%r: 2**%d - 1)
r# int8rB int16uint16int32uint32int64uint64r r hashmax)rT stutsis r) test_intzTestHash.test_int s0 7BHa0 8RY3 8RY3 8RY35 @ @IBA 1a[[
@
@ T""adU))__dAqD5kk%1RG%;= = = = T""QQZ.. 1 14AE
3C3C%0BA;%>@ @ @ @ T""QTAX,, / /adQh %4Aw%>@ @ @ @ AqMM T""QQZ.. 1 14AE
3C3C%0BA;%>@ @ @ @ T""QTAX,, / /adQh %4Aw%>@ @ @ @ @
@ @ @r+ N)ra rb rc r rd r+ r) r r s( @ @ @ @ @r+ r c J e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d S ) TestAttributesc t j d | _ t j d dd | _ t j dt j ddd | _ d S ) NrQ r rW < r r2 )r# r% onereshapetwofloat64threerS s r) rU zTestAttributes.setup_method s^ 9R==9R==((A..Yr444<>>TX]A&&&TX]A&&&TZ_a(((h TX]B'''TX_bf---TX&(?@@@TX]BIbMM22222r+ c : t | j j t j t j t | j j t j t j t | j j j d t | j j j d t | j j j d dv t | j j j d d t | j j j d d d S )Nldr <>r3 r f)
r r r/ r# int_r float_charr strrS s r) test_dtypeattrz TestAttributes.test_dtypeattr: s TX^RXbg%6%6777TZ%rx ':':;;;TX^(#...TZ%*C000
$Q'4/000TX^'*C000TZ%)!,c22222r+ c t t j d }t t |t d S Nr )r# r r r int)rT numpy_ints r) test_int_subclassingz#TestAttributes.test_int_subclassingC s4 GAJJ Jy#.../////r+ c
| j fd}t |ddd t j g d t t
|ddd t t
|ddd t t
|ddd t |ddd
t j dgdz t t
|d d t j dd
g |d
d
d d S )Nc ` t j | t |j z |j z S )N)bufferr/ rK r )r# rE r
r@ )rH rK r xs r)
make_arrayz3TestAttributes.test_stridesattr..make_arrayN s6 :d1C%+AJ%6&-aj&8: : :
:r+ r r5 r r r2 r3 r2 r r r3 r r2 r rW rQ )r r r# arrayr r8 )rT r r s @r) test_stridesattrz TestAttributes.test_stridesattrK s H : : : : :
ZZ1b))28LLL+A+ABBBj*aB777j*aB777j*aA666ZZ1a(("(A3q5//:::j*fa2r(9K9KLLL
1a r+ c | j fd}t |ddd t j g d t |ddd t j g d t t
|ddd t t
|dd
d t t |d dd t j j t j
d d d
d }t t
|dj z j f t j j t j
dt j d dd t t
|d d d d d d d }d|_
d
|d d d
_
t j d }d|_
t t ||d d S )Nc t j | gt |j z }n!# t $ r}t | d }~ww xY w|j z x|_ }|S )N)r/ r rK )r# rE r
r@ Exception RuntimeErrorr )rH rK r rer s r) r z7TestAttributes.test_set_stridesattr..make_array_ su
&JvS&,QZ&79 9 9
&
&
&"1oo%
&")!*"4 4AIHs &*
AAAr r5 r r r3 )r r rW r r r r r2 r rQ rQ r r c || _ d S r" r )r' r s r) set_stridesz8TestAttributes.test_set_stridesattr..set_stridesq s
!CKKKr+ rQ r r r5 )rF r r rd )r r r# r r r8 r lib
stride_tricks
as_stridedr% r@ r r TypeError)rT r r$ rR arr_0dr s @r) test_set_stridesattrz#TestAttributes.test_set_stridesattr\ s H ZZ1b))28LLL+A+ABBBZZ1a(("(3H3H3H*I*IJJJj*aB777j*aB777lJ1a888
F + +BIaLL(F K K " " " j+q2aj=!*2MNNN
F + +BIb,H,H,H,L:?
,
P
Pj+q2w;;;
dddG ##A# !ifd;;;;;r+ c dD ]V}t j d| }t j d| }| d d|d< t || Wd S )Nz?bhilqpBHILQPfdgFDGOr r2 r3 r3 .)r# rA rD r )rT tr ys r) test_fillzTestAttributes.test_fill sh ' AA&&AA&&A
FF1IIIAcF A r+ c t j dt j }t j dt j }d}||d< | | t || d S )Nr- r l .)r# rA r rD r
)rT r r/ values r) test_fill_max_uint64z#TestAttributes.test_fill_max_uint64 sa HYbi 0 0 0 HYbi 0 0 0# u
1a r+ c v t j ddgd }| |d t |d d |d d t j dd d
g }| d t |d d
d
g t |d ddg d S )N)r )r3 ?i4,f8r r f1r3 r2 rR f8r ) @r rR r; r r )r# r rD r r9 r
rT r s r) test_fill_struct_arrayz%TestAttributes.test_fill_struct_array s Hh) 9 9 9 qt QtWQZ4,,,
HQ{K 8 9 9 9 y1S6C:...1S6B8,,,,,r+ c t j d }| d t j t
d 5 | d d d d d S # 1 swxY w Y d S )N Fru z .*read-onlymatchr )r# r9 r~ r raisesr8 rD r s r) test_fill_readonlyz!TestAttributes.test_fill_readonly s HRLL
]:]
;
;
;
FF1III s A))A-0A-N)
ra rb rc rU r r r r r+ r0 r3 r= rC rd r+ r) r r s F F F
3 3 3*3 3 30 0 0 "%< %< %.assign* s AcFFFr+ )r# r% r r r r8 rT rR r{ s r) test_assignment_broadcastingz+TestAssignment.test_assignment_broadcasting s IaLL A & & 1 #QIII.///1 %%a++#QIII./// 1 ddd#++Aq!44#QIII./// j&!RYr]]-B-B1a-K-KLLLLLr+ c G d d }t j d fd}t t t f| | t t
|dg d S )Nc e Zd ZdS )0TestAssignment.test_assignment_errors..CNr` rd r+ r) r, r 1 rf r+ r, r3 c | d< d S r rd )r rR s r) r{ z5TestAssignment.test_assignment_errors..assign5 s AaDDDr+ )r# r9 r AttributeErrorr) r8 )rT r, r{ rR s @r) test_assignment_errorsz%TestAssignment.test_assignment_errors/ s HQKK ~y161133???j&1#.....r+ c T ddl m t fd }t j dg }t j d } |d 5 ||d< d d d n # 1 swxY w Y t |d d t j t j d g d S )Nr )set_string_functionc 3 v K fdd dV dd dS # dd w xY w)z% replace ndarray.__str__ temporarily c S r" rd )r r s r) zLTestAssignment.test_unicode_assignment..inject_str..B s ! r+ F)reprNrd )r r s `r)
inject_strz:TestAssignment.test_unicode_assignment..inject_str? sm
% 8 8 8 8
6##Du555555##Du55555s ) 8testdonebadu åäö)numpy.core.numericr r r# r r )rT r a1da0dr s @r) test_unicode_assignmentz&TestAssignment.test_unicode_assignment; s :::::: 6 6 6 6
6 hx hv
Z
CF SVV$$$ "(>**+,,,,,s A A$'A$c t j dg }t j dg } G d d }t t t j |dg t t t j |dg t t t j |d | t t t j |d | d S )Nr s donec e Zd Zd Zd ZdS )?TestAssignment.test_stringlike_empty_list..bad_sequencec d S r" rd rS s r) __getitem__zKTestAssignment.test_stringlike_empty_list..bad_sequence.__getitem__W s r+ c t r" r rS s r) r1 zGTestAssignment.test_stringlike_empty_list..bad_sequence.__len__X s \ 1r+ Nra rb rc r r1 rd r+ r) bad_sequencer V s" ' ' ' 1 1 1 1 1r+ r r )r# r r r8 r> setitem)rT ur r s r) test_stringlike_empty_listz)TestAssignment.test_stringlike_empty_listQ s HfX HgY 2 2 2 2 2 2 2 2 j("2Aq"===j("2Aq"===j("2Aq,,..IIIj("2Aq,,..IIIIIr+ c t j t j fD ]W}t j t j d d | }t j t j d d | }t j |g }t
|d | ||d< t
|d | ||d< t
|d | |d |d< t
|d | |d |d< t
|d | t j t j | g }t
|d | Yd S )Nr r3 r5 r ..)r#
longdouble longcomplex nextafterastyper r )rT r/ tinybtinyatiny1dr' s r) test_longdouble_assignmentz)TestAssignment.test_longdouble_assignment` sT mR^4 ( (EL q!1!1155<i4rR r rR r )r r# r/ rS s r) test_byteordersz TestDtypedescr.test_byteorders sV 28E??2333, ((BHl^,D,DDEEEEEr+ c ddg}t j d|f }t t | d t j d| }t t | d d S )N)rR .
AaD r+ c | d S r rd r s r) r z5TestZeroRank.test_invalid_subscript.. r r+ c B | t j g t S r" r# r r
r s r) r z5TestZeroRank.test_invalid_subscript.. Abhr3.?.?,@ r+ c B | t j g t S r" r r s r) r z5TestZeroRank.test_invalid_subscript.. r r+ r r
IndexErrorr s r) test_invalid_subscriptz#TestZeroRank.test_invalid_subscript sg v1j..!444j..!444j"@"@!DDDj"@"@!DDDDDr+ c | j \ }}d|d< t |d d|d< t | d d S )N* . r r itemr s r) "test_ellipsis_subscript_assignmentz/TestZeroRank.test_ellipsis_subscript_assignment sM v1#Q#QVVXXr"""""r+ c | j \ }}d|d< t |d d|d< t | d d S )Nr rd r r r s r) test_empty_subscript_assignmentz,TestZeroRank.test_empty_subscript_assignment sM v1"Q"QVVXXr"""""r+ c | j \ }}d }t t ||dd t t ||dd t t ||dd d S )Nc || |< d S r" rd r r r s r) r{ z>TestZeroRank.test_invalid_subscript_assignment..assign AaDDDr+ r r r rd )r r r r8 )rT rR r r{ s r) !test_invalid_subscript_assignmentz.TestZeroRank.test_invalid_subscript_assignment sf v1 j&!Q333j&!Q333j&!R44444r+ c | j \ }}t |t j j d t |dt j f j d t |t j df j d t |dt j f j d t |t j dt j f j d t |dt j t j f j d t |t j t j df j d t |t j fdz j d d S Nr3 .r3 r3 rQ )
r3 r3 r3 r3 r3 r3 r3 r3 r3 r3 r r r# newaxisrF r s r) test_newaxiszTestZeroRank.test_newaxis s v1Qrz]($///QsBJ'-t444Qrz3'-t444QsBJ'-t444Qrz3
2396BBBQsBJ
2396BBBQrz2:s2396BBBQ
}R'(.88888r+ c | j \ }}d }t t ||t j df t t ||t j fdz d S )Nc | | d S r" rd r r s r) subscriptz4TestZeroRank.test_invalid_newaxis..subscript
aDDDDr+ r 2 r r r r# r )rT rR r r s r) test_invalid_newaxisz!TestZeroRank.test_invalid_newaxis s[ v1 j)QQ@@@j)Q
b0@AAAAAr+ c t j d }d|d< t |d d t j d| }d|d< t |d d t j t
5 t j dd d d d n # 1 swxY w Y t j t
5 t j dd d d d d S # 1 swxY w Y d S )Nrd rW )r r r2 r# )r# rE r r rB r8 rT r r/ s r) test_constructorz TestZeroRank.test_constructor sb JrNN"QrUA Jr! $ $ $"QrUA ]:
&
& ) ) JtR ( ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) )
]:
&
& ) ) Jr4 ( ( ( ( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s$ :B B!$B!C%%C),C)c t t j d }t t t j |dg| d S )Nr2 r3 )r# r r r8 addr< s r) test_outputzTestZeroRank.test_output s/ HQKKj"&!aS!44444r+ c B t j d }|j }|j }t |t j d t t
| t j u t |j j d t |j j
d t |t j d t t
| t j u t |j j d t |j j
d d S )N ?r Tr3 ) r# r realimagr r r rE rY
contiguousre )rT r xrxis r) test_real_imagzTestZeroRank.test_real_imag s HRLL
V
VR!%%%RBJ&'''RX($///RX*D111R!%%%RBJ&'''RX($///RX*D11111r+ N)ra rb rc rU r r r r r r r r r r r rd r+ r) r r s 4 4 4" " "$ $ $E E E# # ## # #5 5 5 9 9 9B B B ) ) ) 5 5 52 2 2 2 2r+ r c > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestScalarIndexingc H t j ddg d | _ d S Nr r3 )r# r r rS s r) rU z TestScalarIndexing.setup_method s 1a&!!!$r+ c v | j }t |d d t |d j d d S )N.r rd r r rF r s r) r z*TestScalarIndexing.test_ellipsis_subscript s9 FQsVQ QsV\2&&&&&r+ c v | j }t |d d t |d j d d S )Nrd r r r s r) r z'TestScalarIndexing.test_empty_subscript s9 FQrUA QrU["%%%%%r+ c p | j }t t d | t t d | d S )Nc | d S r rd r s r) r z;TestScalarIndexing.test_invalid_subscript.. r r+ c B | t j g t S r" r r s r) r z;TestScalarIndexing.test_invalid_subscript.. r r+ r r s r) r z)TestScalarIndexing.test_invalid_subscript s9 Fj..!444j"@"@!DDDDDr+ c J | j }d }t t ||dd d S )Nc || |< d S r" rd r s r) r{ zDTestScalarIndexing.test_invalid_subscript_assignment..assign r r+ r r )r r r) r} s r) r z4TestScalarIndexing.test_invalid_subscript_assignment s6 F iAr22222r+ c | j }t |t j j d t |dt j f j d t |t j df j d t |dt j f j d t |t j dt j f j d t |dt j t j f j d t |t j t j df j d t |t j fdz j d d S r r r s r) r z TestScalarIndexing.test_newaxis" s FQrz]($///QsBJ'-t444Qrz3'-t444QsBJ'-t444Qrz3
2396BBBQsBJ
2396BBBQrz2:s2396BBBQ
}R'(.88888r+ c | j }d }t t ||t j df t t ||t j fdz d S )Nc | | d S r" rd r s r) r z:TestScalarIndexing.test_invalid_newaxis..subscript0 r r+ r r r )rT rR r s r) r z'TestScalarIndexing.test_invalid_newaxis- sW F j)QQ@@@j)Q
b0@AAAAAr+ c , t j d }|dd |d d< t |g d t j d }|d d |dd < t |g d t j d }|d d d |d d < t |g d t j d dd }|d d d d df |d d dd d f< t |g d
g d g t j d dd }|d d d d df |d d dd d df< t |g d g d
g t j d }|dd |d d < t |g d t j d }|d d |dd < t |g d t j d }|dd |dd d< t |g d t j d }|dd d |dd < t |g d t j d }|d dd |dd d< t |g d t j d }|dd d |d dd< t |g d d S )Nr r3 r5 r3 r2 r r )r r r3 r2 r r2 r3 r r r2 r rw rx r r rW rt rW )r2 r r r r )r r3 r r3 r2 )r r r2 r r )r r3 r2 r3 r )r# r% r r r s r) test_overlapping_assignmentz.TestScalarIndexing.test_overlapping_assignment6 s IaLL122#2#Q %%% IaLL#2#!""Q %%%
IaLL2w!!!Q %%% IaLL A & &aaa2gJ$$B$qqq& QIII./// IaLL A & &!!!TTrT'
$$B$"*
QIII.///
IaLL!"""1"Q((( IaLL"1"!""Q((( IaLLQRR5!%R%Q((( IaLL!%R%!""Q((( IaLLUaU8!%R%Q((( IaLLQUU8%1R%Q(((((r+ N) ra rb rc rU r r r r r r r rd r+ r) r r s % % %' ' '
& & &
E E E
3 3 3 9 9 9B B B.) .) .) .) .)r+ r c e Zd ZdZd Zd Zd Zej d ej
e d ej
dd g d
Z d Z
ej ej ej j d k d
d Zd Zej j d Zd Zd Zd Zd Zd Zd Z d Z d Z d Z d Z d Z!d Z"d Z#d Z$d Z%d! Z&ej d"e'd#d$d%g ej d&ej( ej) d' g d( Z*ej d)ej+ ej, ej- ej. ej/ g d* Z0d+S ), TestCreationz'
Test the np.array constructor
c p G d d }t t t j | d S )Nc e Zd ZddZdS )+TestCreation.test_from_attribute..xNc d S r" rd )rT r/ s r) __array__z5TestCreation.test_from_attribute..x.__array__m r+ r" ra rb rc r rd r+ r) r r l s(
r+ r )r r8 r# r r< s r) test_from_attributez TestCreation.test_from_attributek sL j"(AACC00000r+ c t j d t j d z }ddg}t j ddgt }|D ]-}d|z }t t j || || .d S )N
AllIntegerFloat123{ r zString conversion for %sr )r# typecodesr r
r )rT r nstrresultr msgs r) test_from_stringz TestCreation.test_from_stringr s \*R\'-BBu~3*C000 J JD,t3C $d333VS I I I I I J Jr+ c t j g d }|j dk sJ t j ddgd }|j dk sJ t j t
5 t j ddgd d d d n # 1 swxY w Y t j t
5 t j ddgd d d d n # 1 swxY w Y t j ddgd d }|j dk sJ t j t
5 t j ddgd d d d d d S # 1 swxY w Y d S )NVr V8s 1234V4s 12345O)r# r r/ r rB r) r rj s r) test_voidzTestCreation.test_voidz sA hr%%%yD h)555yD ]9
%
% 5 5 Hgx( 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5
]9
%
% 5 5 Hh( 4 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 h)555<M8[s]zf4,f4)r# r" r9 r
zeros_liker r/ )rT cr s r) test_zeros_like_like_zerosz'TestCreation.test_zeros_like_like_zeros s e$ : :ACxxa(((A r}Q / / 3 3 3 q))/ 9 9 9 9 HU$ ' ' '2=++Q///R]1%%+QW555 HU$ ' ' '2=++Q///R]1%%+QW555 HU% ( ( (2=++Q///R]1%%+QW555 HU% ( ( (2=++Q///R]1%%+QW555 HU( + + +2=++Q///R]1%%+QW555 HU( + + +2=++Q///R]1%%+QW555 HU' * * *2=++Q///R]1%%+QW55555r+ c x t ddd D ]'}t j |d }t | (d S )NrW d Ur )r r# rA r )rT r r s r) test_empty_unicodez TestCreation.test_empty_unicode sH q#q!! A#&&&A FFFF r+ c t t j dddz g j t t t j dddz dg j t t t j ddz dg j t t t j ddz gdz j t t t j ddgddgg j t
t t j ddgddgg j t
t t j g dg dg dg j t
d S ) Nr r2 P r r3 r rv )r3 r r6 )r r# r r/ r7 complexrS s r) test_sequence_non_homogeneousz*TestCreation.test_sequence_non_homogeneous s$ RXq!R%j))/888RXq!R%m,,2F;;;RXq"uaj))/888RXq"ugk**0&999RX1vr2h /006@@@RXBxA /006@@@RXyyyiii @AAGQQQQQr+ c T G d d } G d d }t j | g }t |j dk t |j t j t
k t
t t j | g dS )a) Should not segfault.
Class Fail breaks the sequence protocol for new style classes, i.e.,
those derived from object. Class Map is a mapping type indicated by
raising a ValueError. At some point we may raise a warning instead
of an error in the Fail case.
c e Zd Zd Zd ZdS )5TestCreation.test_non_sequence_sequence..Failc dS Nr3 rd rS s r) r1 z=TestCreation.test_non_sequence_sequence..Fail.__len__ qr+ c t r" r8 rT indexs r) r zATestCreation.test_non_sequence_sequence..Fail.__getitem__ s ll"r+ Nra rb rc r1 r rd r+ r) Failrf s2
#
#
#
#
#r+ ro c e Zd Zd Zd ZdS )4TestCreation.test_non_sequence_sequence..Mapc dS rh rd rS s r) r1 z.Map.__len__ ri r+ c t r" )KeyErrorrl s r) r z@TestCreation.test_non_sequence_sequence..Map.__getitem__# s jj r+ Nrn rd r+ r) Maprq s2
!
!
!
!
!r+ ru r N)r# r r rF r/ r7 r r8 )rT ro ru rR s r) test_non_sequence_sequencez'TestCreation.test_non_sequence_sequence s # # # # # # # # ! ! ! ! ! ! ! !
HcceeW 4 28F+++,,,j"(TTVVH55555r+ c G d d }t j | | | g }t |j t j t d S )Nc e Zd Zd Zd ZdS )4TestCreation.test_no_len_object_type..Point2c d S r" rd rS s r) rk z=TestCreation.test_no_len_object_type..Point2.__init__. r r+ c * |dv r|S t )Nr_ r )rT inds r) r z@TestCreation.test_no_len_object_type..Point2.__getitem__1 s &== J$,,&r+ N)ra rb rc rk r rd r+ r) Point2ry - s2
'
'
'
'
'r+ r~ )r# r r r/ r7 )rT r~ r s r) test_no_len_object_typez$TestCreation.test_no_len_object_type+ sv ' ' ' ' ' ' ' '
Hffhh&&((3 4 4QWbhv . ./////r+ c G d d }t j | }t t | d d S )Nc e Zd Zd Zd ZdS )/TestCreation.test_false_len_sequence..Cc t r" r| )rT r s r) r z;TestCreation.test_false_len_sequence..C.__getitem__< s r+ c dS )Nr rd rS s r) r1 z7TestCreation.test_false_len_sequence..C.__len__> s rr+ Nr rd r+ r) r, r ; s2
!
!
!
r+ r, r )r# r r r& rT r, rR s r) test_false_len_sequencez$TestCreation.test_false_len_sequence9 s[
HQQSSMMSVVQ r+ c ( G d d }t j d }t t 5 | |d d < d d d n # 1 swxY w Y t j | t | k d S )Nc e Zd Zd Zd Zd ZdS )/TestCreation.test_false_len_iterable..Cc t r" r r< s r) r z;TestCreation.test_false_len_iterable..C.__getitem__G s r+ c t d S Nrd )iterrS s r) __iter__z8TestCreation.test_false_len_iterable..C.__iter__I s
Bxxr+ c dS )Nr2 rd rS s r) r1 z7TestCreation.test_false_len_iterable..C.__len__K ri r+ N)ra rb rc r r r1 rd r+ r) r, r F sA
r+ r, r2 )r# rA r r8 r listr s r) test_false_len_iterablez$TestCreation.test_false_len_iterableD s
HQKK
:
&
& 133AaaaD
aacc""""s AAAc G d d } |g d }t t t j | d d S )Nc e Zd Zd Zd Zd ZdS )0TestCreation.test_failed_len_sequence..Ac || _ d S r" )_data)rT r4 s r) rk z9TestCreation.test_failed_len_sequence..A.__init__W s
!
r+ c H t | | j | S r" )r r )rT r s r) r z.A.__getitem__Y s !tDzz$*T"2333r+ c * t | j S r" )r& r rS s r) r1 z8TestCreation.test_failed_len_sequence..A.__len__[ s 4:&r+ N)ra rb rc rk r r1 rd r+ r) Ar V sA
"
"
"
4
4
4
'
'
'
'
'r+ r rU r )r r& r# r )rT r r s r) test_failed_len_sequencez%TestCreation.test_failed_len_sequenceT sc ' ' ' ' ' ' ' '
AgggJJS!%%q)))))r+ c B t j d }t j t j j }dD ]d}t j | }|j }t j |d||z f| t t t j |d||z dz f| ed S )Nr] )r6 r r r )r r rF r/ r3 )
r# r9 r; r6 r r/ r@ rE r r8 )rT rI max_bytesr/ r@ s r) test_array_too_bigz TestCreation.test_array_too_bigb s hsmmHRW%%) ) I IEHUOOE ~H Jc4 ' 1 35
B
B
B
B *bjd!*H!4q!8 :%
I
I
I
I
I
I Ir+ c t j t d 5 t j | }d d d n # 1 swxY w Y t j |t
S )Nz.*detected shape wasr@ r )r rB r8 r# r r7 )rT seqrR s r) _ragged_creationz TestCreation._ragged_creationq s
]:-C
D
D
D
A x6****s =AAc | dgddg }t |j d t |j t | ddgdg }t |j d t |j t | dddgg }t |j d t |j t d S Nr3 r2 r rR r r rF r/ r7 r s r) test_ragged_ndim_objectz$TestCreation.test_ragged_ndim_objectx s ! !A31+ . .QWd###QWf%%% ! !1qc1+ . .QWd###QWf%%% ! !1a!+ . .QWd###QWf%%%%%r+ c | ddgdgdgg }t |j d t |j t | dgddgdgg }t |j d t |j t | dgdgddgg }|j dk sJ |j t k sJ d S r r r s r) test_ragged_shape_objectz%TestCreation.test_ragged_shape_object s ! !Aq6A3"4 5 5QWd###QWf%%% ! !A3A"4 5 5QWd###QWf%%% ! !A3aV"4 5 5w$w& r+ c t j d d g }t j g d x|d< |d< t j | j dk sJ t j |g j dk sJ t j d d g }t j g d |d< t j g d |d< t j | j dk sJ t j |g j dk sJ d S )NrU r r3 r r3 r2 r3 r2 r r )r# r rF )rT outer outer_raggeds r) test_array_of_ragged_arrayz'TestCreation.test_array_of_ragged_array s $ && hyyy 1 11a58x$,,,,x &&0000xt -- (999-- Q (<<<00 Qx %%+t3333x ''-888888r+ c t j t d ggg }t j dt d g }t j dgt d gg }d S rh )r# r r r s r) test_deep_nonragged_objectz'TestCreation.test_deep_nonragged_object sY H
|n% & & Ha_ % % HqcGAJJ<( ) )r+ r/ zO,OzO,(3)Oz(2,3)Ofunctionc R t j t j | | S r> )r#
empty_likerA rF r/ s r) r zTestCreation. s rxU/K/K/K!L!L r+ c |d| }t j d }||j t | z z }| |k sJ d S )Nr r )r# r tobytesr r& )rT r r/ r' expecteds r) test_object_initialized_to_Nonez,TestCreation.test_object_initialized_to_None sg hq&&&8D>>))++szS]] :;{{}}((((((r+ ri c t j d } |g d| } |g dt | }t || |j |k sJ d S )Nr rU r )r# r/ r r
)rT ri r/ arr1arr2s r) test_creation_from_dtypemetaz)TestCreation.test_creation_from_dtypemeta sp
tIIIU+++tIIIT%[[1114&&&zU""""""r+ N)1ra rb rc __doc__r r& r, r r r paramEllipsisr4 r< r r# r/ r6 r@ rA rL slowrN rP rS r[ r_ rc rv r r r r r r r r r r r7 rE rA r r rn rp rc rf r rd r+ r) r r g s 1 1 1J J JA A A( [ U
V\(u
-
-
-|v|B8/L/L/L MO O O O
A
A
A [))2a7 F H H< <H H<
1 1 1, [ - - - & & & 6 6 6D R R R6 6 68 0 0 0 # # # * * *
I
I
I + + + & & & ! ! ! 9 9 9* * * [ Wvuh&IJJ [ Z J L L*N O O) )O O KJ) [ V
"*bmR-A
& # # # # #r+ r c
e Zd Zd Zd Zd Zej de j
e j g d Z d Z
ej de j
d e j d g d Zd
Zd Zd Zd
Zd Zd Zd Zd Zd Zd Zd Zej d eddgddgd eddgddgd d g g ej d d d!g d" Zd# Z d$S )%TestStructuredc t j ddg }t j d dddd |d< t |j d |d d d
dd | d }t |d j |d j t |j d j |j d j d S )
N)r rW )rR r r3 r r r rW r2 rR r3 r rY r6 )
r# r9 r% r r
rL transposerW r rF r s r) test_subarray_field_accessz)TestStructured.test_subarray_field_access s HV$9#: ; ; ;2 &&q!Q22# 13s8QsV%5%5aAq%A%ABBB
FFF QsV\1S6<000QSX^QSXXZZ_%:;;;;;r+ c V t j g ddddgddggfg ddd d gd d ggfgd
dt fd g }| }t ||k d
d
g t ||k ddg d|d _ t ||k d
dg t ||k dd
g t d D ]V}|d j |d _ d|d j |< t ||k ddg t ||k d
d
g Wt d D ]f}t d D ]T}| }d|d j ||f< t ||k dd
g t ||k d
dg Ugt j
dgdggdg }t j
g ddg }t ||k g dg dg t ||k g dg dg t j
dgdggdg }t j
g ddg }t ||k g dg dg t ||k g dg dg t j
d d gfgddgfggdg }t j
d d gfd dgfddgfgdg }t ||k g dg dg t ||k g dg dg t j
d d gfgddgfggdgd }t j
d d gfd dgfddgfgdg }t ||k g dg dg t ||k g dg dg t j dd d g }t j dd d g }t j
t 5 ||k d d d n # 1 swxY w Y t j dd d g }t j dd d g }t j
t 5 ||k d d d d S # 1 swxY w Y d S )!NrU rR r3 r2 r r r r r r r )rR )f4r )rZ r r TFrZ rW rQ r r r9 )r r r rH rI )rR r: r )rR r: r )TFFrY r. )rR )r r r i1)rR )r r )rR )r r2 r3 )r# rec fromrecordsr7 rW r r r rR rZ r r9 r rB r) )rT rR r r jr r/ s r) test_subarray_comparisonz'TestStructured.test_subarray_comparison s
F ii1v1v .
/)))SAq6Aq6BR1S T#c6]4IJ
L
L
FFHHQ!VdD\***Q!VeU^,,,!Q!VdE]+++Q!VeT]+++q / /AqTVAaDFAaDF1I a% 0 0 0 a$ . . . .q 4 4A1XX
4
4FFHH !!q!t Q!VeT]333 Q!VdE]3333
4
Htftf%k] ; ; ; H''' } = = =Q!V 1 1 13G3G3G HIIIQ!V 1 1 13G3G3G HIII Htftf%.?-@ A A A H'''0A/B C C CQ!V 1 1 13G3G3G HIIIQ!V 1 1 13G3G3G HIII HAykaVI;/8I7J K K K H1vi1a&aVI6?P>Q R R RQ!V 2 2 24H4H4H IJJJQ!V 2 2 24H4H4H IJJJ
HAykaVI;/8I7JRU V V V H1vi1a&aVI6?P>Q R R RQ!V 2 2 24H4H4H IJJJQ!V 2 2 24H4H4H IJJJ
HT"7!E F F F HT"5{!C D D D
]9
%
%
FF
HT"7!E F F F HT"5{!C D D D
]9
%
%
FF s$ #N66N:=N:
P P"%P"c 6 t j ddg }t || t j ddg }t || t j ddg }t || t j ddg }t || d S )Nr )rR TestStructured.test_structured_array_comparison_bad_broadcasts s HQe $ $ $ HRu % % %
]:
&
& Bq!HHH s
A A$'A$c F t j ddgddg }t j ddgddg }t ||k d d
g t ||k d
d g t j ddgd dg }t j ddgdd g }t ||k d d
g t ||k d
d g t j ddgd
dg }t j ddgddg }t ||k d d
g t ||k d
d g d S )NrW r rQ r3 rR >i8r r r )rW + rR zf8FT)rR r r r )rR z10>f8)r z5i8r# r r r s r) *test_structured_comparisons_with_promotionz9TestStructured.test_structured_comparisons_with_promotion sP
Hgw' l/K L L L Hgw' l/K L L LQ!VeT]+++Q!VdE]+++ Hgw' l/K L L L Hgw' l/K L L LQ!VeT]+++Q!VdE]+++
Hgw'
/N O O O Hgw'
/N O O OQ!VeT]+++Q!VdE]+++++r+ c , t j || S r" )r> eqr r/ s r) r zTestStructured.1 hk!Q&7&7 r+ c , t j || S r" )r> ner s r) r zTestStructured.2 r r+ c t j ddg }t j d }t j t 5 ||| d d d n # 1 swxY w Y t j ddg }t j |j |j sJ t j t 5 ||| d d d n # 1 swxY w Y t j dd }t j dd }t j t 5 ||| d d d d S # 1 swxY w Y d S )Nr )rR r r ))titlerR r V7r) )r# r9 r rB r) can_castr/ )rT r r r/ s r) test_void_comparison_failuresz,TestStructured.test_void_comparison_failures0 s
HQ{m , , , HQKK
]9
%
% Bq!HHH
HQ 5 6 7 7 7{17AG,,,,,
]9
%
% Bq!HHH
HQd # # # HQd # # #
]9
%
% Bq!HHH s6
A A"%A":
CCC#
D==EEc t j dgdg }t t j |j dgd | dg }t
|| t
|d d |d d t j d d
gdd g }t j d d
gdd g }t t j |j |j d
t
||k ddg t t j |j |j d
| |j d
}t
||k ddg dd g}t t j |j |d | |d }t
|t j d d
g| k ddg ddg}t t j |j |d | |d }t
|t j d d
g| k ddg ddg}t t j |j |d t t |j |d dd g}t t j |j |d
t t |j |d
ddg}t t j |j |d t t |j |d t t j |j |j d t t |j |j d dD ]]}dg}t t j |j || g d}t t j |j || ^d S )Nr r r r unsafecastingrR r r r r r equivTr safe)r z>f4 same_kindr )r i2r no)r r r r )r r )rZ r ) r# r r r r/ r r byteswap newbyteorderr r) )rT rR r rZ r. r s r) test_castingzTestStructured.test_castingH s
HdVL> 2 2 2 AGl^XFFFGGG
HHl^ $ $Q
1133444QsVAY#q ***
Hgw' l/K L L L Hgw' l/K L L L AGQWg>>>???Q!VdD\*** AGQWg>>>???
HHQWgH . .Q!VdD\*** < ( AGQ777888
HHQH ' 'a28Wg$6a@@@@D\ # # #
< ( AGQ <<<===
HHQ H , ,a28Wg$6a@@@@D\ # # #
< (BKF;;;;<<<i1f====
< (BKG<<<<===i1g>>>>
< (BKK@@@@AAAi1kBBBBBK$????@@@i17DAAAA < B BG A AGQ@@@@ A A A999A AGQ@@@@ A A A A B Br+ c t j g g d }|ddg t j dddg }|dd g d S )
N)rR r r r )rZ r+ r rR r r )r r )rt z|Ort r )r# r r9 )rT rR dat2s r) test_objviewzTestStructured.test_objview sW HR C C C D D D 3*
xJ 455 c3Zr+ c f t j dg }t j dd|fg }t j d| }t j dd |d d < t j d| |d d< t |d d t j dd d S )
N)elemr rW )fieldr rQ structr3 rQ r r r r )r# r/ r9 rJ r )rT struct_dtrK r s r)
test_setfieldz TestStructured.test_setfield s H / 122
X*Hi+@A
B
B HQOO$///!W
)444!XQqT']BGBd$;$;$;<<<<.testassign s ! AaDIIIr+ )r# r9 r% r r r8 )rT r r rZ s @r) test_setfield_objectz#TestStructured.test_setfield_object s
HQzl + + +IaLL!S QqT#Y ! ---
HQ}o . . . % % % % % j*-----r+ c t j dt fdg }t j d| }t |d g d t |d j d g d|d< t |d g d t |d g d d
|d |d dk < t |d g d t |d g d d |d< t |d g d t |d g d t j d|d j }t |j d t |d | t t j dd d
g d j d t t j dd j d t t j dd d j d |d d }t |j d t |ddgddgg t |d d j |j t t t | t t j
| t j }t j || | d t j | }t |j d t || t' d 5 }t j || t j | }t |j d t || d d d d S # 1 swxY w Y d S )NI)SS0r r r )r+ r+ r+ r+ r )rR r rZ r r r r r hellor )rR S0,S0)r ry rR r r zS0,u1f0r3 r+ r z.npy)suffix)r# r/ r
r9 r r@ rE rA r r
evalr dictr ioBytesIOsaveseekloadr )rT rK r r/ xxr yytmps r) test_zero_width_stringz%TestStructured.test_zero_width_string s XSz;/
0
0 HQb ! ! !QsV 1 1 1222QsV_a(((%%%#QsV 1 1 1222QsV\\\*** &#qv{QsV 1 1 1222QsV\\\*** #QsV 1 1 1222QsV\\\***
Jq# - - -QZ###QsVQ RXa(3(5 6 6 669;;CQ H H HRXaw///8!<<<RXaw///5>BBB
sV^^F
#
#R[!$$$R3*sCj1222RU["(+++4R$RX*>*>*>??DDD JLL
2 q
WQZZR[!$$$R
V
$
$
$ ! GC B a ( ( ( R ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !s AM::M>M>c n t j dd }|d }t |j |u d S )Nr zi4,f4r r )r# r9 r r r s r) test_base_attrz TestStructured.test_base_attr s9 HQg & & &
aD! r+ c f
d
t j ddg }t j d| t j dgddg }t j dgddg }t j dgddg }t j dgdg t j dgd d
g } d gd gd
gd }t j ddg| }t
| | t
| | t
| | t
t
fd t
| | d|d d < t t j ddg| t j dgg d }|d dg |dd g< t |d d t j dgddg }|ddg |ddg< t |d d d S ) Nc F | }||d< ||dd < |S r rV )r' r rZ s r) r z2TestStructured.test_assignment..testassign s*
AAaDAabbEHr+ fooi8barr( r2 r r )bazr( r )r' r: )r* r: r* r( r namesformatsoffsetsc S r" rd )r' r v4s r) r z0TestStructured.test_assignment.. s C)<)< r+ r )r3 r rU )r& r) )r+ r r' r r2 r3 r r r r r rR r ) r# r/ rJ r rm r r r) r ) rT rK v1v2v3v5r ansrR r' r r1 s @@@r) test_assignmentz TestStructured.test_assignment se X}m4
5
5gann
Xugm]%C
D
D
D
Xugm]%C
D
D
D
Xugm]%C
D
D
D
Xtfm_
5
5
5
Xugm]%C
D
D
D HHwD6qcJJ K Khe}B///ZZR((#...ZZR((#...ZZR((#...i! ? ? ?3*
3*
QqTYY[[%(((((r+ c t t 5 t j d dd }| d d d d d S # 1 swxY w Y d S )N rW r )r r8 r# r% r itemsetrj s r) test_scalar_assignmentz%TestStructured.test_scalar_assignment s
:
&
& )B--''1--C KKNNN s >A A$'A$c t j dd }t |ddg d |d ddg t |d |d d d S )Nr3 z (2)3S,(2)3Ur r r8 r rd )r# rA r r< s r) test_structuredscalar_indexingz-TestStructured.test_structuredscalar_indexing sb H1M 2 2 2QT{^A&!d4[(9:::QqT1Q48$$$$$r+ c t j dg d t t fd t t fd t t fd ddg d S ) Nr ))rz r rZ r r r r c ddg S )NrR rZ rd r s r) r z7TestStructured.test_multiindex_titles.. C9 r+ c ddg S )NrR rd r s r) r z7TestStructured.test_multiindex_titles.. rC r+ c ddg S )Nr rd r s r) r z7TestStructured.test_multiindex_titles.. s 1c#Y< r+ r rZ )r# r9 r rt r8 r s @r) test_multiindex_titlesz%TestStructured.test_multiindex_titles s} HQ I I I J J Jh 5 5 5 5666h 5 5 5 5666j"7"7"7"7888 3s) r+ c d}d}t j dg||g }t j dg||g }t t t j || t t t j |j |j t t t j |j |j t t j |j |ddg j t |j
d t t j |j |j d t | |j j |j t t j d d
d g d t t j d d d
g d t t j d d d
g d t t j d d d
gd d ||d d < t j dg }t j dg }t t j || t j d
g t t j || t j d
g t t t j |t j dg t t j dd t j d
d g t j d|fg }t j d|fg }t t j || t j dt j d
g fg t j dgdgdgd } t j
d | }
t t j |
|
g j t j d g d S ) Nr r r r rR r NTzf8,i8)r r: )r8 r( )r8 r: )r r( Fr r )r r )r r( )r V3r7 zi8,f4r r r r r r, r )r r )r# r r r) concatenate result_typer/
promote_typesr r r r r rJ ) rT r rt abbadt1dt2dt1nestdt2nestrK rR s r) )test_structured_cast_promotion_fieldorderz8TestStructured.test_structured_cast_promotion_fieldorder sk
Xvhq!f
-
-
-
Xvhq!f
-
-
-i R888i 28DDDi!128RXFFF R%bhC:0DEE"28#8#8#=#=>> @ @ @
R[2844d;;;RYYrx((.999R[< *FGGNNNR[< *FGGNNNR[< *FGGOOOR[< *F)13 3 348 : : : 111 h
|$$h
|$$R%c3//<.1I1IJJJR%c3//<.1I1IJJJi!13* 8N8NOOOR%gw77X|\ :;; = = = (RI;''(RI;''R%gw77Xbh ~&>&> ? @AA C C C X4&aSII
J
J GAR R^QF++128[M3J3JKKKKKr+
dtype_dictrR r r r r] )r- r. r@ r r| r, rG TFc ^ t j || }| dd | dd t j || }t j || }|j |j k sJ |j |j k sJ t j || }||u sJ d S )NrG r@ r/ )r# r/ poprL r@ r )rT rT rG r/ r ress r) test_structured_promotion_packsz.TestStructured.test_structured_promotion_packsG s 5111z4(((y$'''8Je444 ue,,|x 00000zX_,,,, x22hr+ c | t j dgd }|d }|j j rJ t j | j |u sJ t j ||j j J t j t j | }|j j sJ t j | j J d S )Nr3 r2 r r )
r# r rY r rn r r/ r r r )rT r' r0 s r) test_structured_asarray_is_viewz.TestStructured.test_structured_asarray_is_view] s hs%(((Q<''''z&!!&&0000z& 555:BBB fl6 2 233|####z&!!&.....r+ N) ra rb rc r r r r r r r> r r r r r r r r r r! r# r8 r< r> rF rS r rY r[ rd r+ r) r r s0 < < < 8 8 8t
[ THK#=>> ?> , , ,( [ T K77 K77$9 : : : :*8B 8B 8Bt = = =. . . 6! 6! 6!p
#) #) #)J
% % % -L -L -L^ [ \ DSzD#; E E E DSzD#;R
"
"
",# $ $ [ WtUm44 54 $ $
"/ / / / /r+ r c e Zd Zd Zd Zd Zd Zej j d Z
d Z d Z d Z
ej d
d Zej d
d Zd
S )TestBoolc t j d }t j d }t ||u t j d }t j d }t ||u t t j dg d |u t t j d d |u d S )Nr Fr3 Trd )r# bool_r r )rT a0b0a1b1s r) test_test_interningz TestBool.test_test_interningn s
Xa[[
Xe__b
Xa[[
Xd^^b$ #r)***r"b()))))r+ c t j dt }t | |j t |d d d |d d d j t |d d d |d d d j t j dt }t | |j t |d d d |d d d j t |d d d |d d d j d S )Ne r r2 r s )r# rJ rK r rH rH r} rg s r) test_sumzTestBool.test_sumx s GCt $ $ $QUUWWaf%%%QsssVZZ\\1SSqS6;///QtttW[[]]AdddGL111 M+4 8 8 8QUUWWaf%%%QsssVZZ\\1SSqS6;///QtttW[[]]AdddGL11111r+ c R d t | D }t d|z D ]fd|D }t j |t }t j | }t
t j | | | t j }|dz }t
t j | | |dz }t
t j | | d||dk < t
t j | | d S ) Nc g | ]}d |z S r rd .0r s r)
z0TestBool.check_count_nonzero.. s 000Q!q&000r+ r2 c " g | ] }|z d k S r rd )rk r r s r) rl z0TestBool.check_count_nonzero.. s! ...!!a%A ...r+ r r r r )
r r# r rK builtinsrH r rG rm rB ) rT powerlengthpowersr rR rZ avr s @r) check_count_nonzeroz TestBool.check_count_nonzero s 00%--000q%x 1 1A....v...A$'''A QA )!,,a 0 0 0!!B !GB )!,,a 0 0 0 !GB )!,,a 0 0 0 BrQwK )!,,a 0 0 0 0 1 1r+ c 2 | dd d S )Nr| rt rS s r) test_count_nonzerozTestBool.test_count_nonzero s
R(((((r+ c 2 | dd d S )Nrv rw rS s r) test_count_nonzero_allz TestBool.test_count_nonzero_all s
R(((((r+ c t d D ]}t j dt |dz d }d|d |< t t j | t
j | t j dt |dz d }d|d |< t t j | t
j | d S )Nr ) r r3 TF)
r r# r9 rK r rG ro rH tolistrJ )rT orR s r) test_count_nonzero_unalignedz%TestBool.test_count_nonzero_unaligned s q H HAd+++AaCDD1AAbqbE )!,,hl188::.F.F G G GT***1Q3440AAbqbE )!,,hl188::.F.F G G G G
H Hr+ c 6 t d D ]}t j d||f }t t | d t t |d d t | t d t
t | t t j t
|d t t j u t dd D ]}dD ]}t j |||f }t t | d t t |d d t | t d t
t | t t j t
|d t t j
u d S ) Nr r+ Frd r3 r ) a 0 T) r r# r r rK r r r rE False_True_)rT r/ nr vals r) _test_cast_from_flexiblez!TestBool._test_cast_from_flexible s q 5 5Auaj))A a% ( ( ( aee , , , $ / / / Jqxx~~rz:: ; ; ; AbELL&&")3 4 4 4 4 q! 8 8A)
8
8HS5!*-- T!WWd+++ T!B%[[$/// QXXd^^T222
188D>>2:>>???" T**bh67777
8 8 8r+ c D | t j d S r" )r r# voidrS s r) test_cast_from_voidz TestBool.test_cast_from_void s %%bg.....r+ z See gh-9847r c D | t j d S r" )r r# str_rS s r) test_cast_from_unicodez TestBool.test_cast_from_unicode s %%bg.....r+ c D | t j d S r" )r r# bytes_rS s r) test_cast_from_bytesz TestBool.test_cast_from_bytes s %%bi00000r+ N)ra rb rc rd rg rt rx r r r rz r r r xfailr r rd r+ r) r] r] m s * * * 2 2 2
1
1
1 ) ) )
[) ) )
H H H8 8 8(/ / / [m,,/ / -,/ [m,,1 1 -,1 1 1r+ r] c e Zd Zeefd Zd Zd Zd Zd Z d Z
d Z d Z d Z
d
Zd Zd Zej ej d
k d d ZdS )TestZeroSizeFlexiblec t j | }|t j k rt j | |df S t j d|dfg }t j | | d S )Nr r r )r# r/ r r9 r s r) _zeroszTestZeroSizeFlexible._zeros si BG 8E%444 4 3q/*++xU+++C00r+ c 0 | dt }t |j d | dt j }t |j d | dt }t |j d d S )NrQ r )r r r r@ r# r r rT zss r) test_createz TestZeroSizeFlexible.test_create sy
[[U
#
#R[!$$$
[[RW
%
%R[!$$$
[[S
!
!R[!$$$$$r+ c t t j t fD ]W}| d| }t || }t t | }|D ]} |dd|i| ||fd|i| Xd S )NrQ kindrd )r r# r r r getattr) rT namekindskwargsrK r sort_method sort_funcr s r) _test_sort_partitionz)TestZeroSizeFlexible._test_sort_partition s "'3' 3 3BR$$B!"d++K D))I
3
3 000000 "224262222
3 3 3r+ c 4 | dd d S )Nr$ qhsr r rS s r) test_sortz TestZeroSizeFlexible.test_sort s! !!&!66666r+ c 4 | dd d S )Nargsortr r r rS s r) test_argsortz!TestZeroSizeFlexible.test_argsort s! !!)5!99999r+ c 8 | ddgd d S )N partition introselectr2 r kthr rS s r) test_partitionz#TestZeroSizeFlexible.test_partition s% !!+m_!!LLLLLr+ c 8 | ddgd d S )N argpartitionr r2 r r rS s r) test_argpartitionz&TestZeroSizeFlexible.test_argpartition s% !!. Q!OOOOOr+ c t t j t fD ]B}| d| }| d | d Cd S )NrQ r: r )r r# r r r resizerT rK r s r) test_resizez TestZeroSizeFlexible.test_resize s[ "'3' BR$$B IIbMMM IIh r+ c 2 t t j t fD ]|}| d| }t | | j t j | t | |df j d }d S )NrQ r3 r ) r r# r r r r rm r/ rF r s r) test_viewz TestZeroSizeFlexible.test_view s "'3' 7 7BR$$B
*BHRLL 9 9 9
"a))/ 6 6 6 6 7 7r+ c | dt }t |t j | d S rP )r r
r r r r r s r)
test_dumpsz TestZeroSizeFlexible.test_dumps s=
[[S
!
!R bhhjj1122222r+ c 2 t dt j dz D ]x}t t j t fD ]\}| d| }t j || }t j | }t |j |j ]yd S )Nr2 r3 rQ protocol) r r r r r# r r r r r r r/ )rT protorK r pzs2s r) test_picklez TestZeroSizeFlexible.test_pickle s 1f 5 9:: 2 2E bgs+
2
2[[R((Le444 l1oo RXsy1111
2 2 2r+ c t j g dd }t j | }t j | }|j dk sJ dS )z]Checking if an empty array pickled and un-pickled will not cause a
segmentation faulti?B r N)r# r r r r r rH )rT r' pk_dmppk_loads r) test_pickle_emptyz&TestZeroSizeFlexible.test_pickle_empty sX hrll""61-- c"",v&&|q r+ rW requires pickle protocol 5r c t j d }g }t j ||j d }t j || }d|d< |d dk sJ |d d S )NrQ rW )buffer_callbackr buffersr5 r )r# r% r r appendr )rT r r bytes_stringarray_from_buffers r) test_pickle_with_buffercallbackz4TestZeroSizeFlexible.test_pickle_with_buffercallback s "
|E7>-.0 0 0 "L w G G G
a #r)))+d7 Z?d8 Z@d9 ZAd: ZBd; ZCd< ZDe j
dejE d= d> ZFe j
dejE d= d? ZGd@ ZHdA ZIe j
dBejE dC dD ZJe j
dBejE dC dE ZKdF ZLdG ZMdH ZNdI ZOdJ ZPdK ZQe j
dBejE dC dL ZRdM ZSe j
dNejT ejU f dO ZVe j
dNejT ejU f e j
ddP dQ ZWdR ZXdS ZYdT ZZdU Z[dV Z\dW Z]dX Z^dY Z_dZ Z`d[ Zad\ Zbd] Zcd^ Zdd_ Zed` Zfda Zgdb Zhdc Zidd Zjde ZkdS )f TestMethods) quicksortheapsortstablec
t j g dg dg dg }t j g dg dg dg }t j dgdgdgg }dD ]}t | || t j ||d d df d d f | t t j ||| ||d d df d d f | t | |
d t t j ||
d t | d
d t t j |d
d d S ) NTFTFFFTTTFTr Naxiswherer r r )r# r r all)rT rR wh_fullwh_lower_axs r) test_all_wherezTestMethods.test_all_where+ s H)))+++(((*
+
+ ( / / / 1 1 1 / / / 1 2 2 8eW #W "V % & & ; ;C Cx88q!!!A# qqq 1<<<
>
>
> 8<<<8AAAaC=?+//S/99
;
;
;
; QUUU))4000RVAW---t444QUUU''...RVAU+++T22222r+ c
t j g dg dg dg }t j g dg dg dg }t j dgdgdgg }dD ]}t | || t j ||d d d f d d f |
t t j ||| ||d d d f d d f |
t | | d t t j || d t | d d t t j |d d d S ) Nr r r )FTFFTr r r r r )r# r r rI )rT rR r wh_middler s r) test_any_wherezTestMethods.test_any_where@ s H)))+++(((*
+
+ ( 0 0 0 . . . 1 1 1 3 4 4 Hug #f $g ' ( ( < >>
@
@
@ 9===9QQQqS>!!!+,00c0::
<
<
<
<QUUU))5111RVAW---u555QUUU''///RVAU+++U33333r+ c ` g dg}t j d dd }| ddgd }t || ddgd d
gg}| g d d }t || dgd gg}t j d dd }| ddgd }t || t j d dd }| ddg }t |d d S N)rW r r r r rQ r2 rW r r3 r r r r )r r3 r r3 r r# r% r compressr rT rM r' outs r)
test_compresszTestMethods.test_compressT s* imm##Aq))llAq6l**S# 1v1v ll???l33S# sQCjimm##Aq))llAq6l**S# imm##Aq))llAq6""S! r+ c N dt j dt z }dt j dt z }dt j dt z }dt j dt z }t j g d }| ||f }t |g d | ||f }t |g dg dg | ||f }t |g dg dg t j d }t
t |j t j d |g t j d }t j t j d g d
| } || u sJ t |d d
t j d }t j g d|d d |d d |d d g|dd d }t |t j g d d S )Nr2 rR r r r r r r3 r2 r2 r r3 r )rQ r r r rd r rW ru r wrapr modert )
r# rJ r
r chooser r r) r r% )
rT r r/ x2y2r} r onedr rets
r) test_choosezTestMethods.test_chooseg s
bgd#&&& &
bgd#&&& &
rwvS)))
)
rwvS)))
)hyyy!! JJ1v Q """ JJBx QIII./// JJ2w QIII.///wqzziRWQZZ$@@@hqkki \\\s;;;czzzzSWb!!!
IaLL Iggg"1"q!uae 4!AaC&v N N NQ++,,,,,r+ c g d}g dg dg dg}t j t j t j t j t j t j t j t j fD ]}t j || }t j || }|dv r7t t |j t t |j d gt | d d t | d t j g d
| t | d t j g d | d S )
N)r3 r2 rQ r? r rW r r )rW r r r )rQ r r rW )1r r3 r r i g )r $ T r5 ) ib iX )r# r r r r float32r complex64
complex128r r ArithmeticErrorprodr r
)rT rN ba2ctyperR a2s r) test_prodzTestMethods.test_prod sS
$
$
$||\\\===9h 28RYj"*blBMK E EEU##A#u%%B
"" oqv666 orwQ????? QVVV^^U333"2777??#%8,=,=,=u#E#EG G G"2777#3#3#%8OOOU#C#CE E E E E Er+ r/ Nc t j g d| }| d }| g d }t |g d | d }t |g d | ddgd
}t |g d g d g d g | g d
d
}t |g dg dg | dd
}t |g d g d g d g d g | dd
}t |g dg dg d S )Nr3 r2 r r rW r r r r3 r r2 r3 r3 r2
r3 r2 r2 r2 r r r rW r r r2 r3 r3 r2 r2 r r r r rW rW r r r3 r r rU r rW r r3 r r2 r3 r2 r2 r2 r r r rW rW rW r r r3 r3 r2 r2 r r r r rW rW r r )r# r r repeatr )rT r/ mm_rectr s r) test_repeatzTestMethods.test_repeat s H'''u 5 5 56""
HH''' ( (Q ( ( ( ) ) )
HHQKKQ + + + , , ,
MM1a&qM ) )Q""$ % % %
MM)))!M , ,Q++++++- . . .
MM!!M $ $Q"""$ % % %
MM!!M $ $Q++++++- . . . . .r+ c t j g dg dg dg dg }g dg dg}t | dd | g d g d
g d g}t | d d
| g dg dg dg}t | dd | g dg dg dg}t |j dd | d S )NrU r )r r r )rQ r? r| r
)r r r rQ r? r| r2 r r rW r r r )r rQ r? r| r r )r3 rQ r r )r r2 r? r )r rW r r| r r rY r6 )r3 r r rQ )r2 rW r r? )r r r r| r, )r# r r r rL )rT r' rM s r) test_reshapezTestMethods.test_reshape s
h 999iii FGG!!!#8#8#89S[[A&&,,,||\\\???;S[[A&&,,, }}mmm]]];S[[s[33S999 }}mmm]]];SU]]6]55s;;;;;r+ c d } |t j ddg ddg |t j d d |t j ddg dd gd
|t j d d g ddgd |t j d
g dg |t j dg dgd
d S )Nc t | j | | t j | } | j |d|i}t || ||u sJ d S Nr )r roundr# rY )r' r
round_argsr rX s r) check_roundz+TestMethods.test_round..check_round sa J/ : : :-$$C#)Z1S11C h ' ' '#::::::r+ g333333? ?r3 r2 gffffff(@g /@rQ r r5 gL(@gQ /@y @ ?y @ @y )@ /@y $@ 4@rn )rT r$ s r)
test_roundzTestMethods.test_round s BHc3Z((1a&111 BHSMM1%%% BHdD\**RHb999 BHeU^,,tTlA>>> BHj\**VH555 BHl^,,xj"=====r+ c ^ t j dgdgdggg }t | g d t | d dgdgdgg t t
|j d t | d g dg d S ) Nr3 r2 r rU r r r r )r# r r squeezer r8 r s r) test_squeezezTestMethods.test_squeeze s HsQC!o& ' 'QYY[[))),,,QYYDY))QC!qc?;;;j!)$7777QYYDY))III;77777r+ c t j ddgddgg t ddgddgg t t
fd t t
fd t t
fd d S )Nr3 r2 r r c . d S r r r s r) r z,TestMethods.test_transpose.. s !++a.. r+ c 0 dd S r r, r s r) r z,TestMethods.test_transpose.. s !++a*;*; r+ c 2 ddd S )Nr r3 r2 r, r s r) r z,TestMethods.test_transpose.. s !++aA*>*> r+ )r# r r r r r8 r s @r) test_transposezTestMethods.test_transpose s Hq!fq!f% & &Q[[]]aVaV$4555j"8"8"8"8999j";";";";<<<j">">">">?????r+ c $ d}t j t j ddg }t j | }t ||d d d | d}t j dt j }|xj t j t j t j ddddddg z
c_ |xj t j ddt j t j ddddg z
c_ t j | }t ||d d d | d S )Nz Test real sort order with nansr3 r r5 z!Test complex sort order with nansr r ) r# r nanr$ r r9 r r r )rT r% rR r s r) r zTestMethods.test_sort s / Hbfa^ $ $ GAJJQ$$B$%%%1 HQbm , , , 2626261aAq!<< 261aAq!<< GAJJQ$$B$%%%%%r+ c v t j d| }|d d d }| j D ]}}d|z }| }| | t ||| | }| | t ||| ~d S )Nrf r r5 scalar sort, kind=%sr r# r% rW
sort_kindsr$ r rT r/ rR r r r% rZ s r) test_sort_unsignedz TestMethods.test_sort_unsigned s
Ic ' ' '
dddGLLNNO $ $D(4/CA
FFF As # # #A
FFF As # # # # $ $r+ c x t j dd| }|d d d }| j D ]}}d|z }| }| | t ||| | }| | t ||| ~d S )N3 r r5 r3 r4 r5 r7 s r) test_sort_signedz TestMethods.test_sort_signed
s
Ic2U + + +
dddGLLNNO $ $D(D1CA
FFF As # # #A
FFF As # # # # $ $r+ partr r c t j t j t j t j t j t j i| }t j dd| }|d d d }|dz | }|dz | }t ||d t ||d | j D ]}d|d|} | }
|
| t |
|| | }
|
| t |
|| d S )
Nr: r; r r5 ? ?r3 zcomplex sort, z part == 1, kind=r4 )r# singlecsingledoublecdoubler clongdoubler% rW r r r6 r$ r ) rT r= r/ cdtyperR r aibir r% rZ s r) test_sort_complexz TestMethods.test_sort_complex sG
Irz Irz M2>
Ic2U + + +
dddGLLNN4j
(
(4j
(
(D! D! O % %DD;?44FC A
FFF B $ $ $ A
FFF B $ $ $ $ % %r+ c dD ]}t j d D ]k}t j g d||z }| }| d | }t
||| ld S )Nr Complexy ? @y @ @y @ ?r z$byte-swapped complex sort, dtype={0})r# r" r rW r$ formatr )rT
endiannessrK r' rZ r% s r) test_sort_complex_byte_swappingz+TestMethods.test_sort_complex_byte_swapping5 s * *Jl9-
*
*h 4 4 4JOLLLHHJJ<CCBGG QS))))
* * *r+ c t j d t d D | }|d d d }| j D ]}}d|z }| }| | t
||| | }| | t
||| ~d S )Nc 2 g | ]}d t | z S )aaaaaaaachrrj s r) rl z0TestMethods.test_sort_string..B s# >>>aj3q66)>>>r+ rf r r5 kind=%sr4 )r# r r rW r6 r$ r r7 s r) test_sort_stringz TestMethods.test_sort_string? s
H>>5::>>>e L L L
dddGLLNNO $ $Dd"CA
FFF As # # #A
FFF As # # # # $ $r+ c t j dt }t t d |d d < |d d d }dD ]}}d|z }| }| | t ||| | }| | t ||| ~d S )Nrf r rf r5 qhr rT r4 )r# rA r7 r r rW r$ r )rT rR r r r% rZ s r) test_sort_objectz TestMethods.test_sort_objectM s HV6 * * *E#JJ !!!
dddG# $ $Dd"CA
FFF As # # #A
FFF As # # # # $ $r+ rK r r stepr3 r2 c t j d t d|z D | }|d d d }dD ]&}d|z }| d d | }| | }| | t
||d d | | t
|d d | | |d d | | | d d | }| | }| | t
|||dz
d | | t
|d d | | ||dz
d | | (d S ) Nc g | ]}||fS rd rd rj s r) rl z4TestMethods.test_sort_structured..a s 666q!f666r+ rf r r5 rX rT r4 r3 )r# r r rW r r$ r ) rT rK r\ rR r r r% rZ indxs r) test_sort_structuredz TestMethods.test_sort_structured[ s|
H66eCHoo666b A A A
dddG# @ @Dd"C4 A99$9''D
FFF AfffIs + + + 66T64 !FFdF)S 9 9 94 A99$9''D
FFF Ad1fldlOS 1 1 1 66T64 !DFLDL/3 ? ? ? ? @ @r+
datetime64[D]timedelta64[D]c J t j dd| }|d d d }dD ]}}d|z }| }| | t ||| | }| | t ||| ~d S )Nr rf r r5 rX rT r4 )r# r% rW r$ r r7 s r) test_sort_timezTestMethods.test_sort_timep s
IaE * * *
dddG# $ $Dd"CA
FFF As # # #A
FFF As # # # # $ $r+ c t j ddgddgg }t j ddgddgg }t j ddgddgg }| }| d t ||d | }| d t ||d | }| t ||d d S ) Nr r2 r3 r r ztest sort with axis=0ztest sort with axis=1ztest sort with default axis)r# r rW r$ r )rT rR r rZ r s r) test_sort_axiszTestMethods.test_sort_axis~ s
Hq!fq!f% & & Hq!fq!f% & & Hq!fq!f% & &
FFHH AQ2333
FFHH AQ2333
FFHH Q899999r+ c ^ t j g }d|_ t |j |j D ]<}d | }t
t j || || =d}t
t j |d | | d S )Nr z#test empty array sort with axis={0}r z$test empty array sort with axis=None) r# r rF r r rL r r$ ravel)rT rR r r% s r) test_sort_size_0z TestMethods.test_sort_size_0 s HRLL 16'16** 8 8D7>>tDDC ...3 7 7 7 74RWQT***AGGIIs;;;;;r+ c G d d }t j | gdz t }| j D ]B}d|z }| }| | t
||| Cd S )Nc e Zd Zd ZdS )0TestMethods.test_sort_bad_ordering..Boomc dS )NTrd rT others r) __lt__z7TestMethods.test_sort_bad_ordering..Boom.__lt__ s tr+ N)ra rb rc rp rd r+ r) Boomrl #
r+ rq r] r rT r4 )r# r r7 r6 rW r$ r )rT rq rR r r% rZ s r) test_sort_bad_orderingz"TestMethods.test_sort_bad_ordering s
HddffX^6 2 2 2O $ $Dd"CA
FFF As # # # # $ $r+ c t d D ]`}t j ddt j }| d }|d d d at j dg }t d D ]`}t j ddt j }| | }|d d d ad S )Nr i rH r/ r* r5 )r r r )r r# randomrandintrB rm r$ r/ )rT r randr' rK s r) test_void_sortzTestMethods.test_void_sort s q A9$$St28$DDD))D//C "INN
X*+
,
,q A9$$St28$DDD))B--C "INN r+ c t j dt j dgt }| j D ] }t
t |j | G d d t j fdt d D
d }t j | | j D ] }t
t |j | d S )
Nr r3 r r4 c , e Zd Zd ZexZxZxZxZxZZ dS ),TestMethods.test_sort_raises..Raiserc t d )NzSOMETHING ERROREDr) argsr s r) raises_anythingz.Raiser.raises_anything s 3444r+ N)
ra rb rc r __eq____ne__rp __gt____ge____le__rd r+ r) Raiserr} s;
5
5
5BQ QF QV Qf Qv Qr+ r c ( g | ]} |gS rd rd )rk r r s r) rl z0TestMethods.test_sort_raises.. s# 999!1
999r+ rQ r5 )
r# r r nowr7 r6 r r) r$ r r rw shuffle)rT r' r r s @r) test_sort_raisesz TestMethods.test_sort_raises s h8<>>1-V<<<O : :D )SXD 9 9 9 9 9 R R R R R R R R h9999uRyy999::BB2FF
# O : :D )SXD 9 9 9 9 9 : :r+ c p t j d }| }|}|j dk r6|j dz }|d || c||< |d< |d d }|j dk 6t t j | | t |t j | | d S )N@B r r2 r )r# r% rW rH r r$ r )rT r dor mids r) test_sort_degradedz TestMethods.test_sort_degraded s Ig
VVXX
fqjj&A+C bE1S6MAcFAbE#2#A fqjj
RWQZZ$$$Qrz!}}%r*****r+ c F d }d }t j dd } || || d || d || d t j dd } || || d || d || d d S )Nc t | j j t | j j t | j j d S r" r rY fortranre rd r' s r) assert_fortranz-TestMethods.test_copy..assert_fortran sD CI% & & & CI* + + + .. / / / / /r+ c t | j j t | j j t | j j d S r" r r s r) assert_cz'TestMethods.test_copy..assert_c sG )) * * * .. / / / CI* + + + + +r+ r3 rY r6 r, r )r# rA rW )rT r r rR s r) test_copyzTestMethods.test_copy s 0 0 0
, , ,
HV3 ' ' ' qvvc{{###qvvc{{###
HV3 ' ' ' qvvc{{### r+ r+ zi,Oc 4 t j d| }t j |j j d|j | i }d|d< t j t 5 t || d d d d S # 1 swxY w Y d S )Nr r r r ) r# rA ctypesmemsetr4 r __deepcopy__r rB AssertionErrorr
)rT r/ rR r s r) test__deepcopy__z TestMethods.test__deepcopy__ s
HQe $ $ $
ahmQ111
NN2 !
]>
*
* % % q! $ $ $ % % % % % % % % % % % % % % % % % %s /B
BBc G d d }t j d | dgd }t j t 5 | i d d d d S # 1 swxY w Y d S )Nc e Zd Zd ZdS ):TestMethods.test__deepcopy__catches_failure..MyObjc t r" r rT r r s r) r zGTestMethods.test__deepcopy__catches_failure..MyObj.__deepcopy__ s ""r+ N)ra rb rc r rd r+ r) MyObjr s#
#
#
#
#
#r+ r r3 r r+ r )r# r r rB r r )rT r r' s r) test__deepcopy__catches_failurez+TestMethods.test__deepcopy__catches_failure s # # # # # # # # h5577Ac222
]<
(
( ! ! R ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !s A++A/2A/c t j g d }t j g d }t j g d }t j |||gd dg t j t j g d t j t j g d t j t j g d
d g t j t j g d t j t j g d
t j t j g d dg t j t j g d t j t j g d t j t j g d t t dfd t j dk rd}nd}dd|fg}t j g d| d t d g d t t j g d| d S )N) r )myfirstr )@ @@zid,word,number)r- r/ r6 )r r r )r r r )r r r word)r r r )r r r )r r r number duplicatec 4 ddg S )Nr/ r6 )r$ )r s r) r z-TestMethods.test_sort_order.. s AFF$ F.. r+ littler r )r rX col2)rR r3 r rn rZ r r r )r3 r rn r )r r r r )
r# r r
fromarraysr$ r r/ r r r r8 sys byteorder)rT x1r x3strtypemydtyper s @r) test_sort_orderzTestMethods.test_sort_order s
Xlll
#
#
X---
.
.
Xooo
&
& F r2rl2B C C dV QT28LLL11222QVRX&=&=&= > >???QXrx 8 8999 fX QT28LLL11222QVRX&=&=&= > >???QXrx 8 8999 hZ QT28LLL11222QVRX&=&=&= > >???QXrx 8 8999J . . . . 0 0 0 =H $ $GGG!FG#45 HAAA"
$
$
$ V QvY 0 0 0111Q!M!M!M'.0 0 0 1 1 1 1 1r+ c
V t j t j t j fD ]}t j d| }|d d d }| j D ]x}d|d|}t | | || t | | || y|dz dz }|dz dz }| j D ]u}d |z }t | | || t | | || v|dz }|dz }| j D ]u}d |z }t | | || t | | || vd
D ]}t j d D ]} t j
g d || z }
d
| }t |
t j t |
t j
| dt j
fdt d D }|d d d }t j d } | d d d } | j D ]u}d|z }t | | | | t | | | | vdt j
fdt d D t j }|d d d }t j d } | d d d } | j D ]u}d|z }t | | | | t | | | | vt j dt" }t% t d |d d < |d d d }t j d } | d d d } | j D ]u}d|z }t | | | | t | | | | vt j dt( fdt* fg } t j
d t d D | }|d d d }t j d } | d d d } | j D ]u}d|z }t | | | | t | | | | vt j ddd }|d d d }t j d } | d d d } dD ]u}d |z }t | | | | t | | | | vt j ddd }|d d d }t j d } | d d d } dD ]u}d |z }t | | | | t | | | | vt j
d d gddgg }t j
ddgddgg }t j
ddgddgg }
t | d! | t | d! |
t | |
t j
g }d"|_ t |j |j D ]Z}d# | }t t j ||! t j |t j
| [d$}t t j |d ! t j | t j
| t j d% } t j d% }t | d& | t j d%t6 }t | d& | t j
d' t d% D }t | d& | t j
d( t d% D t j }t | d& | d S ))Nrf r r5 zscalar argsort, kind=z, dtype=r4 r r3 zcomplex argsort, kind=%sr rJ rK z'byte-swapped complex argsort, dtype={0}rQ c 4 g | ]}t | z S rd rR rk r r s r) rl z,TestMethods.test_argsort..M # 555Qa#a&&j555r+ zstring argsort, kind=%sc 4 g | ]}t | z S rd rR r s r) rl z,TestMethods.test_argsort..X r r+ zunicode argsort, kind=%srW zobject argsort, kind=%sr r c g | ]}||fS rd rd rj s r) rl z,TestMethods.test_argsort..n s 111q!f111r+ z!structured array argsort, kind=%sr ra rX zdatetime64 argsort, kind=%srb z timedelta64 argsort, kind=%sr r2 r r z&test empty array argsort with axis={0}z'test empty array argsort with axis=Noner] r c g | ]}d S aaaaaaaaard rj s r) rl z,TestMethods.test_argsort.. 666ak666r+ c g | ]}d S r rd rj s r) rl z,TestMethods.test_argsort.. r r+ ) r# r r r r% rW r6 r r r" r rL r& r6 r r rA r7 r r/ floatr
rF r rY rh r9 rb )rT r/ rR r r r% rF rG rM rK r' r rrrZ r r s @r) r zTestMethods.test_argsort% s h 2:6 B BE #U+++A$$B$ A
B
B=ATT55I QVVXX - -4 - 8 8!SAAA QVVXX - -4 - 8 8!SAAAA
B rTAX
rTAXO ? ?D,t3C ***55q# > > > ***55q# > > > >
V
VO ? ?D,t3C ***55q# > > > ***55q# > > > > F FJl9-
F
Fh 4 4 4JOLLL?FFrJJ S[[]] Ys3xxrw ? ? ?F F F F
F
H5555%**555 6 6
dddGLLNN IcNN
tttWO ? ?D+d2C ))t)44a = = = ))t)44b# > > > >
H5555%**555RW E E E
dddG IcNN
tttWO ? ?D,t3C ))t)44a = = = ))t)44b# > > > >
HV6 * * *E#JJ !!!
dddG IcNN
tttWO ? ?D+d2C ))t)44a = = = ))t)44b# > > > > XU|c3Z0
1
1 H11eCjj111 < < <
dddG IcNN
tttWO ? ?D5 > > >
IaO 4 4 4
dddG IcNN
tttW# ? ?D/$6C ))t)44a = = = ))t)44b# > > > >
Ia$4 5 5 5
dddG IcNN
tttW# ? ?D047C ))t)44a = = = ))t)44b# > > > >
Hq!fq!f% & & Hq!fq!f% & & Hq!fq!f% & &QVVXX%%1%--q111QVVXX%%1%--q111QVVXX%%''+++
HRLL 16'16** ? ?D:AA$GGC AD111 q888#
?
?
?
?7RZ---]17799BG<< HaXU + + +
NN28Cu555 6 6Q3 ; HaXU + + +
NN28Cu555 6 6Q3 r+ c v t j d }| g dd }t |g d | g dd }t |g d t j d }| g dd }t |g d | g dd }t |g d t j d }| g dd }t |g d
| g dd }t |g d d S ) Nr rt r ru r r3 r r r3 r3 r2 )r r r2 )r r2 r2 )r# rJ r r r s r) test_searchsorted_n_elementsz(TestMethods.test_searchsorted_n_elements s1 GAJJ
NN999f - -Q """
NN999g . .Q """ GAJJ
NN999f - -Q """
NN999g . .Q """ GAJJ
NN999f - -Q """
NN999g . .Q """""r+ c t j d }t j |j |j z dz d }|dd |j }||d d < | |d }t || | |d }t ||dz | |d }t || | |d }t ||dz d S )NrQ r3 rB r r ) r# r% rA r@ rH rm r/ r r )rT rR r unalignedr s r) !test_searchsorted_unaligned_arrayz-TestMethods.test_searchsorted_unaligned_array s IbMM(1:.2G<<ABBK$$QW-- !!! " "1f - -Q " "1g . .QA
NN9f - -Q
NN9g . .QA r+ c t j d }| g dd }t |g d | g dd }t |g d d S )NrW )r rW r r )rW rW r r )rW rW rW )r# r% r r r s r) test_searchsorted_resettingz'TestMethods.test_searchsorted_resetting
sk IaLL
NN999f - -Q """
NN999g . .Q """""r+ c d d t j d t j d t j d df }|D ]b}|dk rd}|dk r+t j d |
}t j d }n+t j d d |
}t j d }| |d
}t || | |d }t ||dz t j d d| }| |d
}t |t j t | t j
| |d
}t |t j d t j
dd S )Nr r rC Datetime?OMM8[D]?r2 r r rW r r r3 r+ rF r r/ ) joinr# r" r% r r rE r
r9 r& r6 )rT r rK rR r r r s r) test_searchsorted_type_specificz+TestMethods.test_searchsorted_type_specific
s l3R\*5M j149 : : > >BSyy SyyIar***illIa"---illq&))A C q'**A C!G $ $ $
3b999Aq&))A q"(3q66"A"A"A B B Bq&))A q"(1BG"<"<"< = = = =' > >r+ c z t j g dt j t j t }t fdD | t fdD |dz t fd|D | t fd|D |dz d S )N)z'P:\20x_dapi_cy3\20x_dapi_cy3_20100185_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100186_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100187_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100189_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100190_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100191_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100192_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100193_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100194_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100195_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100196_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100197_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100198_1z'P:\20x_dapi_cy3\20x_dapi_cy3_20100199_1r c < g | ]} |d S r r rk r rR s r) rl z9TestMethods.test_searchsorted_unicode..=
s' ;;;AannQ//;;;r+ c < g | ]} |d S r r r s r) rl z9TestMethods.test_searchsorted_unicode..>
s' <<.?
s+ @@@qannQqT622@@@r+ c H g | ] } | d S r r r s r) rl z9TestMethods.test_searchsorted_unicode..@
s+ AAAannQqT733AAAr+ )r# r r r% r& r )rT r} rR s @r) test_searchsorted_unicodez%TestMethods.test_searchsorted_unicode'
s
H
C
C
C 7
$
$
$ iA ;;;;;;;SAAA<<<>III y}} IIKK
!
IIKK(1:.2G<<ABBK$$QW-- !!! " "4 3 3Q " "4! 4 4Qq!!! !!!
NN9fa 0 0Q
NN9gq 1 1Qq!!! l3R\*5M j149 : : > >BSyy SyyHaV2... HaV28444h1v&&H___B777 H___BH===hbg>>>q&!,,A C q'1--A C!G $ $ $
3b999Aq&!BQB%00A q"(3q66"A"A"A B B Bq&!,,A q"(1BG"<"<"< = = = =
H___ % %huBG,,,ADqD "??CCaC !Hhbg666
NN1fa ( (Q
NN1gq ) )Qa r+ c G d dt j }t j d | }t j dd | }t j d | }t t | |d | t t | |d | t t | |d| | t t | |d| | d S )Nc e Zd ZdS )4TestMethods.test_searchsorted_return_type..ANr` rd r+ r) r r
rf r+ r rW r3 r r r )r# rE r% rm r r r )rT r rR r r s r) test_searchsorted_return_typez)TestMethods.test_searchsorted_return_type
s/
IaLL a IaOO # # IaLL a Jq~~a 8 8!<<<===Jq~~a 9 91===>>>Jq~~a ; ;Q???@@@Jq~~a! < d}t
t j ||d | | d S )Nr r rd r z(test empty array partition with axis={0}r z)test empty array partition with axis=None) r# r rF r r rL r r rh rT r r rR r r% s r) test_partition_empty_arrayz&TestMethods.test_partition_empty_array
s hq ***2. HRLL 16'16** B BD<CCDIIC a4888!S A A A A9R\!St444aggiiEEEEEr+ c t j d| d }t j g }d|_ t |j |j D ][}d | }t
t j ||| t j |t j | \d}t
t j ||d t j |
t j | d S )Nr r rd r z+test empty array argpartition with axis={0}r z,test empty array argpartition with axis=None) r# r rF r r rL r r rY r6 rh r s r) test_argpartition_empty_arrayz)TestMethods.test_argpartition_empty_array
s hq ***2. HRLL 16'16** ? ?D?FFtLLC a4888 q888#
?
?
?
?<R\!St444]17799BG<<. s@ $ $ $b"$ $ q' $ $ $ $r+ r r r3 z %d: %r <= %rr% z %d: %r < %r)!r# r% r r) r r8 r r r
rJ rW r$ r rH r rollr1 r isnanrw r AxisErrorr r r r tilemapr r rF rL r )rT r rR r okthr rM r r r tdrK r aaeatr d0r dcs r) r zTestMethods.test_partition
s IbMMi q!!<<<<j",1:FFFFj"/1ajIIIIj!+qqzJJJJj!.!!*MMMM! G J G JA A r|Aqq 9 9 91 = = = rq!! < < !>!> ? !|Aqq 9 9 9
;
;
; (;;;''C88::D L2 , , , sD ) ) )!fq!fq!f-
+
+HQKKgajj"2<11#=#=#=a#@#a&III"2<11#=#=#=a#@#a&III"1R_Q%B%B%B#C#%<11#=#=#=? ? ?"1R_Q%B%B%B#C#%<11#=#=#=? ? ? qv / /AabbEOOAAO...."1c**** iiIIIyyy iiIIIyyyB
+
+HQKKgajj"2<11#=#=#=a#@#a&III"2<11#=#=#=a#@#a&III"2<11#=#=#=a#@#a&III"1R_Q%B%B%B#C#%<11#=#=#=? ? ?"1R_Q%B%B%B#C#%<11#=#=#=? ? ?"1R_Q%B%B%B#C#%<11#=#=#=? ? ? qv / /AabbEOOAAO...."1c**** A r|Aqq 9 9 91 = = = q AA!>!>!> ? !|Aqq 9 9 9
;
;
; "
A a333A6 : : : a!444R8" = = = q AA!>!>!> ? !|Aqq 9 9 9
;
;
; q BQ!?!?!? @ !|Ar : : :
<
<
< "
ddd#A a333A6 : : : a!444R8" = = = q AA!>!>!> ? !|Aqq 9 9 9
;
;
; q BQ!?!?!? @ !|Ar : : :
<
<
<
r|Ar : : : !|Ar : : :
<
<
< r|As ; ; ; !|Ar : : :
<
<
< q BQ!?!?!? @ !|Ar : : :
<
<
<
'""A16Q;''A&A+/C a--c2C 8 8 8 '""A16Q;?++A&A+/C a--c2C 8 8 8 AAaD a11"5q 9 9 9 a11!4a 8 8 8 2?1g667;Q ? ? ? 2?1g667:A > > >6AaD BHQrq' : : ;B ?@@ A A A BHR\!W 5 5b 9:: ; ; ; "
!A'")B--!+,,C I a 16]]
D
D R\!QQ 7 7 7 :CFCCCC q AA!>!>!> ? !|Aqq 9 9 9
;
;
; q BQ!?!?!? @ !|Ar : : :
<
<
<16]]
+
+!""**** q# & & & , , , - -A ..C a1555c:L I I I 2?1c:::;C@, O O O!Q A
KKK " " " *ak1 5 5 5 ", QQ ? ? ? ? *blAq 9 9 9 ", a C C C C *ana 8 8 8 ", B B B B *boq! < < < ",AA F F F F "
%%f--A
KKK * * *
KKK * * * LAAA . . . . LAAA . . . . LADq 1 1 1 1 LADq 1 1 1 1
NN111N - - -
NN111N - - - OAqqq 1 1 1 1 OAqqq 1 1 1 1 OAqt! 4 4 4 4 OAqt! 4 4 4 4 *ak11 = = = = *ak2A > > > > )Q[!$ ? ? ? ? *blAqq A A A A *blAr E E E E *anaa @ @ @ @ *anbq A A A A *boq!! D D D D *boq"4 H H H H$ $RXrz2<$H $ $ $B /
J /
JA( Iar*** !!!$$$WRYq 3 3 3V<<BI%r***\"%% qv &J &JA Q222A 1q)))%aeQqT222%adAa!effI666C1R_Q : : :;<<< R;;;AC!!!Q$1# *;2!F!F!FGGGB!!!RaR%
a1g-2244)Q!!!Q$111bqb5,DDF F F FB!!!QUVV) 111a405577(AqAw!!!QUVV) +GGI I I IC2bi 4 4QQQW =AAA>>> ? @ A A A R;;;AC!QQQ$1# *;2!F!F!FGGGB"1"aaa%AadG+0022)Q!QQQ$2A2qqq5,BBD D D DB!a%&&!!!) qAAAw.3355(AqAAAw!!!QUVV) +EEG G G GC2bob!!! D D D "(1+..tQQQw7 8 9 : : : BLLL+++ R\!QQ%?%?%?@@@BLLL333 R\"aaa%H%H%HIIIBLLL333 R\"aaa%H%H%HIIIIM&J/
JqG J G Jr+ c d}t j | D ]s}t ||| || d|z t ||d || k d|||d || fz |dz }td S )Nr zkth %dr z kth %d, %r not greater equal %dr, r3 )r# r$ r r r )rT r r prevrR s r) assert_partitionedz TestMethods.assert_partitioned s A aQi1x!| D D D D QqrrUad]''))9Q!""qt > > "
ddd#A # #BLQCE$:$:QC @ @ @ # #BLQCEQGO$D$D%&CEQGO
5
5
5 # #Aboa!Q&?&?$@1# F F F # #Aboa!Q!R &I&I$J%&CEQGO
5
5
5 "
A I a 2 //B !2 & & & b!!B C R1---ABIbhqk**111d73OOCaO001 2B q" % % %28A;''
5
5''!AAA#4444 R1---ABOB!444Ibhqk**4623 4B q" % % %28A;''
6
6''!!!Q$5555
6
6r+ c
t j g dg d }t j |ddg }t t j |t |j ddg | t |t j |t |j ddg | t |j D ]n}t t j ||ddg | || t |t j ||ddg | || ot j g d }t j | }t t j |t |j | t |j D ]f}t t j || | || t |t j || | || gd S )N))Galahadg333333?& )Arthurg?r& )Lancelotgffffff?rB ))r z|S10)heightr )ager r rF rE r6 )rA rC zebrarD ) r# r r$ r
r r rH r r )rT r rM rR s r) test_partition_cdtypez!TestMethods.test_partition_cdtype s! H * * *MMM
O
O
O gax 01112<5==/4h.? A A A 1R_Qaf
4983D F F F G qv ! !A a5(2CDDDQGA
2?1ax7HIIIJ1M Q
!
!
!
!
H??? @ @gajj2<5==993???qv > >A a++A.A 7 7 7 2?1a001!4c!f = = = = > >r+ c t j d }d d }t t |j d| t t |j d| d S )NrQ r r r2 r4 )r# r% r r r8 r r r s r) test_partition_unicode_kindz'TestMethods.test_partition_unicode_kind sZ IbMM v & &j!+qq9999j!.!!<<<<<}| | } t || | | t || j
| | t || | j
| t || j
| j
| t || j
| | t || | j
| t || j
| j
| @t j t j t j t j fD ]}| | } | | }
t ||
|
| t ||
j
|
| t ||
|
j
| t ||
j
|
j
| t j t j t j t j fD ]}| | } t || d d dd d f | j
|| d d dd d f | j
t || d d d d df | j
|| d d d d df | j
t || | d d dd d f j
|| | d d dd d f j
t || | d d d d df j
|| | d d d d df j
t || d | j d dz d d f | d d dd d f j
|| d | j d dz d d f | d d dd d f j
t || d d dd d f | d | j d dz d d f j
|| d d dd d f | d | j d dz d d f j
t j t j t j t j fD ]}| | } | | } | | }
t || | j
| t || j
| |
d S )Nr3 r r r r r )7 E )rZ i ' )r[ r] i )r\ r^ r_ i )i @ d )r` iD l )ra re i )rb rf ri i $ T )rc rg rj rl iX )rd rh rk rm rn i r5 r2 )
r# r r% r r r r r r r rL rW rF )rT ri rR r rZ r ddtdtdeteafebfedfeddtfedtdfs r)
test_arr_multzTestMethods.test_arr_multB s Hq!fq!f% & & Hq!fq!f% & & Hq!fq"g& ' ' IbMM ! !!Q ' 'h
%
%
%
%
%
%
%
%
%
%
%
%
'
h
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
-
:rz2< G @ @B((2,,C c3 - - - ceS))3 / / / c35))3 / / / ceSU++S 1 1 1 cejjllC00# 6 6 6 c35::<<00# 6 6 6 cejjllCEJJLL993 ? ? ? ? :rz2< G 2 2B((2,,C((2,,C c3 - - - ceS))3 / / / c35))3 / / / ceSU++S 1 1 1 1 :rz2< G B((2,,C S2qqq\35))S2qqq\&&((#%**,,77
SDDbD\35))SDDbD\&&((#%**,,77
S#dddAAAg,.))S#dddAAAg,.--//00
S#aaa2g,.))S#aaa2g,.--//00
S+#)A,!++QQQ./SSqS!!!V ??S+#)A,!++QQQ./4466CCaCF
8J8J8L8LMM
S1aaa[#&8sy|q'8&8!!!&;"<">??S1aaa[%%''-?cila.?-?-B)C)E)J)J)L)LMM
:rz2< G 2 2B((2,,CJJrNNEJJrNNE c35))5 1 1 1 ceS))5 1 1 1 1 2 2r+ ifdFDc & t j d| dd }t j dd } ||| } ||| }t || ||j |j } ||j |j }t || t j | }t j d|j z dz dz d dd
| }| dd }|d
}t |j d d
|j z d
k ||| } || | }t || ||j |j } ||j |j }t || d S ) N @r r2 r r6 r+ r r3 r r r4 )
r# r% r broadcast_torW r rL r/ r9 r@ rm r rC )rT ri r/ rR r ret1ret2rK s r)
test_no_dgemvzTestMethods.test_no_dgemv s
Ic ' ' ' / /1 5 5 OB ' 'tAqzztAqvvxx T4 tAC~~tACHHJJ$$T4 Xe__ HQ_)A-W = = =abb A F Fu M M
IIaOO
aD%f-a02;>!CDDDtAqzztAFFHHaffhh''T4 tAC~~tACHHJJ
++T4 r+ c t j ddgddgg }t j ddgddgg }t j ddgddgg }t t j || | | t t j t j || | | | | t j | }| || t |t j || t j | }| || t |t j || d S )Nr3 r r rX )r r )r# r r dotrY rT rR r rZ s r) test_dotzTestMethods.test_dot s7 Hq!fq!f% & & Hq!fq!f% & & Hq!fq"g& ' 'RVAq\\15588,,,RVBF1aLL!,,aeeAhhll1oo>>>
M! a Qq! %%%
M! qQq! %%%%%r+ c d}t j dd }t t t j || t t t j || d S Nr6 r r2 r )r# r r r) r rT rZ r s r) test_dot_type_mismatchz"TestMethods.test_dot_type_mismatch sM HU% ( ( (iA...iA.....r+ c t j d d t j d D }|D ] }t j dd | }t
d| }t j dd |d< t j || }t j ||| }t ||t | t t t j |||d d d
t t t j |||j d S ) Nr3 c > g | ]}|d vt j | S )USVMr# r/ )rk codes r) rl z8TestMethods.test_dot_out_mem_overlap.. s4 ) ) )T'' (4..'''r+ rU r rV r .r r r2 )
r# rw seedr" ry r rL r r r r r8 rL )rT dtypesr/ rR r r/ r s r) test_dot_out_mem_overlapz$TestMethods.test_dot_out_mem_overlap s*
q) )R\%-@ ) ) )
=
=E q!$$++E22A vU333AY^^Aq))AcFq! Aq!###A AtE{{ 3 3 3 3
*bfa##A# ? ? ? ? *bfa < < < < <
=
=r+ c V G d dt j }t j d | }t j d | }t j d }t j ||| t j ||| d S )Nc e Zd ZdS ),TestMethods.test_dot_matmul_out..SubNr` rd r+ r) Subr rf r+ r r3 r )r# rE rJ rm r matmul)rT r rR r r s r) test_dot_matmul_outz TestMethods.test_dot_matmul_out s "* GFOO % % GFOO % %gfoo q!
!QC r+ c F G d d }t t t j | | t t t j | | t t t j | | d S )Nc e Zd Zd ZdS )@TestMethods.test_dot_matmul_inner_array_casting_fails..Ac t r" NotImplementedErrorr s r) r zJTestMethods.test_dot_matmul_inner_array_casting_fails..A.__array__ ))r+ Nr rd r+ r) r r #
*
*
*
*
*r+ r )r r r# r r innerrT r s r) )test_dot_matmul_inner_array_casting_failsz5TestMethods.test_dot_matmul_inner_array_casting_fails s * * * * * * * *
)261133<<<)29aacc1133???)28QQSS!!##>>>>>r+ c t j d ddd }t j || }t j ||| }t ||u t || t j d ddd }t j |||d d ddf }t |j |j u t || d S )Nr| r2 r r r5 .)r# r% r r r r r r s r) test_matmul_outzTestMethods.test_matmul_out s IbMM ! !!Q * * IaOO Ia " " "QQ IbMM ! !!Q * * Ia$$B$) - - -!& !!!Qr+ c t j d d }t | g d t | d g d t | d g d t | d dd g t t j |j dd
t t j |j d
d t t j |j d
d
t t |j dd t j d d
}t | ddgddgg t | d ddgddgg t | d dgdgg t | d dgd
gg t t |j dd t | ddd ddgddgg t | ddd ddgddgg t | ddd dgdgg t | ddd ddgddgg d S )Nr| r )r rW rQ r r3 )r3 r r? r5 r r rW )axis1axis2r r2 r2 r2 r r r2 r rK r r )r# r% r r diagonalr r/ r8 r s r)
test_diagonalzTestMethods.test_diagonal
sx IbMM ! !& ) )QZZ\\:::...QZZ]]JJJ///QZZ]]JJJ///QZZ^^aV,,,blAJaqAAAAblAJaqAAAAblAJaqAAAAj!*AQ???? IaLL + +QZZ\\QFQF#3444QZZ]]aVaV$4555QZZ]]aS1#J///QZZ^^qcA3Z000j!*AQ????QZZ1a((Aq6Aq6*:;;;QZZ1a((Aq6Aq6*:;;;QZZqZ;;qcA3ZHHHQZZ1a((Aq6Aq6*:;;;;;r+ c " t j d }t |j j t |j j t j t j d }t |j j t |j j t j t j d }t |j j t |j j d S Nr )r# eyer r rY rZ r diagr s r) test_diagonal_view_notwriteablez+TestMethods.test_diagonal_view_notwriteable
s F1II AG%%&&&AGO#$$$ Kq " "AG%%&&&AGO#$$$ GBF1II AG%%&&&AGO#$$$$$r+ c . t j d }t r%t t j | dk t
d D ]}| t r't t j | dk d S d S )Nr] r] r r] )r# r9 r r r getrefcountr r )rT rR r s r) test_diagonal_memleakz!TestMethods.test_diagonal_memleak$
s HZ - COA&&+ , , ,s A
JJLLLL - COA&&+ , , , , , - -r+ c t j g t j }t j d }t d D ]}t j ||| t
r't
t j | dk d S d S )Nr @r] r r ) r# r r r r r r r r )rT rR r _s r) test_size_zero_memleakz"TestMethods.test_size_zero_memleak.
s
HRrz * * * HSMMs A F1aQ - COA&&+ , , , , , - -r+ c x t j d d }t | d t | d d t | d d t | d d t j d d
}t | d d g t | d d d g t | d d d
g t | d ddg t | ddd d d g t | ddd ddg t | ddd d
dg t | ddd dd
g t j d }| | }||u sJ d S )Nr| r r! r r3 r| r5 rE r r r r2 r r rW r r? r r )r# r% r r tracer )rT rR r r r s r)
test_tracezTestMethods.test_trace9
s IbMM ! !& ) )QWWYY###QWWQZZ$$$QWWQZZ$$$QWWR[["%%% IaLL + +QWWYYA'''QWWQZZ!Q(((QWWQZZ!Q(((QWWR[[1a&)))QWWQ1%%1v...QWWQ1%%1v...QWWQ1%%2w///QWWAQaW881a&AAAhqkkgg#g czzzzzzr+ c G d dt j }t j d d | }| }t
t || d S )Nc e Zd ZdS )0TestMethods.test_trace_subclass..MyArrayNr` rd r+ r) MyArrayr Q
rf r+ r r r )r# rE r% r rm r r r )rT r r r. s r) test_trace_subclassz TestMethods.test_trace_subclassN
s bj
IaLL + + 0 0 9 9
GGII
1g&&'''''r+ c t j d }t j d }||z dz D ]}t j g d| }t j d| }| g dg d t || t j d| }| g dg d t || d d
d D ]}t j g d | }t j d| }| g dd
gd
z t || t j d| }| g dd
gd
z t || d d
t j d }d|j _ t t |j g dg d g d}t t t j |dd gd d S )Nr rC r+ )r r3 r r r rW r r )r3 r rW r r2 r r )FTFTFTTFrU r rW ) r# r" r r9 putr r rY rZ r r8 r) )rT icodesfcodesrK rM rR bad_arrays r) test_putzTestMethods.test_putX
s l+ j)6/C' / /B(---R888C "%%%A
EE)))YYY ' ' ' C r***A
EE)))YYY ' ' ' CKK1 - - . . . . / /B(BBB"MMMC "%%%A
EE)))dVAX & & & C r***A
EE)))dVAX & & & CKK1 - - . . . .
HQKK !j!%III>>>
II iQFA>>>>>r+ c r t j ddgddgg }t | g d t | j j t | d g d t | d g d t | d g d t | d
g d t | d
j j t | d g d t | d j j t | | d t j ddgddggd }t | g d t | d
g d t | d g d t | d
j j t | d j j t | | d t | d
| d d
t j ddgddgg d d d d d f }t | g d
t | d g d
t | d g d t | d
g d
t | d g d
t | d j j t j d d d d }t | d j j t | d j j t | d j j t j d d d d }| dddd d d }t |j }d|d< ||_ t | d j j t | d t j ddd t j d }| dddd d d }t |j }d|d< ||_ t t j | d | t | d t j d t j d d d d dd }t | d j j t | d j j t | d g d t | d g d t j dgg }d|_ t j
d j dk rht t j | d | t | d j |j j f dD ]o}t j d }t | | dg t t j | | | pt j d d d d dddd }|dd d df }t | d g d t | d g d t | d
g d t | d g d |d d ddf }t | d g d t | d g d t | d
g d t | d g d d S ) Nr r3 r2 r )r r3 r2 r rY )r r2 r3 r r, r6 r Kr5 )r2 r r r3 )r2 r r r3 rQ r r r! r! r r r )r! i r r, rY r r r .)r r r r| r r r )r r r r r r r| r )r r2 r r r rQ r| r )r r r r| r2 rQ r r )r# r r rh r rY r r r% swapaxesr r may_share_memoryrJ r/ r@ )rT rR r r0 r s r)
test_ravelzTestMethods.test_ravel
s$ Hq!fq!f% & &QWWYY ---AGGIIO++,,,QWWS\\<<<000QWW3W'' 666QWW3W'' 666QWW3W'' 666AGG#G&&,44555QWW3W'' 666AGG#G&&,44555QWWYY "
... Hq!fq!f%S 1 1 1QWWYY ---QWW3W'' 666QWW3W'' 666AGG#G&&,44555AGG#G&&,44555QWWYY "
...QWW3W''2S)A)ABBB Hq!fq!f% & &tttQQQw /QWWYY ---QWW3W'' 666QWW3W'' 666QWW3W'' 666QWW3W'' 666c""(0111
IbMM##A# "*+++ "*+++ "*+++
Ih ! $
IIaAq ! ! * *2r 2 2qy//
c""(0111QWWS\\29QA#6#6777
IdOO
IIaAq ! ! * *2r 2 2qy//
#AGG#G$6$6::;;;QWW3W''4 999
IaLL2 & &q! , ,c""(0111c""(0111QWWS\\<<<000QWWS\\<<<000
HqcUOO 71:: % % B' a88 9 9 9 -0@/B C C C) < #>#>???QWWS\\#>#>#>???QWWS\\#>#>#>???QWWS\\#>#>#>?????r+ c G d dt j }t j d | }t t | d | t t | d | t t | d | t t | d | t j d d d d | }t t | d | t t | d | t t | d | t t | d | d S ) Nc e Zd ZdS )6TestMethods.test_ravel_subclass..ArraySubclassNr` rd r+ r)
ArraySubclassr
rf r+ r rQ r, rY r r r2 )r# rE r% rm r r rh )rT r rR s r) test_ravel_subclassz TestMethods.test_ravel_subclass
s BJ
IbMM } - -
1773<< 77888
1773<< 77888
1773<< 77888
1773<< 77888 IbMM##A# # #M 2 2
1773<< 77888
1773<< 77888
1773<< 77888
1773<< 7788888r+ c t j d dddd }t j |j }t
|j d | }t t j |j
dd t t j |j
dd t t j |j
dd t t j |j
dd t d d D ]f}t d d D ]Q}t ||f D ];\ }}|
|| }t |j } |j | | |< |j | | |< t |j | t |||f d
|j D \ }
} } }
d |j D \ }}}}t ||| || || || f |||
|| || ||
f t |||f t
|j d t |||f |dk r|}=Shd S ) Nr r3 r2 r r OWNDATAr c g | ]}|d z
S r rd rk dims r) rl z-TestMethods.test_swapaxes.. s %?%?%?c!e%?%?%?r+ c g | ]}|d z
S r rd r s r) rl z-TestMethods.test_swapaxes.. s %A%A%Ac!e%A%A%Ar+ )r# r% r rW indicesrF r rY r r/ r r enumerater r r )rT rR r- r r r rR srcrZ rF i0r i2i3j0j1j2j3s r)
test_swapaxeszTestMethods.test_swapaxes
sV Ig & &q!Q 2 2 7 7 9 9j!! "###
FFHHblAJA666blAJ1555blAJ2666blAJ1555r1 A2q\\
'A / / FAs Q**A OOE "y|E!H "y|E!H %aAY @@@%?%?qw%?%?%?NBB%A%Asy%A%A%ANBB SWc"gs2wB%G!H!"3r7CGSWc"g#E!F!$aAY 1 1 1 2 2CAq NNCCCAvv #
r+ c B t j g d }t j |j t j |j t | t |t j t j g dd }t j |j t j |j t | t |t j t j g d }t | t | t |t j t j g d }t | t | t |t j t j g dt }t |d D t | t |t j t j g dt t t fd t t fd d S )
N ? r? y 7@ 7@rY rU r6 r @)r r? r3 r c 6 g | ]}| S rd conjugate)rk rR s r) rl z.TestMethods.test_conjugate..( s 333A!++--333r+ )r r3 r r c , S r" )conjr s r) r z,TestMethods.test_conjugate..- s r+ c , S r" r r s r) r z,TestMethods.test_conjugate... s r+ )
r# r r r r r r r7 r r) )rT acrR s @r) test_conjugatezTestMethods.test_conjugate s H+++ , ,
VVXXQVRW%%%QVbgX&&&R '''R a ))) H+++S 1 1
VVXXQVRW%%%QVbgX&&&R '''R a ))) HYYY
VVXXQR '''R a ))) H___ % %
VVXXQR '''R a ))) H)))6 2 2
VVXXR33333444R '''R a ))) H(((& 1 1i!1!1!1!1222i!6!6!6!677777r+ c t j g d }t j | }| | }||u sJ t || d S )Nr )r# r r r r
)rT rR r rX s r) test_conjugate_outz TestMethods.test_conjugate_out0 se
H+++ , ,mA kk# czzzz3
.....r+ c F g d}|D ]}t j d| }t j dg| }t j dggggg| }d | }t | }t ||| t t 5 t | }d d d n # 1 swxY w Y t ||| t t 5 t | } d d d n # 1 swxY w Y t | || d S )Nr r r r( ry u2u4u8r r grY DGr r+ r r z
dtype: {0})r# r rL rb r r
r )
rT r rK rR r rZ r% apbpcps
r) test__complex__zTestMethods.test__complex__9 s % %B"%%%A!B'''AaSE7)B///A %%b))CB Q $ $ $ 011
QZZ
Q $ $ $ 011
QZZ
Q $ $ $ $ % %s$ B77B; >B; 'DD
D c g d}|D ]5}t j g d| }t t t | 6t j ddg }t j d| }t t t | t j ddg| }t t t | t j d }t t t | t j dgd }t
t 5 t t t | d d d d S # 1 swxY w Y d S )
Nr rU r r9 r )r6 r )gMb`?r z1+1jr^ )r# r r r) rb r/ r
r )rT r rK rR r rZ r r s r) test__complex__should_not_workz*TestMethods.test__complex__should_not_workO so 1 1B"---A )Wa 0 0 0 0
X{K0
1
1 HXR ( ( (i!,,, Hh *" 5 5 5i!,,, HV i!,,, HfXs # #
,
-
- 1 1 )Wa 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s D99D= D=)lra rb rc r6 r r r r r r r r r7 r r r& r) r/ r r# rB r r r float16r r r r8 r r r r r< rH rN r r rU r[ r/ r r
r` rd rf ri rs rz r r r r r r r r r r1 r r r r r r r r r r
r" r r r r r r r r: r? rH rJ rL rN rV r r rw r~ r r r r r r r r r r r r r r r r r r r r rd r+ r) r r ' s 444J3 3 3*4 4 4( & - - -<E E E$ [ WtVn55 . . 65 .>
<
<
< > > >"8 8 8@ @ @& & &2 [ WrxBIry')z2:rz')}'6 7 7
$
$7 7
$ [ W grx28RZ j"*bm EF F
$
$F F
$ [ Wrz2:r}&MNN [ Vff%566% % 76 ON%.* * * [ Wry"'&:;; $ $ <; $ $ $ $ [ T BHsElS#J/ 0 0 BHsElS&M2 3 3$5 6 6 [ VaV,,@ @ -,6 6@" [ W8H&IJJ $ $ KJ $: : : < < < $ $ $
:
:
:
+
+
+ 2 [ WsBHe&<==
%
% >=
%! ! !#1 #1 #1JG- G- G-R5 5 5 [ S!Q rz222!Q rz222!Q #
0# # #( "# # #> > >2L L L6 R R R K! K! K!Z
B
B
B [ Wbl5&9::7 7 ;:7 [ Wbl5&9::4 4 ;:4 9 9 96 6 6 [ [",|*DEE F F FE F [ [",|*DEE C C FE CNJ NJ NJ` B6 B6 B6H> > >2= = = L L L [ [",|*DEE4 4 FE4 6 6 6 [ Vbfbi%899K2 K2 :9K2Z [ Vbfbi%899 [ Wg..! ! /. :9!8& & &$/ / /= = =,
!
!
! ? ? ?
< < <. % % %- - - - - - *( ( (&? &? &?PY@ Y@ Y@v9 9 9 @#8 #8 #8J/ / /% % %,1 1 1 1 1r+ r c e Zd Zd ZdS )TestCequenceMethodsc t dt j d dd v t dt j d dd v d S )N @g 0@r 4@)r r# r% r rS s r) test_array_containsz'TestCequenceMethods.test_array_containsg s^ ry~~--a222333BIcNN 2 21Q 7 7788888r+ N)ra rb rc r rd r+ r) r r f s# 9 9 9 9 9r+ r c e Zd Zd Zd Zej dddg d Zd Z d Z
d Z d
Z d Z
d ZdS )
TestBinopc t t j dg t j ddg z ddg t j ddg d d d }t ||t j ddg z z ddg t j dg }t j dg }||z }||z
}||z }||z }t || t |d ||dz |z |z z ||z z
}t || t |d t j dg }t j dg }||z |z }t |d t |d t |d d S ) N ?r6 r r2 ?rW r g @)r r# r r r )rT r rR r rZ s r) test_inplacezTestBinop.test_inplacel s !"(C5//BHc3Z4H4H"H#&* . . .
Hc3Z 1 %!!q28S#J+?+?'?"@#'+ / / /
HcUOO HcUOO
E
E
E
EQAr"""
B
Q Q &QQ
HaSMM HaSMM
UaKAv&&&QQr+ c t j dt ft j dt ft j dt ft j dt ft j dt ft j dt ft j dt ft j dt ft j dt ft j dt ft j
dt ft j dt ft j dt ft j dt fd G d dt" fdd d d d
} d fd } dfd
} | |t$ ddd | |t$ d ddd | |t$ d ddd | |t j d dddd | |t$ d| ddd | |t j d| ddd | |t$ d ddd | |t j d dddd d S )NTF)r subr? truedivfloordivmoddivmodpowlshiftrshiftandxororr c e Zd ZdS )7TestBinop.test_ufunc_binop_interaction..CoercedNr` rd r+ r) Coercedr rf r+ r c r" rd )rT r s r)
array_implz:TestBinop.test_ufunc_binop_interaction..array_impl s Mr+ c dS )Nforwardrd rn s r) op_implz7TestBinop.test_ufunc_binop_interaction..op_impl 9r+ c dS )Nreverserd rn s r) rop_implz8TestBinop.test_ufunc_binop_interaction..rop_impl r" r+ c dS )Nzin-placerd rn s r) iop_implz8TestBinop.test_ufunc_binop_interaction..iop_impl s :r+ c d||||fS )N__array_ufunc__rd )rT ufuncmethodr r s r) array_ufunc_implz@TestBinop.test_ufunc_binop_interaction..array_ufunc_impl s %ufdFC Cr+ __main__c di}|dur||d< D ]J}
|d | < |d | < |d | < K|dur||d< | ||d}t d | }t |t j r.make_obj s *J 7O U**8F 45
B
B8? 0 0 4 459A
1 1" 5 569A
1 1" 5 566%''5@ 12&*1@*8 N D(* *F&"*--
yA..q!4499&AAA vxxr+ c l D ]\ \ }}}dd|d|d|}t j dd| dd g} |r!| | d d | D ]0}
t |
d
} fd } |r t | | t | n|r!t | | d d | nt | t j
rVt | j t j
j u r1 | | | }
t |
j | j u | nt t f| | |
t |
d }|r< || }
t |
d d |
t |
d ||
nt | t j
rVt | j t j
j u r1 | || }
t |
j | j u | nt t f|| |
|rgt |
t j
rLt |
d }|r"t | | t |
n|r || }
t |
d d | t |
d || t t |
d d t" u | t |
d d d |
u | nut | t j
rAt | j t j
j u r t || |
u | nt t f|| |
t t$ d }|t t$ dz d }|t t& }t || |
d| t | t j
sA|r t ||
| d| n#|r!t ||
| d d | |r!t || |
d d | 2d S )Nzop: z , ufunc: z, has_inplace: z , dtype: r r r r2 r r0 c h dk r*t t | t | d S | S )Nr r )r r tuple)r$ r s r)
first_out_argzLTestBinop.test_ufunc_binop_interaction..check..first_out_arg s3 >> #Jvu$=$= > > >#)!9 ,#)Mr+ r) r r1 r3 r2 r5 r r r r$ )itemsr# r% r r r rL r NotImplementedr rE r r) r __class__r r) r> r> ro )objbinop_override_expectedufunc_override_expectedinplace_override_expected check_scalarr* has_inplacer/ r
check_objsr'
arr_methodr? rX arr_rmethod arr_imethodop_fnr r r: s @r) checkz5TestBinop.test_ufunc_binop_interaction..check s^ 3699;; Y
. Y
.//UK bb%%%ee = i1E : : : B B1a H H I
8 %%jmA&6777% S. S.C!(i.>.>r.B.B!C!CJ* * * * * / L$ZZ__ngNNNN0 L$ZZ__Q%79J%,. . . . 'sBJ 7 7 L !#YY 6 Z 7 8 8 #0-
3"@"@C #CMS]$BG L L L L )9g*>*4c7 L L L L #*#z/@/@/D/D"E"EK. M )k# . .$SV->-46 6 6 6$SVUGDDDDD &sBJ 7 7 M!%c!:!#!;"< "< #0- C0@0@"A"AC #CMS]$BG L L L L *9g*>*5sG M M M M
# ?z#rz'B'B ?&-c:3D3DR3H3H&I&I 4 ? (C .18 : : : : : 4 ?"-+c"2"2C (Q1BG L L L (Q @ @ @ #DR $8$8E$A7 K K K #CGEN1$5$$&J$>&? &? !( C(8(8C(? I I I I -y'.B.936=!? !? !? !? $Hb$ 7 7E } '"s(D A A } '" 5 5 sC )WEEE%c2:66 22 2 (sC )W M M M M 4 2 (sC );=N)0 2 2 2. .$UU3__Q%79J%,. . .eS.
Y
. Y
.r+ i )r5 r3 )rG )r5 r6 )r6 )FFr- )T)r# r r subtractmultiply true_divide floor_divide remainderr rp r
left_shift right_shift bitwise_and bitwise_xor
bitwise_orr r r7 rE )
rT r, r; rN r r r' r! r: r% s
@@@@@@r) test_ufunc_binop_interactionz&TestBinop.test_ufunc_binop_interaction s( u-dE2dE2 u5 $6 tU3E514- c2 s3 s3 s3 c2D%0
. i D D D >C$. 2 ;?[ . [ . [ . [ . [ . [ . [ .| hhv ue444 hhvf555ueUKKK
hhva000$tDDD
hhrz!444eUE " " " " hhva#35 5 56;T5 J J J
hhrz!#35 5 56;T5 J J J hhv4000$uEEE
hhrzt444dE5 " " " " " "r+ priorityN
runtime errorc G fdd } G d dt j }t j d | | z }|j dk sJ |d dk sJ d S ) Nc 0 e Zd Ze fd Zd ZdS )BTestBinop.test_ufunc_binop_bad_array_priority..BadPriorityc 2 dk rt d S )Nr[ z#RuntimeError in __array_priority__!r )rT rZ s r) r/ zUTestBinop.test_ufunc_binop_bad_array_priority..BadPriority.__array_priority__P s" ..&'LMMM r+ c dS )Nr$ rd rn s r) __radd__zKTestBinop.test_ufunc_binop_bad_array_priority..BadPriority.__radd__V s xr+ N)ra rb rc propertyr/ ra )rZ s r) BadPriorityr^ O sG
X
r+ rc c e Zd ZdZdS )BTestBinop.test_ufunc_binop_bad_array_priority..LowPriorityiNra rb rc r/ rd r+ r) LowPriorityre Y s !& r+ rg r rR r r$ )r# rE r% rm rF )rT rZ rc rg rX s ` r) #test_ufunc_binop_bad_array_priorityz-TestBinop.test_ufunc_binop_bad_array_priorityI s ' ' ' ' '"* ' ' '
ill ,,{{}}<yD 1v!!!!!!r+ c G d d } | }t j |dg }t d|vod|v t j |dgd }t d|vod|v t |d d t j |dgd }t d|vod|v t |d d d S ) Nc e Zd Zd ZdS )DTestBinop.test_ufunc_override_normalize_signature..SomeClassc |S r" rd rT r* r+ inputskws r) r) zTTestBinop.test_ufunc_override_normalize_signature..SomeClass.__array_ufunc__f s r+ Nra rb rc r) rd r+ r) SomeClassrk e s#
r+ rq r3 sig signaturezii->i)rr )rs )r# r r r )rT rq rR ro s r) 'test_ufunc_override_normalize_signaturez1TestBinop.test_ufunc_override_normalize_signaturec s
IKK
VAs^^R9Kr$9:::
VAs
(
(
(R5K2$5666R _g...
VAsg
.
.
.R5K2$5666R _g.....r+ c h G d d } | }t j d }t t j | d t t j || d t t j || d t t j ||f d t t j || d t t j || d t t j ||f d t t j || d t t j |d | d t t j ||| d t t j ||d f d t t j |||f d t t j |d |f d t t j |||f d t t j |||f d t t 5 t j || d d d n # 1 swxY w Y t t t j ||f t t j || d t t j || d t t j ||| d t t j ||| d t t j ||| d t t j |||f d t t j ||| d d S ) Nc e Zd Zd ZdS )4TestBinop.test_array_ufunc_index..CheckIndexc t | D ]
\ }}|| u r|c S t |d D ]\ }}|| u r|fc S d S r! )r )rT r* r+ rn ro r rR r s r) r) zDTestBinop.test_array_ufunc_index..CheckIndex.__array_ufunc__x sx %f - - ! !DAqDyy ! &bi 0 0 $ $DAqDyy !t !$ $r+ Nrp rd r+ r)
CheckIndexrw w s#
$
$
$
$
$r+ ry r r r r r r3 ) r# r% r sinmodfr r) r8 r )rT ry rR dummys r) test_array_ufunc_indexz TestBinop.test_array_ufunc_indexs s $ $ $ $ $ $ $ $
JLL "
RVAYY"""RVE1%%t,,,RVEq)))4000RVEt,,,d333RVAq\\1%%%RVA1%%%q)))RVAA4(((!,,,RWUA&&---RWUD!,,d333RWUE1--t444RWUD 222D999RWUE
333T:::RWUq 222D999RWU
333T:::RWQUAJ///333
9
%
% " " GEq ! ! ! ! " " " " " " " " " " " " " " " j"'5qd;;;; RVAu%%q)))RVE1%%q)))RVE5!,,d333RVE1a((!,,,RVE5a000$777RVE5qd333T:::RVAu!,,,a00000s #JJ
J
c
G d dt j }t j dg | }t j dg }t j dg }t j ||| t |d d t
t || d|d< t j ||| t |d d t
t || d S )Nc e Zd Zd ZdS )-TestBinop.test_out_override..OutClassc d|v rR| }| d t || } ||i ||d d d< d S d S )Nr r .)rW rW r )rT r* r+ rn ro tmp_kwri s r) r) z=TestBinop.test_out_override..OutClass.__array_ufunc__ sg B;; WWYYFJJu%%%"5&11D(,f(?(?(?BuIaL%%% ;r+ Nrp rd r+ r) OutClassr s(
@
@
@
@
@r+ r r rW r r r )r# rE r rm rP r r r )rT r r rt r, s r) test_out_overridezTestBinop.test_out_override s @ @ @ @ @rz @ @ @
HaSMM x ( ( HaSMM HaSMM
Aq! QqT2
1h''(((!
Aqa QqT2
1h''(((((r+ c G d dt j }t j dt j | }t |dz d t
t 5 |dz d d d n # 1 swxY w Y t
t 5 |dz d d d n # 1 swxY w Y t
t 5 |d z d d d n # 1 swxY w Y t
t 5 |d
z d d d n # 1 swxY w Y t
t 5 |d z d d d d S # 1 swxY w Y d S ) Nc e Zd Zd ZdS ):TestBinop.test_pow_override_with_errors..PowerOnlyc 0 |t j urt dS )NPOWER!)r# rp r rm s r) r) zJTestBinop.test_pow_override_with_errors..PowerOnly.__array_ufunc__ s ((-- xr+ Nrp rd r+ r) PowerOnlyr s#
r+ r @r @r r r r3 r5 r2 )r# rE r r rm r r r )rT r rR s r) test_pow_override_with_errorsz'TestBinop.test_pow_override_with_errors s
HRrz * * * / / : :Q#Xx(((
.
/
/
HH
.
/
/
FF
.
/
/
FF
.
/
/
GG
.
/
/
FF sZ 3BB B $B66B:=B:C''C+.C+DD D 7E
EEc G fdd d }t j d d d g }t |dz |d| t |dz |d| t |dz |d| t |d z |d | t |dz |d| d S )
Nc < e Zd ZddZd Zd Z fdZ fdZeZdS )8TestBinop.test_pow_array_object_dtype..SomeClassNc || _ d S r" r r s r) rk zATestBinop.test_pow_array_object_dtype..SomeClass.__init__ s
r+ c t d )Nz __mul__ should not be calledr rn s r) __mul__z@TestBinop.test_pow_array_object_dtype..SomeClass.__mul__ $%CDDDr+ c t d )Nz __div__ should not be calledr rn s r) __div__z@TestBinop.test_pow_array_object_dtype..SomeClass.__div__ r r+ c , | j |z S )Nr r )rT exprq s r) __pow__z@TestBinop.test_pow_array_object_dtype..SomeClass.__pow__ s yTX_5555r+ c H t | r| j |j k S d S r" )r r )rT ro rq s r) r z?TestBinop.test_pow_array_object_dtype..SomeClass.__eq__ s. eY// 1 8uy001 1r+ r" ) ra rb rc rk r r r r __rpow__)rq s r) rq r s
E
E
E
E
E
E
6
6
6
6
6
1
1
1
1
1 HHHr+ rq c D t j fd|D S )Nc g | ]}|z S rd rd )rk r r s r) rl zJTestBinop.test_pow_array_object_dtype..pow_for.. s 3 3 3!Q#X 3 3 3r+ rn )r r' s ` r) pow_forz6TestBinop.test_pow_array_object_dtype..pow_for s( 8 3 3 3 3s 3 3 344 4r+ r3 r2 r r r r5 r )rT r obj_arrrq s @r) test_pow_array_object_dtypez%TestBinop.test_pow_array_object_dtype s ( 5 5 5 (IIaLL))A,, ! EFFW^WWS'%:%:;;;W\771g#6#6777W\771g#6#6777W]GGB$8$8999W\771g#6#677777r+ c G d dt j }t j d | }t t
5 |
d d d d S # 1 swxY w Y d S )Nc e Zd Zd ZdS )2TestBinop.test_pos_array_ufunc_override..Ac B t || d |D i |S )Nc L g | ]!}| t j "S rd )rm r# rE rj s r) rl zVTestBinop.test_pos_array_ufunc_override..A.__array_ufunc__.. s6 0= 0= 0=01 12rz0B0B 0= 0= 0=r+ )r )rT r* r+ rn r s r) r) zBTestBinop.test_pos_array_ufunc_override..A.__array_ufunc__ sH -wuf-- 0= 0=5;0= 0= 0= H@FH H Hr+ Nrp rd r+ r) r r s(
H
H
H
H
Hr+ r r' )r# rE r rm r r) )rT r tsts r) test_pos_array_ufunc_overridez'TestBinop.test_pos_array_ufunc_override s H H H H H
H H H huoo""1%%
9
%
%
DD s A%%A),A))ra rb rc r
rY r r r rh rt r} r r r r rd r+ r) r r k s Jv" v" v"p [ Z$)@AA" " BA"2/ / / .1 .1 .1`) ) )* * 8 8 8B r+ r c > e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d S )
TestTemporaryElidec ddl m} t j d } || \ }}||z t || t |||z d S )Nr ) incref_elide順 )r r r# rJ r
)rT r r origrX s r) test_extension_incref_elidez.TestTemporaryElide.test_extension_incref_elide sg >===== GFOO LOO c A4###3A&&&&&r+ c ddl m} ddddt j d g} || }t |d t j d t ||d |d z d S )Nr )incref_elide_lr3 r r )r r r# rJ r
)rT r r rX s r) !test_extension_incref_elide_stackz4TestTemporaryElide.test_extension_incref_elide_stack
s| @?????
1a ) nQ 1Q4 1113!qt ,,,,,r+ c $ t j dt j }t ||z ddz z j t j d ||z dz }t |j t j d t j ||z d }t |j t j d ||z dz }t |j t j d ||z dz }t |j t j t j t j dt j }t ||z | t j z j t j d | t j }t ||z |z j |j t j dt j }t ||z |z j |j t j
t j t j fD ]\}t j d| }t |dz }t |j t j d |j dz z ]d S )
Ni@
r r2 r+ r: r r zf%d)r# rJ r r r/ rQ r r r r r r rD absr@ )rT r r r r rK rZ s r) test_temporary_with_castz+TestTemporaryElide.test_temporary_with_cast s GF"( + + +q1u3&-rx}}===!eq[QWbhtnn--- NAEA & &QWbhtnn---!er\QWbhtnn---!e\QWbhrx 0 0111
GF"* - - -q1u 4 44;RXd^^LLL
HHRZ q1uk(!'222 GF"- 0 0 0q1uk(!'222 < ? G GBb)))AAG A "(5AJ!O+D"E"E F F F F G Gr+ c t j dt }t j dt }d|z
|z }t |d t |j d d S )N) r3 r r r3 )r r )r# rJ r
rK r rF )rT r r r s r) test_elide_broadcastz'TestTemporaryElide.test_elide_broadcast: sa
GIS ) ) ) GT$ ' ' '
UaKQQWl+++++r+ c t j }t t ||z t j u d S r" )r# r_ r r r s r) test_elide_scalarz$TestTemporaryElide.test_elide_scalarD s4 HJJq1uX"(*+++++r+ c Z t j dt j }|j dz d S )Nr r r2 )r# rA r r r s r) test_elide_scalar_readonlyz-TestTemporaryElide.test_elide_scalar_readonlyI s)
HV2: . . . ! r+ c t j t j t j d d j dz }t |d d S )Nr3 r r5 r )r# rn r{ r9 flatr )rT r s r) test_elide_readonlyz&TestTemporaryElide.test_elide_readonlyR sC Jrrx{{F;;@ A AC GQr+ c t j d d d d }|j dz }~t |d d S )Ni r2 r3 )r# rJ r r r r s r) test_elide_updateifcopyz*TestTemporaryElide.test_elide_updateifcopyW sK GENN33Q3
F "
Qr+ N) ra rb rc r r r r r r r r rd r+ r) r r s
'
'
'
-
-
- G G GB, , ,, , ,
r+ r c 8 e Zd Zd Zej dej ej g d Z
ej dej ej g ej e oe
j j dk d d ZdS ) TestCAPIc ddl m} t |d t |d t |ddz t |d t |d d S ) Nr )IsPythonScalars foobarr3 r2 ra r rR )r r r )rT r s r) test_IsPythonScalarz TestCAPI.test_IsPythonScalar_ s ??????y))***q!!"""q"u%%&&&r""###s##$$$$$r+ converterc |d dk sJ |d dk sJ |t j d dk sJ |d dk sJ |t j dg dk sJ d S ) NrQ r r5 r% r! )r! r? )r? rn rT r s r) test_intp_sequence_convertersz&TestCAPI.test_intp_sequence_convertersg s
y}}%%%%y}}%%%% y# ''61111y 5((((y2$ ((E111111r+ )r r r z PyPy bug in error formattingr c t j t d 5 |t d d d n # 1 swxY w Y t j t d 5 |d d d d n # 1 swxY w Y t j t d 5 |dg d d d n # 1 swxY w Y t j t d 5 |d d d d d S # 1 swxY w Y d S )Nz5expected a sequence of integers or a single integer, r@ z?expected a sequence of integers or a single integer, got '32.0'g @@z2'float' object cannot be interpreted as an integerzMaximum allowed dimension )r rB r) r7 r8 r s r) $test_intp_sequence_converters_errorsz-TestCAPI.test_intp_sequence_converters_errorsu s, ]9MO O O Ifhh ]9#$ $ $
IcNNN ]9JL L L Ise ]:13 3 3
Ie sG A AA& A>>BB$
B==CC# C<<D D N)ra rb rc r r r r _multiarray_testsrun_scalar_intp_converter run_scalar_intp_from_sequencer r r r implementationversionr rd r+ r) r r ^ s % % % [ [ 9 =?@ @ 2 2@ @ 2 [ [ 9 =?@ @ [KC$6$>)$K1 3 3 3 3@ @
r+ r c e Zd Zd ZdS )TestSubscriptingc t j g d }t t |d t j t t |d t j u d S )NrU r r )r# r r r r r rE r< s r) test_test_zero_rankz$TestSubscripting.test_test_zero_rank sX HYYY
1Q4))***QvY2:-.....r+ N)ra rb rc r rd r+ r) r r s# / / / / /r+ r c e Zd Zej ej dk d d Zd Z ej ej dk d d Z
d Z d Z d
Z
d Zd Zd
Zd Zd Zd Zd Zd ZdS ) TestPicklingrW zSthis tests the error messages when trying toprotocol 5 although it is not availabler c . t j d }d S rP r
)rT r s r) $test_correct_protocol5_error_messagez1TestPickling.test_correct_protocol5_error_message s "
r+ c , t }t j |ddfgdt fdt fdt fg }t j dgdt
fdt fdt fg }t
dt j dz D ]}t j t j
|| }t j t j
|| }t |j |j t |j |j d S )
Nr3 r rR r rZ r xxxr3 r r2 r )
r7 r# r r
r r r r r r r r r/ )rT my_objectarr_with_objectarr_without_objectr depickled_arr_with_object depickled_arr_without_objects r) #test_record_array_with_object_dtypez0TestPickling.test_record_array_with_object_dtype s8 HH (Q$%V}sCj3,?A A A X !SzC:U|< > > > 1f 5 9:: = =E(. L 5AAA)C )C %+1<L!3eDDD,F ,F (
.28
:
:
: +15;
=
=
=
= = =r+ r c t j g dg dgd }g }t j |d|j }t | dk sJ t j || }t || d S ) NrU r rY r6 rW )r r r r )r# r r r r r& r r )rT f_contiguous_arrayr r depickled_f_contiguous_arrays r) test_f_contiguous_arrayz$TestPickling.test_f_contiguous_array s Xyyy)))&~~~t |D ]} | J d S )Nr2 r3 r r r r r r rR r rZ r c 6 g | ]}t j | S rd )weakrefrefrk rR s r) rl z/TestPickling.test_roundtrip.. s 111qGKNN111r+ r %rr )
r r r r# r r r r
r r r r r )rT r r DATArefsrR r s r) test_roundtripzTestPickling.test_roundtrip s; 1f 5 9:: % %EX1v1v1v 677F V$$/*C#q?S#J47<3A B B BD 21D111D
*
* 6< Q(G(G(GHH $q* * * * * 4 NNN
%
%suu}}}}
%# % %r+ c . t j |d S )Nlatin1)encoding)r r rT rC s r) _loadszTestPickling._loads s |C(3333r+ c d}t j g dt j }| | }t || d S )Ns} cnumpy.core._internal
_reconstruct
qcnumpy
ndarray
qK UbRq(Kcnumpy
dtype
qUi1K KRq(U|NNJJtbUtb.r r r# r r r r rT r rR r s r) test_version0_int8z TestPickling.test_version0_int8 sI
s H\\\ 1 1 1 KKNNQr+ c d}t j g dt j }| | }t || d S )Ns cnumpy.core._internal
_reconstruct
qcnumpy
ndarray
qK UbRq(Kcnumpy
dtype
qUf4K KRq(U*O P P P KKNNQr+ c L t j dggd }| d }|j d |_ | d t
j t
j | }t |j |j d S ) Nz 2015-02-24T00:00:00.000000000zdatetime64[ns]r r r6 r Tinplace)
r# r rW r/ r r r r r r )rT originaloriginal_byte_reversednews r) test_datetime64_byteorderz&TestPickling.test_datetime64_byteorder s 8 = > ?GWXXX!) S !9!9 '='C'P'PQT'U'U $ '''555l6<(>??@@X^SY/////r+ N)ra rb rc r r r r r r r r r r r r r r r r
r r r rd r+ r) r r sW [/14!J L L L L = = =* [/!3 ; = =G G= =G O O O% % %*4 4 4
0 0 0 0 0r+ r c 2 e Zd Zd Zd Zd Zd Zd Zd ZdS )TestFancyIndexingc t j d }d|d d dgf< t |t j dgg t j d }d|d d d d dgf< t |t j dggg d S )Nr r r rv r# rJ r
r r< s r) test_listzTestFancyIndexing.test_list s GFOO!!!aS& 1bhw / /000 GI !!!QQQ) 1bh#y 1 122222r+ c t j d }d|d d df< t |t j dgg t j d }d|d d d d df< t |t j dggg d S )Nr r r rv r r< s r)
test_tuplez TestFancyIndexing.test_tuple( s GFOO!!!T'
1bhw / /000 GI !!!QQQ*
1bh#y 1 122222r+ c t j g d }t j g dt }t || t j dg d S )Nr r r3 r r r2 r# r rK r
rT r r s r) test_maskzTestFancyIndexing.test_mask0 sN H\\\ " " H\\\4 ( (1Q41# /////r+ c t j g dg dg }t j ddgt }t j g dg dgt }t j g dg dgt }t || t j g dg t || t j dd g t || t j dg d S )
Nr r r r3 r r3 r r r r r2 rW r )rT r r m2m3s r)
test_mask2z TestFancyIndexing.test_mask25 s HlllLLL1 2 2 HaVT " "
X|||\\\2D
9
9
X|||\\\2D
9
91Q4<<<.!9!9:::1R5"(Aq6"2"23331R5"(A3--00000r+ c t j g d }t j g dt }d||< t |t j g d d S )Nr r rW )r3 rW r r r r s r) test_assign_maskz"TestFancyIndexing.test_assign_mask> sW H\\\ " " H\\\4 ( (!1bh||| 4 455555r+ c x t j g dg dg }t j ddgt }t j g dg dgt }t j g dg dgt }| }d||< t |t j g dg d g | }d||< t |t j g d
g d g | }d||< t |t j g d
g dg d S ) Nr r r r3 r r# r rQ )rQ rQ rQ rQ )r3 rQ r r )rQ r r r )r# r rK rW r
)rT xorigr r$ r% r s r) test_assign_mask2z#TestFancyIndexing.test_assign_mask2D s/ ,,, 566 HaVT " "
X|||\\\2D
9
9
X|||\\\2D
9
9 JJLL!1bh 6F6F6F'G H HIII JJLL"1bh
}}}'E F FGGG JJLL"1bh
|||'D E EFFFFFr+ N) ra rb rc r r r! r& r( r+ rd r+ r) r r ss 3 3 33 3 30 0 0
1 1 16 6 6
G
G
G
G
Gr+ r c e Zd Zd Zd Zd ZdS )TestStringComparec t j g d t j g d t k fddD t k fddD t k fddD t k fddD t k fddD t k fd dD d S )
NThisisexampler0 wasr2 c 4 g | ]}| | k S rd rd rk r g1g2s r) rl z1TestStringCompare.test_string..X % %H%H%Hber!un%H%H%Hr+ rt c 4 g | ]}| | k S rd rd r6 s r) rl z1TestStringCompare.test_string..Y r9 r+ c 4 g | ]}| | k S rd rd r6 s r) rl z1TestStringCompare.test_string..Z r9 r+ c 4 g | ]}| | k S rd rd r6 s r) rl z1TestStringCompare.test_string..[ r9 r+ c 4 g | ]}| | k S rd rd r6 s r) rl z1TestStringCompare.test_string..\ % $F$F$FqRURU]$F$F$Fr+ c 4 g | ]}| | k S rd rd r6 s r) rl z1TestStringCompare.test_string..] r> r+ r[ rT r7 r8 s @@r) test_stringz TestStringCompare.test_stringU s>
X///
0
0
X000
1
128%H%H%H%H%Hi%H%H%HIII28%H%H%H%H%Hi%H%H%HIII28%H%H%H%H%Hi%H%H%HIII28%H%H%H%H%Hi%H%H%HIII27$F$F$F$F$FI$F$F$FGGG27$F$F$F$F$FI$F$F$FGGGGGr+ c t j g d }dt |k fd|D t |k fd|D t |k fd|D t |k fd|D t |k fd|D t |k fd|D d S ) N)spamspaspammerzand eggsrC c g | ]}|k S rd rd rk r r8 s r) rl z0TestStringCompare.test_mixed..b %:%:%:!a2g%:%:%:r+ c g | ]}|k S rd rd rG s r) rl z0TestStringCompare.test_mixed..c rH r+ c g | ]}|k S rd rd rG s r) rl z0TestStringCompare.test_mixed..d $8$8$8QV$8$8$8r+ c g | ]}|k S rd rd rG s r) rl z0TestStringCompare.test_mixed..e rK r+ c g | ]}|k S rd rd rG s r) rl z0TestStringCompare.test_mixed..f rH r+ c g | ]}|k S rd rd rG s r) rl z0TestStringCompare.test_mixed..g rH r+ r[ r@ s @r)
test_mixedz TestStringCompare.test_mixed_ s
X<<<
=
=
28%:%:%:%:r%:%:%:;;;28%:%:%:%:r%:%:%:;;;27$8$8$8$8R$8$8$899927$8$8$8$8R$8$8$899928%:%:%:%:r%:%:%:;;;28%:%:%:%:r%:%:%:;;;;;r+ c t j g d t j g d t k fddD t k fddD t k fddD t k fddD t k fddD t k fd dD d S )
Nr/ r3 c 4 g | ]}| | k S rd rd r6 s r) rl z2TestStringCompare.test_unicode..l r9 r+ rt c 4 g | ]}| | k S rd rd r6 s r) rl z2TestStringCompare.test_unicode..m r9 r+ c 4 g | ]}| | k S rd rd r6 s r) rl z2TestStringCompare.test_unicode..n r9 r+ c 4 g | ]}| | k S rd rd r6 s r) rl z2TestStringCompare.test_unicode..o r9 r+ c 4 g | ]}| | k S rd rd r6 s r) rl z2TestStringCompare.test_unicode..p % %G%G%Gbebem%G%G%Gr+ c 4 g | ]}| | k S rd rd r6 s r) rl z2TestStringCompare.test_unicode..q rV r+ r[ r@ s @@r) test_unicodez TestStringCompare.test_unicodei s>
X///
0
0
X000
1
128%H%H%H%H%Hi%H%H%HIII28%H%H%H%H%Hi%H%H%HIII28%H%H%H%H%Hi%H%H%HIII28%H%H%H%H%Hi%H%H%HIII27%G%G%G%G%GY%G%G%GHHH27%G%G%G%G%GY%G%G%GHHHHHr+ N)ra rb rc rA rO rX rd r+ r) r- r- T sI H H H< < <I I I I Ir+ r- c l e Zd Zg dZej d ej d eD ej de j
e j g d Z ej dddg d Z
ej dd d
g d Zej d d
dg ej dd d
g d Zej de j d ddfe j d
ddfg d Zej dd e j
fd
e j fg d Zej d ej ddd g d Zd S ) TestArgmaxArgminCommon)rd rR r r2 r rV r2 r r )r r r2 r )r2 r r r3 )r r r3 r2 )r r3 r2 r )r )r ru z
size, axisc g | ]Jfd t t t t dgz D KS )c g | ]}|fS rd rd )rk r rH s r) rl z,TestArgmaxArgminCommon..{ s5 >A >A >A @DTl >A >A >Ar+ N)r r r& )rk rH s @r) rl z!TestArgmaxArgminCommon.{ sx = = = >A >A >A >AD z3t995566$?>A >A >A = = =r+ r+ c P t j | }|'d t t | D }nt | }d||< t
| } ||| }| | } |||d }t || t |j
|k t j |j
|j } |||| d }
t |
| u t || t | d k r|t | } |d
| |< nd
| d < t j | |j } t j t 5 ||j || d d d d n # 1 swxY w Y |'d t t | D }n t | d d d }d||< t
| } ||j | }| | } ||j |d }t || t |j
|k t j |d d d |j } | j } ||j || d }
t |
| u t || t | d k rFt j t 5 ||d || d d d d n # 1 swxY w Y t | d k rt | } |d
| |< nd
| d < t j | |j } t j t 5 ||j || d d d d d S # 1 swxY w Y d S d S )
NrH c g | ]}d S r rd rk r s r) rl zITestArgmaxArgminCommon.test_np_argmin_argmax_keepdims.. 555q555r+ r3 r Tr keepdimsr )r r rf r r2 c g | ]}d S r rd rc s r) rl zITestArgmaxArgminCommon.test_np_argmin_argmax_keepdims.. rd r+ r5 )r# rw normalr r& r r> r r r rF rA r/ r rB r8 rL )
rT rH r r+ r' new_shape _res_origres_origrX outarrayres1 wrong_shapewrong_outarrays
r) test_np_argmin_argmax_keepdimsz5TestArgmaxArgminCommon.test_np_argmin_argmax_keepdims{ s i D )) <55E#d))$4$4555II T
I IdO)$$ F3T*** $$Y//fStd333S(### Y&'''8CISY777vc(%)+ + + !!!S(### t99q== y//K $% D!!!" A Xk C C CN z**
;
;su4 *T; ; ; ;
;
;
;
;
;
;
;
;
;
;
;
;
;
;
;
<55E#d))$4$4555II T
44R4(I IdO)$$ F35t,,, $$Y//fSU555S(### Y&'''8IdddO39===:vce$H%)+ + + !!!S(### t99q== z**
5
5s1vD $t5 5 5 5
5
5
5
5
5
5
5
5
5
5
5
5
5
5
5 t99q== y//K $% D!!!" A Xk C C CN z**
;
;su4 *T; ; ; ;
;
;
;
;
;
;
;
;
;
;
;
;
;
;
;
;
;
; =s6 8FF F 'L L
L
7NN N r minc t j ddd }t |d|z }t || }t |j D ]} || } || }t
t |j }| | t t j | |j
|j |g|R k d S )Nr r3 )r rW r r r r( ) r# rw rh r r r r remover r r r ) rT r+ rR
arg_method
val_methodr a_maxmin aarg_maxminaxess r) test_allz TestArgmaxArgminCommon.test_all s I Q? 3 3 Q//
Q''
qv B BA!z!}}H$*Q--Kaf
&&D KKNNN BF8'9{'9)4Q)>)>)>)>(@ @ A A
B
B
B
B B Br+ argmaxargminc $ t j d }t || }t j dt j }t t
|d| t j dt j }t t
|d| t j dt j }t t
|d| t j dt j } |d| t
| |d d S ) N)rQ rW r? r r5 r) r3 rQ rQ r )r# rJ r r r r8 r )rT r+ rR rt r s r) test_output_shapez(TestArgmaxArgminCommon.test_output_shape s
GG Q''
gb(((j*b#666gfBG,,,j*b#666 ggRW---j*b#666gb(((
23 S**R..)))))r+ r r r3 c t j dd|z z }t || }t j d|z t j } |d| }||u sJ d S )Nr r] r r r r )r# rJ r rA r6 )rT r r+ rR rt r r s r) test_ret_is_outz&TestArgmaxArgminCommon.test_ret_is_out sj
GD6$;& ' ' Q''
hvd{"'222jaS)))czzzzzzr+ znp_array, method, idx, vali6 asiq 0c z |dd }t || }|||< t | | d S )Ni z>F; F; ?> F;P [ Xu~66
B
B 76
B [ X(';<<* * =<*& [ VaV,, [ X(';<< =< -, [ 9
(HdD )
'8T3 ' )* *( (* *( [ 4
BI
BI ! !! !! !!& [!!)C!DD [ ^ & & ED & & &r+ rZ c
e Zd Zg ddfg ddfg ddfg ddfgZeg ddfg d dfgz Zd
ej eej ej ej
ej f D Z e d ej eej
ej ej ej ej ej f D z Z e d ej dd
ddej gdfdd
dej dgdfej dd
ddgdfej dej ddgdfd
gdz ej gz dfd
gdz ej gz dfd
gdz ej gz dfd
gdz ej gz dfd
gdz ej gz dff ej ej f D z Z e dd
dd edej gdfdd
dd eej d gdfdd
d eej d dgdfdd
d edej dgdf edej dd
ddgdf eej ej dd
ddgdf eej d eej d eej d
gdf eej ej eej d eej d
gdf eej d eej d eej ej gdf edd edd edd
gd
f ed
d edd edd
gdf ed
d edd ed
d
gdf ej d ej d ej d ej d ej d ej d gdf ej d ej d ej d ej d ej d ej d! gdf ej d" ej d" ej d ej d" ej d# ej d$ gdf ej d% ej d& ej d" ej d' ej d( ej d) gdf ej dd* ej d
d* ej d"d* ej dd* gdf ej d"d* gdz df edd+, edd-, ed.d/, gdf ed
d0, ed1d, edd+, gd
f ed1d2, ed1d, ed1d0, gdfg d3dfg d4dfg d5dfg d6dfgz Zej d7e d8 Z d9 Z d:S );
TestArgmaxr3 r3 r3 r3 r3 r3 r3 r3 r r r r r r r r r r r r r2 r2 r2 r2 r r3 r2 r r rW r r r r r rW r r r2 r3 r r3 r2 r r r r r r5 r r3 r2 r r r5 r r r c \ g | ])\ }}t j |d | |d f*S r r r3 rn rk r r. s r) rl zTestArgmax.* D 1RXad!
$
$
$ad + r+ c \ g | ])\ }}t j |d | |d f*S r rn r s r) rl zTestArgmax./ D tq!BHQqT+++QqT2 r+ c \ g | ])\ }}t j |d | |d f*S r rn r s r) rl zTestArgmax.4 D tq!BHQqT+++QqT2 r+ r3 r2 r r r( ' O 1923-04-14T12:43:121994-06-21T14:43:152001-10-15T04:10:321995-11-25T16:02:162005-01-04T03:14:122041-12-03T14:05:03rW 1935-09-14T04:40:111949-10-12T12:32:112010-01-03T05:14:12z2015-11-20T12:20:59z1932-09-23T10:10:13z2014-10-10T03:50:30NaT2015-09-23T10:10:131932-10-10T03:50:302059-03-14T12:43:121996-09-21T14:43:152022-12-25T16:02:161963-10-04T03:14:122013-05-08T18:15:23r r daysseconds# r5 r rQ r )FFFFT)FFFTF)TFFFF)TFTFFr4 c d |\ }}t 5 }| t d t j | }d d d n # 1 swxY w Y t t j | |d|z t |t j | |d|z t j |d }|dz }t t j | |d|z t |t j | |d|z t j t j | d }t j ||f }|}t t j | |d|z t |t j | |d|z d S Nz#invalid value encountered in reducer r i )
r filterRuntimeWarningr# r r rz r rq rJ ) rT r4 r' possupr rarrrpospadds r) test_combinationsz TestArgmax.test_combinations s S
C JJ~=
?
?
?&++C
RYs^^S$*====S3 (#tczBBBB yc""SyRYt__dD4K@@@@T")D//*CEEEEyc**~sDk**RYt__dD4K@@@@T")D//*CEEEEEE 0AAAc t j g dt j }t t j | d | d }t t j | d t j g dt j }t t j | d | d }t t j | d t j g dt j }t t j | d | d }t t j | d t j g dt j }t t j | d | d }t t j | d d S )N)r3 r r3 r )r3 )r3 r7 )r3 r9 ) r# r r r rz r r r r r s r) test_maximum_signed_integersz'TestArgmax.test_maximum_signed_integers sx H))) 9 9 9RYq\\1%%%
HHSMMRYq\\3''' H+++28 < < <RYq\\1%%%
HHSMMRYq\\3''' H+++28 < < <RYq\\1%%%
HHSMMRYq\\3''' H+++28 < < <RYq\\1%%%
HHSMMRYq\\3'''''r+ N) ra rb rc usg_datasg_datar productr# rB r r r darrr r r r r r r1 rb
datetime64 timedelta64r nan_arrr r r r r rd r+ r) r r s 6 6 6: % % %q) % % %q) % % %q) H % % %q) % % %q) G ( HbiBI%
D
' GRXrx2:rz$
D
Aq"&
!1 %Arvq
!1 %faAq
!1 %faA
& * SE]bfX
%u -SE]bfX
%u -SE]bfX
%u -SF^rvh
& /SF^rvh
& /
J
D"
Q1gga(( )1-
Q1ggbfa(( )1-
Q77261%%q )1-
Q771bf%%q )1-
'!RV
aAq )1-
'"&"&
!
!1aA .2
'"&!
ggbfa 0 0''"&!2D2D EqI
'"&"&
!
!77261#5#5wwrvq7I7I JAN
'"&!
ggbfa 0 0''"&"&2I2I JAN
'!Q--A 1
6:
'!Q--A 1
6:
'!Q--A 1
6:
"--
.
.
"--
.
.
"--
.
.
"--
.
.
"--
.
.
"--
.
.
0
23 4 "--
.
.
"--
.
.
"--
.
.
"--
.
.
"--
.
.
"--
.
.
0
23 4 "-
"-
"--
.
.
"-
"--
.
.
"--
.
.
0
23 4 "--
.
.
"--
.
.
"-
"--
.
.
"--
.
.
"--
.
.
0
23 4 ".C
".C
".
$
$
".C
" $% & ".
$
$ % )1-
)B
'
'
'2)F)F)F
)R
(
(
(
*+, .
)B
'
'
'A)F)F)F
)B
'
'
'
)*+ -
)R
(
(
())Q*G*G*G
)R
(
(
(
*+, .
, + +Q/ + + +Q/ + + +Q/ * * *A.q9 9Gv [ VW--F F .-F.( ( ( ( (r+ r c
e Zd Zg ddfg ddfg ddfg ddfgZeg d dfg d
dfgz Zd ej eej ej ej
ej f D Z e d ej eej
ej ej ej ej ej f D z Z e d
ej ddddej gdfdddej dgdfej ddddgdfej dej ddgdfdgdz ej gz dfdgdz ej gz dfdgdz ej gz dfdgdz ej gz dfdgdz ej gz dff ej ej f D z Z e dddd edej gdfdddd eej d gdfddd eej d dgdfddd edej dgdf edej ddddgdf eej ej ddddgdf eej d eej d eej d gdf eej ej eej d eej d gdf eej d eej d eej ej gdf edd edd edd gdf edd edd edd gdf edd edd edd gdf ej d ej d ej d ej d ej d ej d gdf ej d ej d ej d ej d ej d ej d! gd"f ej d# ej d# ej d ej d# ej d ej d! gdf ej d$ ej d% ej d# ej d& ej d' ej d( gdf ej dd) ej dd) ej d#d) ej dd) gdf ej d#d) gdz df ed"d*+ edd,+ ed-d.+ gdf edd/+ ed0d"+ ed"d*+ gdf ed0d1+ ed0d"+ ed0d/+ gdfg d2dfg d3dfg d4dfg d5dfgz Zej d6e d7 Z d8 Z d9S ):
TestArgminr r r r r r r r r r c \ g | ])\ }}t j |d | |d f*S r rn r s r) rl zTestArgmin. r r+ c \ g | ])\ }}t j |d | |d f*S r rn r s r) rl zTestArgmin. r r+ c \ g | ])\ }}t j |d | |d f*S r rn r s r) rl zTestArgmin. r r+ r3 r2 r r r( r r r r r r r r r r r r z2014-11-20T12:20:59r r rW r r r r r r r r r r r5 r r rQ r )TTTTF)TTTFT)FTTTTFTFTTr4 c d |\ }}t 5 }| t d t j | }d d d n # 1 swxY w Y t t j | |d|z t |t j | |d|z t j |d }|dz }t t j | |d|z t |t j | |d|z t j t j | d }t j ||f }|}t t j | |d|z t |t j | |d|z d S r )
r r r r# rq r r{ r r rJ ) rT r4 r' r r min_valr r r s r) r z TestArgmin.test_combinations s S
"C JJ~<
>
>
>fSkkG " " " " " " " " " " " " " " "
RYs^^S$*====S3 ('4#:FFFF yc""SyRYt__dD4K@@@@T")D//*GTD[IIIIyc**~sDk**RYt__dD4K@@@@T")D//*GTD[IIIIIIr c t j g dt j }t t j | d | d }t t j | d t j g dt j }t t j | d | d }t t j | d t j g dt j }t t j | d | d }t t j | d t j g dt j }t t j | d | d }t t j | d d S )N)r3 r ir r r3 r )r3 r ir )r3 r i r7 )r3 r l r9 ) r# r r r r{ r r r r r s r) test_minimum_signed_integersz'TestArgmin.test_minimum_signed_integers& sx H444BG D D DRYq\\1%%%
HHSMMRYq\\3''' H777rx H H HRYq\\1%%%
HHSMMRYq\\3''' H777rx H H HRYq\\1%%%
HHSMMRYq\\3''' H777rx H H HRYq\\1%%%
HHSMMRYq\\3'''''r+ N) ra rb rc r r r r r# rB r r r r r r r r r r r1 rb r r r r r r r r r rd r+ r) r r s 6 6 6: % % %q) % % %q) % % %q) H % % %q) % % %q) G ( HbiBI%
D
' GRXrx2:rz$
D
Aq"&
!1 %Arvq
!1 %faAq
!1 %faA
& * SE]bfX
%u -SE]bfX
%u -SE]bfX
%u -SF^rvh
& /SF^rvh
& /
J
D"
Q1gga(( )1-
Q1ggbfa(( )1-
Q77261%%q )1-
Q771bf%%q )1-
'!RV
aAq )1-
'"&"&
!
!1aA .2
'"&!
ggbfa 0 0''"&!2D2D EqI
'"&"&
!
!77261#5#5wwrvq7I7I JAN
'"&!
ggbfa 0 0''"&"&2I2I JAN
'!Q--A 1
6:
'!Q--A 1
6:
'!Q--A 1
6:
"--
.
.
"--
.
.
"--
.
.
"--
.
.
"--
.
.
"--
.
.
0
23 4 "--
.
.
"--
.
.
"--
.
.
"--
.
.
"--
.
.
"--
.
.
0
23 4 "-
"-
"--
.
.
"-
"--
.
.
"--
.
.
0
23 4 "--
.
.
"--
.
.
"-
"--
.
.
"--
.
.
"--
.
.
0
23 4 ".C
".C
".
$
$
".C
" $% & ".
$
$ % )1-
)B
'
'
'2)F)F)F
)R
(
(
(
*+, .
)B
'
'
'A)F)F)F
)B
'
'
'
)*+ -
)R
(
(
())Q*G*G*G
)R
(
(
(
*+, .
) ( (!, ( ( (!, ( ( (!, ) ) )1-q9 9Gv [ VW--J J .-J,( ( ( ( (r+ r c e Zd Zd Zd Zd ZdS )
TestMinMaxc t t j t j dd t t j t j dd t t j dd d t t j dd d t t j dd d t t j dd d d S )Nr3 r r )r r# r/ amaxaminr rS s r) test_scalarzTestMinMax.test_scalar> s blBGQ222blBGQ222RWQQ'''+++RWQQ'''+++RWQT***A...RWQT***A.....r+ c t t j t j g dd t t j g dgd d d S )NrU r r3 r r )r r# r/ r r rS s r) test_axiszTestMinMax.test_axisG sI blBGYYY===RWiii[q111155555r+ c dD ]}t j d | }t t j | |d t t j | |d d|d< t t j | |d t t j | |d d S )N)m8[s]zm8[Y]rQ r r r r )r# r% r r r r )rT r/ rR s r)
test_datetimezTestMinMax.test_datetimeK s ' + +E "
$$U++A QqT * * * QqT * * *AaD QqT * * * QqT * * * *
+ +r+ N)ra rb rc r r r rd r+ r) r r < sA / / /6 6 6+ + + + +r+ r c e Zd Zd ZdS ) TestNewaxisc t j g d }d|d d t j f z }t | d|z d S )N)r gg? )r# r r r rh )rT skrX s r)
test_basiczTestNewaxis.test_basicW sP
Xnnn
%
%"QQQ
]##CIIKKR00000r+ Nra rb rc r rd r+ r) r r V s# 1 1 1 1 1r+ r c 8 e Zd Zd Z d dZd Zd Zd Zd ZdS )
TestClipc t t j ||k t t j ||k d S r" )r r# r )rT r cmincmaxs r) _check_rangezTestClip._check_range^ s@ qDy!!"""qDy!!"""""r+ FNc " ||}||}t j | D ]}t j dk rddg} nddg} | D ]}
t j | |
} t j d |z | } |r| ||| d n| || } d}
| j j dk rd}
t | j j |
|
| || | S ) Nr =>
>
> OOq!W
6
6
6
OOtT3
9
9
9 OOtQ7
4
4
4
OOaG
5
5
5 OOdCq
J
J
J
J J Jr+ c t j ddgg d }|d dd }| |dd d S )Nr r r r r r )r r )r/ r zr r r g333333ӿr )r# r r r )rT r r/ s r) test_record_arrayzTestClip.test_record_array s` h8 G G GI I I HMM$ $ $ !T3'''''r+ c t j g d }| d }t t j |dk | d }t t j |dk | d }t t j |dk d S )Nr r )rq r )r )r# r r r r )rT r r s r) test_max_or_minzTestClip.test_max_or_min s h///00 HHQKKqAv HHHOOqAv HHHOOqAv r+ c
t j dt j dddt j g }| dd }t j dt j dddt j g }t || d S ) N r r r r5 r3 r6 )r# r r1 r r
)rT input_arrr$ r s r) test_nanzTestClip.test_nan sk Hc263D"& ABB A&&8S"&#r4 @AA68,,,,,r+ )FNN) ra rb rc r r r r r r rd r+ r) r r ] s # # #
0537 DJ J J"( ( ( - - - - -r+ r c e Zd Zd Zd Zd ZdS ) TestCompressc g dg}t j d dd }t j ddg|d }t || ddgd d
gg}t j g d |d }t || d S r r r s r) r zTestCompress.test_axis s imm##Aq))k1a&#A...S# 1v1v k///3Q777S# r+ c dgdgg}t j d dd }t j ddg|d }t || d S )Nr3 r rQ r2 rW r r r r s r)
test_truncatezTestCompress.test_truncate s\ sQCjimm##Aq))k1a&#A...S# r+ c t j d dd }t j ddg| }t |d d S )NrQ r2 rW r r3 r rT r' r s r) rV zTestCompress.test_flatten sJ imm##Aq))k1a&#&&S! r+ N)ra rb rc r r rV rd r+ r) r r sA r+ r c e Zd Zd Zd Zd Zej dd d Z d Z
d Z d Z d
Z
d S ) TestPutmaskc t j ||| t || t j || d S r" )r# putmaskr r )rT r rL maskr s r) tst_basiczTestPutmask.tst_basic s<
1dC QtWbhsA . ./////r+ c t t t j g}t j d dz }|dk }dD ]}t j D ]}|D ]}||vr||dk r9t j | j dk r t j | j
dz
}| |
| ||| t j d }| |
| |j || | d d
d S )
Nr r] r )r r! r r c S3r )r r r# r rw r r r/ r r; r r$ rW r r )rT unchecked_typesr r# r r rL rK s r)
test_ip_typeszTestPutmask.test_ip_types s9 #rw/ I T " "3 &2v
J
JC**,,
I
I I IA// 77rx{{'73'>'>"$(1++/B"6C qvvxxq'9'91dCHHH I $ B NN188B<<$ RaR8H I I I I
J
Jr+ c t t t t j t j g d dgd d S )NrU TrW )r r8 r# r" r rS s r) test_mask_sizezTestPutmask.test_mask_size s0 j"*bhyyy.A.AD61MMMMMr+ r/ r r c t j g d| }t j |g dd t |g d d S )NrU r r5 )r5 r2 r5 r# r r" r
rT r/ r s r) test_byteorderzTestPutmask.test_byteorder sL HYYY & &
1)))2...1kkk*****r+ c t j ddgg d }t j |d ddgd t |d dd g t |d
d d g t |d
ddg t j |d
ddgd t |d dd g t |d
dd g t |d
ddg d S )Nr r r
)r/ r r r r TFrQ rW r/ r2 r r r r? r/ )rT r s r) r z TestPutmask.test_record_array s h8 F F FH H H
3s8dE]B///3s8b!W---3s8aV,,,3s8aV,,,
3s8dE]B///3s8b!W---3s8b!W---3s8aV,,,,,r+ c t j g d }t j |dd g d|d d t |t j g d t j g d }t j |dd |d d g d t |t j g d d S ) NTFTFr3 r r r )TTFTr )TTTT)r# r r" r r< s r)
test_overlapszTestPutmask.test_overlaps s H/// 0 0
1QqS6---q!u555Q!:!:!:;;<<< H/// 0 0
1QqS61RaR5"5"5"5666Q!9!9!9::;;;;;r+ c t j d }d|j _ t j t 5 t j ||dk d d d d d S # 1 swxY w Y d S )NrW Fr2 r )r# r% rY rZ r rB r8 r" r s r) r\ zTestPutmask.test_writeable s IaLL !
]:
&
& % % Jq!q&! $ $ $ % % % % % % % % % % % % % % % % % %s A""A&)A&c 0 t j ddg }t j |ddgddg t |ddg t j ddg }t j |ddgddg t |ddg t j ddg }t j |ddgddg t |ddg t j t 5 t j |ddgddg d d d d S # 1 swxY w Y d S )Nr r3 r5 r )r# r )r r# )rR r r# )r# r r" r
r rB r) r< s r) test_kwargszTestPutmask.test_kwargs sd HaV
1q!fr2h'''1q"g&&& HaV
1Aq62r(33331q"g&&& HaV
1b"XaV44441q"g&&&
]9
%
% ; ; JB8Aq6 : : : : ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;s " D DDN)ra rb rc r$ r* r, r r r r1 r r6 r\ r9 rd r+ r) r r s 0 0 0J J J$N N N [ Wn55+ + 65+
- - -< < <% % %; ; ; ; ;r+ r c e Zd Zd Zd Zd Zd Zd Zej
dd d Z d Z d
Z
ej
d g d d
ZdS )TestTakec t t |j d }t | |d | d S )Nr r )r r rF r
take)rT r r} s r) r$ zTestTake.tst_basic sC 5$$%%166#A6..22222r+ c t t t j g}t j d dz }d|_ t j D ]m}|D ]@}||vr:| |
| A| |
d nd S )Nr r] r\ r( ) r r r# r rw rF r r r$ rW r )rT r) r r rL s r) r* zTestTake.test_ip_types s #rw/ I R $Z&&(( + +E
7
7O++NN16688??1#5#5666
NN188D>> * * * *
+ +r+ c 6 t j d dz }d|_ t t |j g dd t t |j dgd t
| dgd d |d d S )
Nr r] r\ rt r r r r5 r3 )r# rw rF r r r= r
r< s r)
test_raisezTestTake.test_raise# s I R $j!&)))!<<<<j!&2$Q77771662$Q6//2AaD99999r+ c $ t j d dz }d|_ t | dgdd d |d t | dgdd d |d d S )
Nr r] r\ r5 r r r r r2 r3 r# rw rF r
r= r< s r) test_clipzTestTake.test_clip* s I R $1662$QV6<!EEEEEr+ c t j d dz }d|_ t | dgdd d |d t | d gdd d |d t | d
gdd d |d d S ) Nr r] r\ r5 r r rB r3 r2 r rC r< s r) test_wrapzTestTake.test_wrap0 s I R $1662$QV6<!EEE1661#AF6;;A>!EEEEEr+ r/ r- c t j g d| }t | g d g d d S )NrU )r r2 r3 r )r# r r
r= r0 s r) r1 zTestTake.test_byteorder7 sA HYYY & &166))),,iii88888r+ c t j ddgg d }| dg }t |d dk o |d d k d S )
Nr r r3 r r3 r r r/ r )r# r r= r )rT r rec1s r) r zTestTake.test_record_array< sg h8 F F FH H Hxx}}S S 5T#Y#%566666r+ c t j d }t j |g d|dd d }t |t j g d d S )NrW rU r2 r r )r# r% r= r r r s r) test_out_overlapzTestTake.test_out_overlapC sW IaLL GAyyya!f6 : : :Q++,,,,,r+ rF )r r rd c t j d }t j |t j }t j ||j }t j ||| }||u sJ d S )NrW r r )r# r% r9 r6 r/ r= )rT rF r indsr r s r) r zTestTake.test_ret_is_outI sd
IaLLxRW---huAG,,,ga3'''czzzzzzr+ N)ra rb rc r$ r* r@ rD rF r r r r1 r rK r rd r+ r) r; r; s 3 3 3 + + +: : :F F F F F F [ Wn559 9 6597 7 7- - - [ W&8&8&899 :9 r+ r; c e Zd Zej dej ej ej ej
ej ej ej
ej ej ej ej g d Zd Zd Zd Zd ZdS ) TestLexsortr/ c . t j g d| }t j g d| }t j ||f }t j g d }t || t || t j | d S )Nr3 r2 r3 r r3 rW r r r rW r r2 r r r r2 r3 r rW )r# r lexsortr
r$ )rT r/ rR r r- expected_idxs r) r zTestLexsort.test_basicT s
H'''u 5 5 5 H'''u 5 5 5j!Q x 2 2 233 3 ---1S6271::.....r+ c t j g d }t j g dd }t j ||f }t j g d }t || d S )NrQ rR ra r rS r# r rT r
rT rR r r- rU s r) rO zTestLexsort.test_mixeda sr H''' ( ( H''' ? ? ?j!Q x 2 2 233 3 -----r+ c t j g dd }t j g dd }t j ||f }t j g d }t || t j g dd }t j g dd }t j ||f }t j g d }t || d S )Nru ra r rx rb rW rX s r) r zTestLexsort.test_datetimei s HWWWO 4 4 4 HWWWO 4 4 4j!Q x ** 3 --- HWWW$4 5 5 5 HWWW$4 5 5 5j!Q x ** 3 -----r+ c
t j dd }t j g dd }||fD ]P}t j | d f }t j |d }t
|| Q||f||ffD ]\ }}t j ||f }t
|t j | d |f t
|t j || d f t j |d t j |d }}t
|t j ||f d S ) NrQ r )abcxywzefghiqwstr r+ mergesortr4 r7 r )r# rw choicerT r r r
r )rT rR r r r r r r- s r) test_objectzTestLexsort.test_objectv sc I R & & I F F F M MA , ,A:qxx}} .//DJq{333E tU + + + +FQFN 8 8DAq*aV$$C sBJ
q/A$B$B C C C sBJ188C==/A$B$B C C C8AX...(0K0K0KqA sBJ1v$6$6 7 7 7 7 8 8r+ c t j ddd dd }t t j t j |d d S )Nr5 r6 ~ r r r2 r )r# r r r r/ rT r< s r) test_invalid_axisz TestLexsort.test_invalid_axis sF KB % % - -b! 4 4blBJ::::::r+ N)ra rb rc r r r r# rB r r r r r r r r r r r rO r rb re rd r+ r) rO rO S s [ W
")RY
28RX
BJ
&
/ /
/. . . . . .8 8 8 ; ; ; ; ;r+ rO c e Zd ZdZ ej d Z ej ddg d Zd Zd Z d Z
d
Z d Z d Z
d
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Z ej e d d Z d Z! ej d d!g d" Z"d# Z#d$ Z$d% Z%d& Z&d' Z'd( Z(d) Z)d* Z*d+ Z+d, Z,d- Z-d. Z.d/ Z/d0 Z0d1 Z1d2 Z2d3 Z3d4 Z4d5 Z5d6S )7TestIOz.Test tofile, fromfile, tobytes, and fromstringc d}t j j } || || t dz z }t j t j t j t j g|dd d df< |S )N)r2 r r r r r3 )r# rw r rb r1 inf)rT rF ry r s r) r zTestIO.x si y DKK$$u++,,W55: :fbfrvgrv6!QQQ'
r+ stringpath_obj)paramsc # P K |dz }|j dk rt | }|V d S )Nfilerj )r r )rT tmp_pathrequestfilenames r) tmp_filenamezTestIO.tmp_filename s7
f$ =H $ $8}}Hr+ c t j t t t j t j d t j d }t t fd| d S )Nra r c . | S r" tofile)r r s r) r z$TestIO.test_nofile.. s ! r+ )r r r OSErrorr# fromfilerB rJ )rT r r s @r) test_nofilezTestIO.test_nofile sY
JLLgr{Arx<<< GAJJg 4 4 4 4a88888r+ c t j g dt j }t j ddt j }t || d S )Nr5 r z 1 0 -2.3 0.0 sepr/ )r# r r_ r r
)rT r r/ s r) test_bool_fromstringzTestIO.test_bool_fromstring sM H///rx @ @ @ M.c B B B1a r+ c t j dt j d }t j ddgt j }t || d S )Nz#9923372036854775807 104783749223640r{ r/ r} l *M M<3} r )r# r r r r
r\ s r) test_uint64_fromstringz TestIO.test_uint64_fromstring sT M? " s
4
4
4 H)?;29 M M M1a r+ c t j dt j d }t j ddgt j }t || d S )Nz -25041670086757 104783749223640r{ r le o[r r )r# r r r r
r\ s r) test_int64_fromstringz TestIO.test_int64_fromstring sS M; "c
3
3
3 Ho7rx H H H1a r+ c d t j ddt j d }|j dk sJ d S )Nz1,2,r )r} r/ countr )r# r r rF rg s r) test_fromstring_count0z TestIO.test_fromstring_count0 s/ M%S B B Bw$r+ c t |d 5 } d d d n # 1 swxY w Y t j | }t |j dk d d S )Nr r Array not emptyopenr# rx r rH rT rr r r/ s r) test_empty_files_textz TestIO.test_empty_files_text s
,
$
$ K % %! .///// ##c t |d 5 } d d d n # 1 swxY w Y t j |d }t |j dk d d S )Nwbr{ r} r r r r s r) test_empty_files_binaryz TestIO.test_empty_files_binary s
,
%
% K # . . .! ./////r c 0 t |d 5 }| | d d d n # 1 swxY w Y t |d 5 }t j ||j }d d d n # 1 swxY w Y t ||j d S )Nr rbr )r rv r# rx r/ r
r rT r rr r r/ s r) test_roundtrip_filezTestIO.test_roundtrip_file s
,
%
%
HHQKKK ,
%
% . AQW---A . . . . . . . . . . . . . . .1af%%%%%s 377 A66A:=A:c | | t j ||j }t ||j d S r> )rv r# rx r/ r
r )rT r rr r/ s r) r zTestIO.test_roundtrip sB K AG 4 4 41af%%%%%r+ c t j | }| | t j |d }t || d S )NT) allow_pickle)pathlibPathdumpr# r r
)rT r rr r r/ s r) test_roundtrip_dump_pathlibz"TestIO.test_roundtrip_dump_pathlib sJ L & & q GAD ) ) )1a r+ c 4 | }t j ||j }t ||j | d }t j ||j }t || d d S )Nr rY )r r# r} r/ r
r rP rT r r r/ s r) test_roundtrip_binary_strz TestIO.test_roundtrip_binary_str s|
IIKK M!17 + + +1af%%%
IIcNN M!17 + + +1aiinn-----r+ c V |j }d t t | }t j |d }t j | }t || || t || || d d S )N@r rW decimal)
r rh r r1 r r# r isfiniter
r )rT r r r/ nan_masks r) test_roundtrip_strzTestIO.test_roundtrip_str s
FLLNN HHSa[[ ! ! M! % % %KNN?1X;( 444!!XI,8) aHHHHHHr+ c |j }d t t | }t j |d }t || d S )Nr r )r rh r r1 r r# r r
r s r) test_roundtrip_reprzTestIO.test_roundtrip_repr sT
FLLNN HHSq\\ " " M! % % %1a r+ c | | d }t j |dd 5 }||_ ||_ t t t j ||j d d d d S # 1 swxY w Y d S )Nc t d )NzCan not tell or seek)rw r s r) failz-TestIO.test_unseekable_fromfile..fail s 011 1r+ r r bufferingr )
rv r r r tellr rw r# rx r/ )rT r rr r r s r) test_unseekable_fromfilez TestIO.test_unseekable_fromfile s 2 2 2 W\41
5
5
5 BAFAF '2; A A A A B B B B B B B B B B B B B B B B B Bs 6A33A7:A7c | | t j |dd 5 }t j ||j }t
||j d d d d S # 1 swxY w Y d S )Nr r r r rv r r r# rx r/ r
r r s r) test_io_open_unbuffered_fromfilez'TestIO.test_io_open_unbuffered_fromfile s
W\41
5
5
5 * AQW---A q!& ) ) ) * * * * * * * * * * * * * * * * * *s 1A++A/2A/c t j d }| | t t j | |j t |t j | t |d 5 }| |j | | t t j | |j dz d d d n # 1 swxY w Y t |d t |d 5 }| | d d d n # 1 swxY w Y t |t j | t |d 5 }| | d d d n # 1 swxY w Y t t j | |j dz d S )Ni @ r+br2 r rM )
r# r9 rv r ospathgetsizer r
rx r r close)rT rr r r s r) test_largish_filezTestIO.test_largish_file sZ H] # # RW__\22AH===1bk, 7 7888
,
&
& F!
FF18
HHQKKK 661 E E E F F F F F F F F F F F F F F F
\3 %%'''
,
%
%
HHQKKK 1bk, 7 7888
,
%
%
HHQKKK RW__\22AHqLAAAAAs7 A%C??DD<E E"%E"F==GGc | | t j |dd 5 }t j ||j }d d d n # 1 swxY w Y t
||j d S )Nr r5 r r r r s r) test_io_open_buffered_fromfilez%TestIO.test_io_open_buffered_fromfile s
W\42
6
6
6 .! AQW---A . . . . . . . . . . . . . . .1af%%%%%s AA Ac , t j dz t j t j dz g}|D ]}t |d 5 }| |dz
| d d d d n # 1 swxY w Y dD ]}d||fz }t || 5 }| d t
j |t j d | }d d d n # 1 swxY w Y t |d |
d S ) Nr r r3 )r r z%d %sr2 r/ r rQ r ) r DEFAULT_BUFFER_SIZEr r rv readr# rx r r r )rT rr r rH r r r r s r) !test_file_position_after_fromfilez(TestIO.test_file_position_after_fromfile! s '*'')+ 7 7DlD))
QtAv
&
7
7!T4L0,-- #FF1IIIK1====&&((C# # # # # # # # # # # # # # # S"g66666
7 7 7s$ .A88A< ?A< A C55C9 <C9 c ~ t j dz t j t j dz g}|D ]}d|fz }t |d 5 }| |dz
| d | d | d t j dgt
j | | }d d d n # 1 swxY w Y t |d
| t |d 5 }| d
| dd t j dgt
j | | }d d d n # 1 swxY w Y t |d| d S )Nr z%dr r3 r rQ s 12r r r r r r2 ) r r r r rv r# r r rv r r r )rT rr r rH r r r s r) test_file_position_after_tofilez&TestIO.test_file_position_after_tofile5 s '*'')+ 3 3DdWnGlD))
QtAvr
!BJ///66q999ffhh
j' : : : :lE**
aq q! !BJ///66q999ffhh
b' 2 2 2 2 2# 3 3s& B C11C5 8C5 A4F F" %F" c H t |d 5 } d d d n # 1 swxY w Y t |d 5 }t t dt j |t
d d d n # 1 swxY w Y t t dt j |t
d S )Nr r z Cannot read into object arrayr )r r r8 r# rx r7 )rT rr r s r) test_load_object_array_fromfilez&TestIO.test_load_object_array_fromfileN s'
,
$
$ ,
%
% >
,K " Qf
>
>
>
> > > > > > > > > > > > > > > > J(G K V E E E E E Es ##)A//A36A3c
0 t |d 5 }| | d d d n # 1 swxY w Y t |d 5 }t j ||j d }t ||j d d d n # 1 swxY w Y t |d 5 }t |j dz }t |j dz }|j j |z }t j ||j || }t ||j |||z |j j }t j ||j | }t ||j ||z dz d d d d n # 1 swxY w Y t |d 5 }| |d
d d d n # 1 swxY w Y t |d 5 }t t d t j ||j d d d d d d S # 1 swxY w Y d S )
Nr r r )r/ rK r r )r/ r rK r3 r r z1'offset' argument only permitted for binary files)r/ r} rK ) r rv r# rx r/ r
r r& r@ r r) ) rT r rr r r/ count_items offset_items offset_bytesr s r) test_fromfile_offsetzTestIO.test_fromfile_offset[ sX
,
%
%
HHQKKK ,
%
% * AQWQ777A q!& ) ) ) * * * * * * * * * * * * * * * ,
%
% G af++*K qv;;!+L 7+l:L L A
16,|K'??@
7+L AQW\BBBA q!& k)A!)C)D)D"E F F F G G G G G G G G G G G G G G G ,
%
% !
HHQCH ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ,
%
% ' GK QWA
'
'
'
' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' 'sU 3772B BB'CE99E= E=F88F<?F<+H HHz bug in PyPy's PyNumber_AsSsize_tr c ` d }d }t j } t |d 5 }| | |t f|t
ffD ],\ }}|t _ t
|t j | - d d d n # 1 swxY w Y |t _ d S # |t _ w xY w)Nc dS )Nr[ rd fds r) dup_strz-TestIO.test_fromfile_bad_dup..dup_str s 5r+ c ddz S )Nr2 D rd r s r)
dup_bigintz0TestIO.test_fromfile_bad_dup..dup_bigint s b5Lr+ r ) r dupr rv r) rw r r# rx ) rT r rr r r old_dupr r excs r) test_fromfile_bad_dupz TestIO.test_fromfile_bad_dup} s & lD))
7Q ")9!5
G7L M 7 7HC BF!#r{A66667
7
7
7
7
7
7
7
7
7
7
7
7
7
7
7 BFFFWBF s/ B AB: B B
B
B
B B-c 6 d|vrt j |fi |}nt j |fi |}t || t |d 5 }| | d d d n # 1 swxY w Y t j |fi |}t || d S )Nr} r )r# r} r r
r rv rx )rT r r2 rq ro r/ r s r) _check_fromzTestIO._check_from s ??
a&&2&&AA
a&&2&&A1e$$$
(D
!
! Q
GGAJJJ K ' 'B ' '1e$$$$$s
A,,A03A0periodcommac # K |j dk rdV dS |j dk r-t 5 dV ddd dS # 1 swxY w Y dS J |j )a
Including this fixture in a test will automatically
execute it with both types of decimal separator.
So::
def test_decimal(decimal_sep_localization):
pass
is equivalent to the following two tests::
def test_decimal_period_separator():
pass
def test_decimal_comma_separator():
with CommaDecimalPointLocale():
pass
r Nr )r r )rT rp s r) decimal_sep_localizationz TestIO.decimal_sep_localization s ( =H $ $ EEEEE
]g
%
%(**
('- ' '5s ?AAc
| dt j t j t j t j t j t j t j g|d d S )Ns3 nan +nan -nan NaN nan(foo) +NaN(BAR) -NAN(q_u_u_x_)r{ r )r r# r1 rT rr r s r) r zTestIO.test_nan sR B
VRVRVRVRVRVRV D r+ c
| dt j t j t j t j t j t j t j g|d d S )Ns. inf +inf -inf infinity -Infinity iNfInItY -inFr{ r )r r# ri r s r) test_infzTestIO.test_inf sX =
VRVbfWbfrvgrvw G r+ c < | dg d|d d S )Ns' 1.234 -1.234 .3 .3e55 -123133.1231e+133)X9v?gX9vg333333?g. LKR?KgY]xr{ r r r s r) test_numberszTestIO.test_numbers s; 6 9 9 9 r+ c ` | dt j g d |d d S )Ns ? @ @@ @r r r )r r# r rT rr s r) test_binaryzTestIO.test_binary sC @ H\\\ " " r+ c < | dg d|d d S )N 1,2,3,4r r r r r s r) rA zTestIO.test_string s, %5%5%5|MMMMMr+ c | dg d|dd | dg d|dd | dg d|dd d S ) Nr r r r )r r} r r r5 r r s r) test_counted_stringzTestIO.test_counted_string s (((,aS J J J l! F F F (((,bc K K K K Kr+ c H | dg d|t d d S )N 1 2 3 4 r r{ r r r
r s r) test_string_with_wszTestIO.test_string_with_ws s= l#3 P P P P Pr+ c J | dg d|dt d d S )Nr rU r r{ )r r/ r} r r s r) test_counted_string_with_wsz"TestIO.test_counted_string_with_ws s;