Po T d Z ddlZddlmZmZmZmZ ddlm Z m
Z
m Z m Z m
Z
mZmZmZ ddlZ G d d Z G d d ZdS ) z-Test functions for 1D array set operations.
N)assert_array_equal assert_equal
assert_raisesassert_raises_regex)ediff1d intersect1dsetxor1dunion1d setdiff1duniquein1disinc
e Zd Zd Zd Zd Zd Zd Zej
d e j g de j
d e j d
f e j g de j
e j g d e j d d f e j g d
e j e j e j d fg d Zej
d e j g de j dd e j g de j f e j g de j e j dge j d e j g de j f e j g de j dd e j g de j f e j g de j
dd e j g de j
fg d Zej
dg d d Zej
dg d d Zd Zej
dg d d Zej
dg d d! Zd" Zej
dg d d# Z ej
dd d$g d% Z d& Z ej
d'e j e j fe j e j fe j e j fe j e j fe j e j fe j e j fg ej
dg d d( Z!ej
dg d d) Z"d* Z#d+ Z$d, Z%d- Z&d. Z'd/ Z(d0 Z)d1 Z*d2 Z+d3 Z,d4 Z-d S )5
TestSetOpsc t j g d }t j g d }t j g d }t ||d }t || t j g d }t j g d }t j g d }t || }t || t g t g g d S )N r r r )r r r T
assume_unique)r r r r r )r r r r r r r )nparrayr r )selfabecceds t/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib64/python3.11/site-packages/numpy/lib/tests/test_arraysetops.pytest_intersect1dzTestSetOps.test_intersect1d s H\\\ " " H___ % %
Xiii
1D 1 1 11b!!!
H___ % % H*** + +
Xiii
1 1b!!!2{2r 2 233333 c G d d } | }t || }t || t g dg d }t |g d d S )Nc e Zd Zd ZdS )4TestSetOps.test_intersect1d_array_like..Testc * t j d S )Nr )r aranger s r$ __array__z>TestSetOps.test_intersect1d_array_like..Test.__array__% s y||#r& N)__name__
__module__ __qualname__r- r& r$ Testr) $ s#
$
$
$
$
$r& r2 r r r )r r )r r2 r ress r$ test_intersect1d_array_likez&TestSetOps.test_intersect1d_array_like" s $ $ $ $ $ $ $ $
DFF!Q 3""")))YYY//3 *****r& c Z t j g d }t j g d }t ||dd \ }}}t j g d }t || t || | t || | t j g d }t j g d }t ||d \ }}}t j g d }t || t || | t || | t j g d g d
g }t j g d g d g }t ||dd \ }}}t j ||j }t j ||j } t j g d
}
t |
|| t |
|| t j g dg dg }t j g dg dg }t ||d \ }}}t j ||j }t j ||j } t j g d }
t |
|| t |
|| d S )Nr r r r )r r r T)r return_indices)r r r )r r r r r r r )r r r r r r r )r9 )r r r r8 )r r: r )r r r r8 )
r: )r r8 r r: )r r r r8 r8 )r r r: r r )r r r r )r< r= r: r )r r r: )r r r r
unravel_indexshape) r r r r" i1i2eeefui1ui2eas r$ test_intersect1d_indicesz#TestSetOps.test_intersect1d_indices. s H\\\ " " H\\\ " " 1DNNN 2r
Xiii
1b!!!1R5"%%%1R5"%%%
H*** + + H--- . . 1T::: 2r
Xlll
#
#1b!!!1R5"%%%1R5"%%%
HlllMMM2 3 3 HlllNNN3 4 4 1DNNN 2r r17++ r17++
Xlll
#
#2qv&&&2qv&&&
Hooo7 8 8 HlllNNN3 4 4 1T::: 2r r17++ r17++
Xiii
2qv&&&2qv&&&&&r& c t j g d }t j g d }t j g d }t || }t || t j g d }t j g d }t j g d }t || }t || t j g d }t j g d }t j g d }t || }t || t g t g g d S ) Nr r )r r r r3 )r8 r r )r r r r r r8 )r r: r r )r8 r r r: )r r r r r r r r! r" s r$
test_setxor1dzTestSetOps.test_setxor1dU s* H\\\ " " H___ % %
Xiii
QNN1b!!! HYYY HYYY
X(((
)
) QNN1b!!! H\\\ " " H\\\ " "
X(((
)
) QNN1b!!!2xB / /00000r& c t j g }t j dg }t j ddg }t g t | t dgt |d t dgt |d t ddgt |dd t g t | t dgt | t g dt |d d
t g d t |d d
gd dg t dd
gt |d
t g dt |d dg t d dgt |d t g dt |d d
g d S )Nr r r )to_begin)to_end)rM rN )r r r> r r> )r r8 r r r: r r8 r: )r r r: )r r8 r )r r r r )r zero_elemone_elemtwo_elems r$ test_ediff1dzTestSetOps.test_ediff1dm s HRLL 8QC==8QF##2wy 1 1222A3 A > > >???A3 ! < < <===B7GI1$M$M$MNNN2wx 0 0111A3 1 1222999gh1&M&M&MNNN???"8q!faVLLL N N NAq678A#>#>#>???999gh1v&F&F&FGGGAq678a#@#@#@AAA999gh!Q&H&H&HIIIIIr& z ary, prepend, append, expectedr3 dtypeNrN )r r r rM ) ?g @g "@c d | }t t | 5 t ||| d d d d S # 1 swxY w Y d S )Nz dtype of `{}` must be compatiblearyrN rM )formatr TypeErrorr )r rY prependappendexpectedmsgs r$ !test_ediff1d_forbidden_type_castsz,TestSetOps.test_ediff1d_forbidden_type_casts s : 177AA
C
0
0 & & ! $
&
&
&
& & & & & & & & & & & & & & & & & & &s A AAzary,prepend,append,expectedi i )r r r r r )r r r r )r r r r r )r r r ra c x t j ||| }t || |j |j k sJ d S )NrX )r r r rU )r rY r\ r] r^ actuals r$ test_ediff1d_scalar_handlingz'TestSetOps.test_ediff1d_scalar_handling sO : #)%,. . . VX&&&|x~------r& kind)Nsorttablec
d }t j |t gdh
fd}t j d g d }t j g dg dg d g } ||| d
d g}d } ||| t j d
} ||| ||| ||| |d| ||d |dd dk r&g } || | ||| || | t t j t j fD ]}
dk r|
t j k r|
t j t j hv rt j g d|
} n |
t hv rt j g d } t j g |
} || | || | || | d S )Nc z t j | }| |v S )N)r asarrayflattentolist)r r s r$
_isin_slowz(TestSetOps.test_isin.._isin_slow s0
1
%%''..00A6Mr& r )otypesexcludedc d t | | } | | }t || d S )Nre )r r )r r xy isin_slowre s r$ assert_isin_equalz/TestSetOps.test_isin..assert_isin_equal s< Q%%%A !QA q! $ $ $ $ $r& r r r )r< )r r r ) ! )r> r: )r r8 )r> r r r r8 rg rT TFF)r vectorizeboolr+ reshaper int64float64)r re rm ru r r r" dfrr rU ar empty_arrayrt s ` @r$ test_isinzTestSetOps.test_isin sr
LTFaSIII % % % % % %
IbMM ! !))) , , HlllIII|||< = =!Q V !Q
HQKK!Q !Q !Q !Q !Q !Q 7?? A a # # # a # # # a # # # BHbj1
8
8Ew5BJ#6#62:...Xlll%8884& X 2 2 233 (2U333K k2 . . . b+ . . . k; 7 7 7 7
8
8r& c dD ]q}g d}g d|z }t j g d }t ||d| }t || d|d< t j g d }t ||d| }t || d
\ |d< |d < t j g d }t ||d| }t || t j g d
}g d|z }g d}t ||| }t || |g d|z z }g d}t ||| }t || t j g d }t j g d|z }t j g d }t ||| }t || t j g d }t j g d|z }t j g d }t ||| }t || t j ddg }t j ddg|z }t j ddg }t ||| }t || st j dg }t j dg }t j dg }t ||| }t || |dv r"t t g g | g d S d S )Nr r< r r TFTTTr re r: r )FFTT)r r: r )TFTFr r r r r r r r r r r r r r rw )FTFTTTTTTFTFFFrq )r r r )TTTTTTTTTTTFTT)r r r r r )r r r r r r )TFTTTr r F> Nrf r r r
r )r re multr r r! r" s r$ test_in1dzTestSetOps.test_in1d sU / & / &D A $&A33344BQD999A q" % % %AaD44455BQD999A q" % % % JAaD!A$44455BQD999A q" % % %CCCDDA D A4 4 4BQ%%%A q" % % %III$$A+ + +BQ%%%A q" % % % &&A 4/00A33344BQ%%%A q" % % % ))A+++d233A999::BQ%%%A q" % % %!Q A!Q$ ''A5%.))BQ%%%A q" % % % % HaSMM HaSMM
Xug
AD ! ! !1b!!! > ! ! tB 6 6 6 ; ; ; ; ; " !r& c t j g d }t j ddg }t j g d }t || }t || d S )N)r r r" r er" r r r r" )TFTFFTFFr rJ s r$ test_in1d_char_arrayz TestSetOps.test_in1d_char_arrayA sd H=== > > Hc3Z
XKKK
L
L AJJ1b!!!!!r& c
dD ]b}t j g d }g d|z }t t j t ||| t ||d| c|dv rdD ]}t j g dt j }g d|z }t j |t j }t t j t ||| t ||d| d S d S )
z Test in1d's invert parameterr r rw rq Tinvertre > Nrf rT N)r r r r r
float32)r re r r r s r$ test_in1d_invertzTestSetOps.test_in1d_invertJ sG
C CDCCCDDA D A rya)>)>)> ? ? #AqD A A A
C
C
C
C > ! !
G
GH G G G"$*. . . II$HQbj111"29T!QT-B-B-B#C#C#'1T#E#E#EG G G G
" !
G
Gr& c . t j d dd }t j dd dd }t j dd dd }t j g d }t t ||d| | t t ||d
| | t t ||d| | t t ||d
| | d S ) Nr8 r r r> ? ry )FFFTTTTr F)r r+ r r r r
)r re r r long_br! s r$ test_in1d_ravelzTestSetOps.test_in1d_ravel_ s:
IaLL A & & IaOO # #Aq ) )1b!!))"a00
X===
>
>41DtDDD 41E%) + + + 46%) + + + 46%) + + + r& c ` t j g dt j }t j g dt j }t j g dt }t |t || t t j | t ||d t j g dt j }t j g dt j }t j g d }t ||d
}t || d S ) z(Hit the standard isin code with integers)r r r r r r eArT )r r r r )r r r r r r r Tr r )r r r r r r r r N)r r r r r r
r )r r r r^ r! r" s r$ !test_in1d_hit_alternate_algorithmz,TestSetOps.test_in1d_hit_alternate_algorithmt s
H,,,BH = = = H^^^28 4 4 48 1 1 1>>>8T!QZZ00029X..Q$0G0G0GHHH H\\\ 2 2 2 H))) : : :
X///
0
0 AT * * *1b!!!!!r& c 0 t j ddg }t j g d }t j ddg }t |t ||| t t j | t ||d| dS )z&Test that in1d works for boolean inputTF)FFFrq r N)r r r r
r )r re r r r^ s r$ test_in1d_booleanz TestSetOps.test_in1d_boolean s
HdE] # # H*** + +8UDM**8 14000 2 2 229X.. 1T=== ? ? ? ? ?r& rf c X t j d }| ddd }| ddd }t || }| d }| d }t
|t ||| dS )z(Test that in1d works for timedelta inputr d r< )sizetimedelta64[s]rq N)r random RandomStaterandintr
astyper )r re rstater r truth a_timedelta b_timedeltas r$ test_in1d_timedeltaz TestSetOps.test_in1d_timedelta s &&q)) NN1cN + + NN1cN + +Q
hh /00 hh /00 5${Kd"K"K"KLLLLLr& c t j g dd }|}t j t 5 t ||d d d d d S # 1 swxY w Y d S )Nr r r r rT rg rq )r r pytestraises
ValueErrorr
r r r s r$ test_in1d_table_timedelta_failsz*TestSetOps.test_in1d_table_timedelta_fails s HYYY&6 7 7 7
]:
&
& % % AG $ $ $ $ % % % % % % % % % % % % % % % % % %s AAAz
dtype1,dtype2c t j |t j }t j g d| }|rt j g d| }nt j g d| }t j g d }|dk oLt |t j k o|t j k |t j k o|t j k f }|rHt j t d 5 t ||| d
d
d
d
S # 1 swxY w Y d
S t t ||| | d
S ) z7Test that in1d works as expected for mixed dtype input.)r r r r rT )ir )r r )TTFFrg zexceed the maximum)matchrq N) r
issubdtype
signedintegerr anyint8int16r r RuntimeErrorr
r ) r dtype1dtype2re is_dtype2_signedar1ar2r^ expect_failures r$ test_in1d_mixed_dtypez TestSetOps.test_in1d_mixed_dtype s =1ABBh|||6222 8(>>>888CC(===777C8 6 6 677 S bg 4&BH"4 bh 46RW#42
. .
D |3GHHH
*
*S#D))))
*
*
*
*
*
*
*
*
*
*
*
*
*
*
*
*
*
*
tC4 8 8 8( C C C C Cs 1DDDc t j d D ]}t j g dt }t j g d| }t j g dt }t t ||| | ||}}t j g dt }t t ||| | dS ) z4Test that in1d works as expected for bool/int input.
AllIntegerr} rT r r r r )FTTrq )TTTTN)r typecodesr r r r
)r re rU r r r^ s r$ test_in1d_mixed_booleanz"TestSetOps.test_in1d_mixed_boolean s \,/ @ @E---T:::A U333Ax 3 3 34@@@H tAqt 4 4 4h ? ? ?aqAx 8 8 8EEEH tAqt 4 4 4h ? ? ? ? @ @r& c d g}t j dgdz }t j dg }t j || }t || d S Nr r< Fr r r r r^ results r$ test_in1d_first_array_is_objectz*TestSetOps.test_in1d_first_array_is_object sV fhs2v 8UG$$c""68,,,,,r& c d}t j d gdz }t j dg }t j || }t || d S r r r s r$ test_in1d_second_array_is_objectz+TestSetOps.test_in1d_second_array_is_object sT hvby!!8UG$$c""68,,,,,r& c d g}t j d gdz }t j dg }t j || }t || d S )Nr< Tr r s r$ test_in1d_both_arrays_are_objectz+TestSetOps.test_in1d_both_arrays_are_object sV fhvby!!8TF##c""68,,,,,r& c $ t j dt fdt fg }t j dg| }t j dgdz | }t j dg }t j || }t
|| d S )Nfield1field2)r NrT r< T)r rU intobjectr r
r )r dtr r r^ r s r$ +test_in1d_both_arrays_have_structured_dtypez6TestSetOps.test_in1d_both_arrays_have_structured_dtype s X#6(:;
<
<h {"---h {2~R0008TF##c""68,,,,,r& c V t j ddgt }t j ddgt }t j ddg }t j || }t || t j ||d }t |t j | t j g dt }|d d }t j g dt }|d d }t j ddg }t j || }t || t j ||d }t |t j | t j g dt }|d d }t j ddgt }t j dd g }t j || }t || t j ||d }t |t j | d S )
Nr r rT Tr )r )r r r rO )r r r r F)r r r r
r r r s r$ 'test_in1d_with_arrays_containing_tuplesz2TestSetOps.test_in1d_with_arrays_containing_tuples s hay///hay///8T4L))c""68,,,c$///629X#6#6777 h(((777#2#hh(((777#2#h8T4L))c""68,,,c$///629X#6#6777h(((777#2#hhay///8T5M**c""68,,,c$///629X#6#677777r& c t j g d }t j g d }t t t ||d t j g dt
}t j g dt
}t t t ||d t j t j fD ]}t j g d| }t j d t j | j g| }t t t ||d t j ||d
}t |d gd gd
z z t j ||d }t |d gd gd
z z d
S )z&Test that in1d raises expected errors.)r r r r r )r r r8 r: r< quicksortrq )r r r r r rT rg )rO r r r r rO NTFr rf ) r r r r r
r int32r iinfomaxr r )r r r obj_ar1obj_ar2rU overflow_ar2r s r$ test_in1d_errorszTestSetOps.test_in1d_errors s h''h'''((j$S{CCCC (...f===(...f===j$wGGGGh) = =E(+++5999C 8R% )<$=UKKKL
c r: r r r r r )r r r r r
r r )r r r c1aux1aux2c2s r$
test_manywayszTestSetOps.test_manywaysc sy H___ % % H*** + +
a^^1a q!}}
tT
"
"2r"""""r& ).r. r/ r0 r% r5 rH rK rS r mark parametrizer r r nanr r r` r r r rd r r r r r r r r r uint8uint16r r r r r r r r r r r r r r1 r& r$ r r s~ 4 4 4$
+
+
+%' %' %'N1 1 10J J J& [ =
)))28 , , ,
)))28 , , , )))2: . . .
,,,bg . . . %@ .
&
&/ .
& [ %
"(999BH
-
-
-
"(<<=:8x [ V%<%<%<==;< ;< >=;=G( [ V%<%<%<== >= (" " " [ V%<%<%<==? ? >=? [ VdF^44M M 54M% % % [
Wbh
Xrw
Xry !
Y !
Xrx
Xrx
[ V%<%<%<==D D >=
D. [ V%<%<%<==
@
@ >=
@- - -- - -- - -- - - 8 8 8B = = =D0 0 0$8 8 8$' ' '= = =
# # # # #r& r c e Zd Zd Zd Zd Zd Zej dddg d Z
d Z d
Z d Z
d Zd
ZdS )
TestUniquec d }g ddz }g d}g d}g ddz }t j g dd }g }| t j d | t j d | d
| d |D ]<}t j || } t j || }
|| |
|||| =d }t j t | | } || d d < t j t | | }
||
d d < || |
|||| d
d
g}t j t t || | } t j t t || | }
|| |
|||| g d} t t j | ddg g d}g d} g d} t | }
t |
| t |d \ }}t || t || t j d }d|d< t j |d \ }}t |t j
d g }t j |d d }t j |d d }t j |dd dd \ }}t |j t j t |j t j t |j t j t |j t j d t j d t j g}d d t j g}g d }g d }g d }t t j | | t t j |d ||f t t j |d ||f t t j |d! ||f d"t j d#t% d$t j t% d t j g}d#d"t% d$t j g}g d%}g d&}g d'}t t j | | t t j |d ||f t t j |d ||f t t j |d! ||f t j d( }t j d) |t j d* |g}t j d* t j d) |g}g d }g d }g d }t t j | | t t j |d ||f t t j |d ||f t t j |d! ||f t j d( }t j dd+ |t j dd, |g}t j dd, t j dd+ |g}g d }g d }g d }t t j | | t t j |d ||f t t j |d ||f t t j |d! ||f t j gd-z }t j g}d.g}g d/}d-g}t t j | | t t j |d ||f t t j |d ||f t t j |d! ||f d S )0Nc ` d}| d| }t | }t ||| | d| }t | ddd \ }} t ||| t | || | d| }t | ddd \ }} t ||| t | || | d| }t | ddd \ }} t ||| t | || | d| }t | ddd \ }}
} t ||| t |
|| t | || | d | }t | ddd \ }}
} t ||| t |
|| t | || | d
| }t | ddd \ }}
} t ||| t |
|| t | || | d | }t | ddd \ }}
} } t ||| t |
|| t | || t | || d S ) Nz check {0} failed for type {1}values return_indexTFreturn_inverse
return_countsz return_index and return_inversez return_index and return_countsz return_inverse and return_countsz.return_index, return_inverse and return_counts)rZ r r )
r r rA rB r" r base_msgr_ vjj1j2j3s
r$ check_allz,TestUnique.test_unique_1d..check_allr s 6H//(B//Cq A q!S ) ) )//."55C!T5%00DAq q!S ) ) ) q"c * * *//"2B77C!UD%00DAq q!S ) ) ) q"c * * *///266C!UE400DAq q!S ) ) ) q!S ) ) )//"CRHHC q$e44IAr2 q!S ) ) ) r2s + + + r2s + + +//"BBGGC q$t44IAr2 q!S ) ) ) r2s + + + r1c * * *//"DbIIC q%t44IAr2 q!S ) ) ) r2s + + + r1c * * *// $79;= =C "1dD$ 7 7MAr2r q!S ) ) ) r2s + + + r2s + + + r1c * * * * *r& )r r r r r r r r< )r r r r )r r r r )r r r r r r r )r r r r r AllFloat
datetime64[D]timedelta64[D]O) i) ? ? r r r )r r r r r3 )r r r r r r T)r r r .)r r )r r g @rV )r r r )r r r r )r r r )r
y @ y ? ? )r r r )r r r r r )r r r natz
2020-12-26z
2020-12-24Dhr r r )r multiplyextendr r] r emptylenlistzipr r chararrayzerosr rU intpr complex
datetime64 timedelta64)r r r r rA rB r" typesr aabbunqinva1a2a2_inva1_idxa3_idxa3_invuaua_idxua_invua_cntr all_nanss r$ test_unique_1dzTestUnique.test_unique_1dp s . + . + . +`
" ! !" $ LL
\\
"
"
"2
% K b ) )
R\,/000
R\*-...
_%%%
%&&& - -B!R B!R B Ib"b"a , , , ,
Xc!ffb
!
!111
Xc!ffb
!
!111 "b"b!R(((
#
Xd3q!99oor
*
*
Xd3q!99oor
*
* "b"b!R((( $
#
#29R==8X*>???
% $ $ii
AYY2s###Ad333
F2s###63'''
LOO#Yq666
F628A;;///
1400031T2221514268 8 889<V\27+++V\27+++V\27+++V\27+++ "&#rv &3
RYq\\2&&&RYqt444r6lCCCRYq666V EEERYq555F|DDD RVVWS"&%9%973;O;O Pfgc26 2 2
3 RYq\\2&&&RYqt444r6lCCCRYq666V EEERYq555F|DDD mE""
]<
(
(#r}\/J/JC PmL))2= +F+F
L RYq\\2&&&RYqt444r6lCCCRYq666V EEERYq555F|DDD nU##
^As
#
#S".C*@*@# FnQ$$bnQ&<&>>i!5SzC=1 3 3 3 blFBIbMMBBBBblFBIbMMCCCCCCr& c d}g dg dg}t j | }t t |d t |d | t t |d t |d | d S )Nz Unique failed on list of listsr r r r r= r )r rj r r )r r_ inpinp_arrs r$ test_unique_axis_listz TestUnique.test_unique_axis_list! s .yy)))$*S//6#A...wQ0G0G0GMMM6#A...wQ0G0G0GMMMMMr& c g }| t j d | t j d | d | d | dt fdt fg | dt fdt
fg |D ]}| | d}t j dt j d
d
t }t j d d
gd
d
ggt }t t |d || d}t j ddgddgddgddgg }t j ddgg }t t |d || d S )Nr r r r r r z&Non-bitwise-equal booleans test failedr< rT rO r FTr r= z"Negative zero equality test failedg r )r r r r] r floatr@ r+ r r viewr r r r )r r+ rU r_ datar s r$ test_unique_axiszTestUnique.test_unique_axis( s
R\,/000
R\*-...
_%%%
%&&&
sCj3*-...
sCj3,/000 ( (E ' ' ' '6y28,,,44R;;@@FFE4=4,7tDDD6$Q///===2x$sDkD#;d LMMD#;-((6$Q///=====r& r> r rO c x t j g d }t || }t |g d d S )N)r r r r r r r r r= r7 r r r r )r r> rr uniqs r$ test_unique_1d_with_axisz#TestUnique.test_unique_1d_with_axis> sD H--- . .ad###4 .....r& c Z t j dt j }t |dddd \ }}}}t |j |j t
|t j d t
|t j dg t
|t j ddg t
|t j dg t |d ddd \ }}}}t |j |j t
|t j d t
|t j g t
|t j g t
|t j g d
}t j | }t t | D ]Z}t | } || dk rd| |< nd | |< t
t || t j | [d S ) N)r r )r@ rU r Tr> r r r
)r r )r@ r r )r r r r r r r r= ) r r! r r r rU r r ranger" r# )
r single_zerorO idxr/ cntr@ multiple_zerosr> expected_shapes
r$ test_unique_axis_zerosz!TestUnique.test_unique_axis_zerosD s hV27;;; $[qt48 N N Nc3 TZ!23334!7!7!78883!
...3!Q 0 01113!
... %[qt48 N N Nc3 TZ!23334!7!7!78883 ---3 ---3 --- &...#e**%% ? ?D !%[[NT{a '(t$$'(t$ vn4 @ @ @ !xn = = =
?
?
?
? ? ?r& c D t j g dd }t j |d }t j | }t j |dd \ }}}d}t |j |j | t |j |j | d S )N)@ r r r r r r r r r r r r r r r rT r T)r r
z6Unique returned different results when asked for index)r r ma masked_equalr r rK mask)r rr rs r v2r r" r_ s r$ test_unique_maskedz TestUnique.test_unique_maskedh s HFFF"
$
$
$ E q! $ $ IaLL9QTFFFAqF1627C0001627C00000r& c d}dD ]E}t j dgdgg| }t j |d }t ||||z Fd S )Nz*sort order incorrect for integer type '%s'bhilqrO r r= rN )r fmtr r r s r$ test_unique_sort_order_with_axisz+TestUnique.test_unique_sort_order_with_axisu sh ; / /B2$b))A !!$$$A q!S2X . . . . / /r& c t j g dg dg dg dg | }d}t j ddg }t t | | | | d}t j g dg dg }t t |d | | | d}t j g d g d
g d g d
g }t t |d | | | d }t j ddgddggddgddggg | }t j |ddgd }t t |d || t |dd
d
d
\ }}}} d}t || || d}t || | d}t | t j d d g | t |dd
d
d
\ }}}} d}t |d d |f | d}t |d d |f | d}t | t j g d | d S )N)r r r r )r r r r z&Unique with 1d array and axis=0 failedr r z&Unique with 2d array and axis=0 failedr= z&Unique with 2d array and axis=1 failed)r r r rD z&Unique with 3d array and axis=2 failedr TrR z-Unique's return_index=True failed with axis=0z/Unique's return_inverse=True failed with axis=0z.Unique's return_counts=True failed with axis=0z-Unique's return_index=True failed with axis=1z/Unique's return_inverse=True failed with axis=1z.Unique's return_counts=True failed with axis=1)r r r )r r r r r take)
r rU rK r_ r data3drO rU r/ rV s
r$ r@ zTestUnique._run_axis_tests~ s x % % % ' ( ( )/u
71a&!!6$<< u)=)=sCCC6<<< 6776$Q/// u1E1EsKKK6999iiiIIIFGG6$Q/// u1E1EsKKK6QF F $ F F $% & & '-fUmm !Qa0006&q11163??? $T48 N N Nc3=49dC000?49d+++>3!Q 0 0#666 $T48 N N Nc3=43<...?43<...>3 3 3S99999r& c t j ddt j t j t j g }t j | }t j |d }t |t j dt j g t |t j dt j t j t j g d S )Nr F) equal_nan)r r r r r )r r r. not_unqs r$ test_unique_nanequalsz TestUnique.test_unique_nanequals s HaBFBFBF3 4 4ill)A///3!RV 5 56667BHa-H$I$IJJJJJr& N)r. r/ r0 r; rB rG rL r r r rP rY r` rd r@ rk r1 r& r$ r r n s gL gL gLRD D DN N N> > >, [ VaW--/ / .-/
"? "? "?H 1 1 1/ / /*: *: *:XK K K K Kr& r )__doc__numpyr
numpy.testingr r r r numpy.lib.arraysetopsr r r r
r r r
r r r r r1 r& r$ rp s? ? ? ? ? ? ? ? ? ? ? ? ?
] # ] # ] # ] # ] # ] # ] # ] #@BK BK BK BK BK BK BK BK BK BKr&