U
|
P ±dø ã @ st d Z ddlZddlmZmZmZmZ ddlm Z m
|
Z
|
m Z m Z m Z mZmZmZ ddlZG dd dZG dd dZdS ) z-Test functions for 1D array set operations.
|
|
é N)Úassert_array_equalÚ assert_equalÚ assert_raisesÚassert_raises_regex)Úediff1dÚ intersect1dÚsetxor1dÚunion1dÚ setdiff1dÚuniqueÚin1dÚisinc @ s0 e Zd Zdd Zdd Zdd Zdd Zd d
|
Zej
|
d e j d d dge j dde jdfe j d d dge j de j ddd ge jdddfe j dddge jde je jdfg¡dd Zej
|
de j d d dge jddd e j d d d d ge jdfe j d d dge jde j dge jdde j dd d ge jdfe j d d dge jdd d e j d d d d ge jdfe j d d dge j ddd e j dd d d ge j dfg¡d d! Zej
|
d"dd#d$g¡d%d& Zej
|
d"dd#d$g¡d'd( Zd)d* Zej
|
d"dd#d$g¡d+d, Zej
|
d"dd#d$g¡d-d. Zd/d0 Zej
|
d"dd#d$g¡d1d2 Z ej
|
d"dd#g¡d3d4 Z d5d6 Z ej
|
d7e je jfe je jfe j e j fe j e j fe j e jfe je j fg¡ej
|
d"dd#d$g¡d8d9 Z!ej
|
d"dd#d$g¡d:d; Z"d<d= Z#d>d? Z$d@dA Z%dBdC Z&dDdE Z'dFdG Z(dHdI Z)dJdK Z*dLdM Z+dNdO Z,dPdQ Z-dS )RÚ
|
TestSetOpsc C s² t ddddg¡}t dddddg¡}t dddg¡}t||dd}t|| t dddddg¡}t dddddddg¡}t dddg¡}t||}t|| tg tg g d S ) Né é é é é é T©Ú assume_unique)ÚnpÚarrayr r )ÚselfÚaÚbÚecÚcZed© r úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\numpy/lib/tests/test_arraysetops.pyÚtest_intersect1d s
|
|
|
zTestSetOps.test_intersect1dc C sR G dd d}| }t ||}t|| t dddgdddg}t|dddg d S )Nc @ s e Zd Zdd ZdS )z4TestSetOps.test_intersect1d_array_like.<locals>.Testc S s
|
t d¡S )Nr )r Úarange©r r r r Ú __array__% s z>TestSetOps.test_intersect1d_array_like.<locals>.Test.__array__N)Ú__name__Ú
|
__module__Ú __qualname__r# r r r r ÚTest$ s r' r r r )r r )r r' r Úresr r r Útest_intersect1d_array_like" s
|
|
z&TestSetOps.test_intersect1d_array_likec
|
C s* t ddddg¡}t ddddg¡}t||ddd\}}}t dddg¡}t|| t|| | t|| | t dddddddg¡}t ddddddddg¡}t||dd \}}}t ddddg¡}t|| t|| | t|| | t ddd
|
dgd ddd gg¡}t ddd dgd dddgg¡}t||ddd\}}}t ||j¡}t ||j¡} t ddd dg¡}
|
t|
|
|| t|
|
|| t ddd
|
ddgdd dd dgg¡}t ddd d gd ddd gg¡}t||dd \}}}t ||j¡}t ||j¡} t dd dg¡}
|
t|
|
|| t|
|
|| d S )Nr r r r é T)r Úreturn_indicesé )r+ r r é é
|
é é )r r r r Z unravel_indexÚshape) r r r r Úi1Úi2ÚeeZefZui1Zui2Zear r r Útest_intersect1d_indices. s<
|
|
" z#TestSetOps.test_intersect1d_indicesc C sö t ddddg¡}t dddddg¡}t dddg¡}t||}t|| t dddg¡}t dddg¡}t ddddddg¡}t||}t|| t ddddg¡}t ddddg¡}t ddddddg¡}t||}t|| tg tg g d S ) Nr r r r r r r* r, )r r r r ©r r r r r r r r Ú test_setxor1dU s
|
|
|
|
|
|
zTestSetOps.test_setxor1dc C s8 t g ¡}t dg¡}t ddg¡}tg t| tdgt|dd tdgt|dd tddgt|ddd tg t| tdgt| tddd gt|dd d td
|
d ddd gt|d
|
d gdd gd tdd gt|d d tddd gt|dd gd tddgt|dd td
|
d dgt|d
|
d gd d S ) Nr r r )Úto_begin)Úto_endéÿÿÿÿ)r8 r9 r r0 r r* r, )r r r r )r Z zero_elemZone_elemZtwo_elemr r r Ú test_ediff1dm s"
|
ÿ zTestSetOps.test_ediff1dz ary, prepend, append, expectedr r r ©ÚdtypeNr9 r r r8 ç ð?g @g "@c C s2 d |¡}tt| t|||d W 5 Q R X d S )Nz dtype of `{}` must be compatible©Úaryr9 r8 )Úformatr Ú TypeErrorr )r r@ ÚprependÚappendÚexpectedÚmsgr r r Ú!test_ediff1d_forbidden_type_casts s
|
þz,TestSetOps.test_ediff1d_forbidden_type_castszary,prepend,append,expectedi i r r i÷ÿÿÿc C s. t j|||d}t|| |j|jks*td S )Nr? )r r r r= ÚAssertionError)r r@ rC rD rE Úactualr r r Ú test_ediff1d_scalar_handling£ s þ
|
z'TestSetOps.test_ediff1d_scalar_handlingÚkindÚsortÚtablec s dd }t j|tgdhd fdd}t d¡ dd d
|
g¡}t d d d gddd gdddgg¡}||| ddg}d}||| t d ¡}||| ||| ||| |d| ||d |dd dkròg } || | ||| || | tt jt jfD ]}
|
dkr |
|
t jkr q |
|
t jt jhkrFt jd d d g|
|
d} n |
|
thkrbt dddg¡} t jg |
|
d} || | || | || | q d S )Nc S s t |¡ ¡ ¡ }| |kS )N)r ÚasarrayÚflattenÚtolist)r r r r r Ú
|
_isin_slowË s z(TestSetOps.test_isin.<locals>._isin_slowr )ZotypesZexcludedc s& t | |d} | |}t|| d S )N©rK )r r )r r ÚxÚy©Z isin_slowrK r r Úassert_isin_equalÐ s
|
z/TestSetOps.test_isin.<locals>.assert_isin_equalé r r r r. é é r é é é! )r0 r, )r r* )r0 r r r* rM r< TF)r Z vectorizeÚboolr! Úreshaper Úint64Úfloat64) r rK rQ rV r r r ÚdÚfrS r= ÚarZ empty_arrayr rU r Ú test_isinÆ s@ "
|
|
|
|
|
|
|
|
|
|
|
|
|
|
zTestSetOps.test_isinc C sÎ dD ]h}ddddg}dddddg| }t dd ddg¡}t||d|d
|
}t|| d |d < t d d ddg¡}t||d|d
|
}t|| d \|d < |d< t dd dd g¡}t||d|d
|
}t|| t ddddddddddddddg¡}dddg| }d dd ddddddd dd d d g}t|||d}t|| |dddg| }dddddddddddd ddg}t|||d}t|| t ddddg¡}t dddddg| ¡}t dd ddg¡}t|||d}t|| t dddddg¡}t ddddddg| ¡}t dd dddg¡}t|||d}t|| t ddg¡}t ddg| ¡}t d d g¡}t|||d}t|| qt dg¡}t dg¡}t d g¡}t|||d}t|| |dkrÊttg g |dg d S )N©r r. r r r r r r TF©r rK r, r )r r, rR > NrL ©r r r r )r rK Úmultr r r r r r r Ú test_in1d sp
|
|
|
|
& ÿ
|
ÿ
|
|
|
|
|
zTestSetOps.test_in1dc
|
C sZ t ddddddddg¡}t ddg¡}t ddddddddg¡}t||}t|| d S )Nr r r ra ÚeTFrg r6 r r r Útest_in1d_char_arrayA s
|
|
z TestSetOps.test_in1d_char_arrayc C sê dD ]^}t ddddddddddddddg¡}dddg| }tt t|||d¡t||d|d q|d
|
krædD ]t}t jddddddddddddddgt jd }dddg| }t j|t jd }tt t|||d¡t||d|d qpd S ) z Test in1d's invert parameterre r r r r r rR T©ÚinvertrK > NrL r< N)r r r rm r Úfloat32)r rK rh r r r r r Útest_in1d_invertJ s &ÿ"ÿÿzTestSetOps.test_in1d_invertc C s¬ t d¡ dd¡}t dd¡ dd¡}t dd¡ dd¡}t ddddddg¡}tt||d|d | tt||d|d | tt||d|d | tt||d|d | d S )
|
Nr* r r r0 é? rY FTrf )r r! r^ r r r )r rK r r Zlong_br r r r Útest_in1d_ravel_ s, ÿ
|
ÿþ
|
ÿþ
|
ÿþzTestSetOps.test_in1d_ravelc C sÞ t jdddddddgt jd}t jddddgt jd}t jdddddddgtd}t|t|| tt |¡t||d d
|
t jdd ddgt jd}t jddddddgt jd}t d d d d g¡}t||d d }t|| dS )z(Hit the standard isin code with integersr r r g eÍÍAr< r r r T©rm r Fr N)r r r_ r] r r rm )r r r rE r r r r r Ú!test_in1d_hit_alternate_algorithmt s z,TestSetOps.test_in1d_hit_alternate_algorithmc C s` t ddg¡}t dddg¡}t ddg¡}t|t|||d tt |¡t||d|d dS )z&Test that in1d works for boolean inputTFrR rl N)r r r r rm )r rK r r rE r r r Útest_in1d_boolean s ÿ
|
ÿz TestSetOps.test_in1d_booleanc C sb t j d¡}|jdddd}|jdddd}t||}| d¡}| d¡}t|t|||d dS )z(Test that in1d works for timedelta inputr éd r. )Úsizeútimedelta64[s]rR N)r ÚrandomZ RandomStateÚrandintr Úastyper )r rK Zrstater r ÚtruthZ a_timedeltaZ b_timedeltar r r Útest_in1d_timedelta s
|
|
|
z TestSetOps.test_in1d_timedeltac C s@ t jdddgdd}|}t t¡ t||dd W 5 Q R X d S )Nr r r rw r< rM rR )r r ÚpytestÚraisesÚ
|
ValueErrorr ©r r r r r r Ú test_in1d_table_timedelta_fails s z*TestSetOps.test_in1d_table_timedelta_failsz dtype1,dtype2c C sÞ t |t j¡}t jddddg|d}|r>t jdddg|d}nt jdddg|d}t ddddg¡}|d kot|t jko|t jk|t jko|t jkf}|rÆtjt d
|
d t
|
|||d W 5 Q R X nt t
|
|||d | d S )z7Test that in1d works as expected for mixed dtype input.r r r< iÿÿÿé éÿ TFrM zexceed the maximum)ÚmatchrR N) r Z
|
issubdtypeZ signedintegerr ÚanyÚint8Úint16r} r~ Ú RuntimeErrorr r ) r Zdtype1Zdtype2rK Zis_dtype2_signedÚar1Úar2rE Zexpect_failurer r r Útest_in1d_mixed_dtype¡ s
|
þz TestSetOps.test_in1d_mixed_dtypec C s t jd D ]}t jdddgtd}t jddddg|d}t jdddgtd}tt|||d| || }}t jddddgtd}tt|||d| q
|
dS )z4Test that in1d works as expected for bool/int input.Ú
|
AllIntegerTFr< r rR N)r Ú typecodesr r] r r )r rK r= r r rE r r r Útest_in1d_mixed_booleanÄ s
|
z"TestSetOps.test_in1d_mixed_booleanc C s< d g}t dgd ¡}t dg¡}t ||¡}t|| d S ©Nr r. Frg ©r r r rE Úresultr r r Ú test_in1d_first_array_is_objectÑ s
|
z*TestSetOps.test_in1d_first_array_is_objectc C s: d}t d gd ¡}t dg¡}t ||¡}t|| d S r rg r r r r Ú test_in1d_second_array_is_objectØ s
|
z+TestSetOps.test_in1d_second_array_is_objectc C s< d g}t d gd ¡}t dg¡}t ||¡}t|| d S )Nr. Trg r r r r Ú test_in1d_both_arrays_are_objectß s
|
z+TestSetOps.test_in1d_both_arrays_are_objectc C s` t dtfdtfg¡}t jdg|d}t jdgd |d}t dg¡}t ||¡}t|| d S )NZfield1Zfield2)r Nr< r. T)r r= ÚintÚobjectr r r )r Údtr r rE r r r r Ú+test_in1d_both_arrays_have_structured_dtypeæ s z6TestSetOps.test_in1d_both_arrays_have_structured_dtypec C sf t jddgtd}t jddgtd}t ddg¡}t ||¡}t|| t j||dd}t|t |¡ t jdddgtd}|d d
}t jdddgtd}|d d
}t ddg¡}t ||¡}t|| t j||dd}t|t |¡ t jdd dgtd}|d d
}t jddgtd}t dd
|
g¡}t ||¡}t|| t j||dd}t|t |¡ d S ) N)r r r< Trr )r r r r: ©r r F)r r r r r rm r r r r Ú'test_in1d_with_arrays_containing_tuplesð s0
|
|
|
z2TestSetOps.test_in1d_with_arrays_containing_tuplesc C s" t dddddg¡}t dddddg¡}ttt||d d
|
t jdd dd dgtd }t jdd dd dgtd }ttt||dd
|
t jt jfD ]}t jdddddg|d }t jdt |¡j g|d }tt
|
t||dd
|
t j||dd
|
}t |dgdgd t j||dd
|
}t |dgdgd qdS )z&Test that in1d raises expected errors.r r r r r r* r, r. Z quicksortrR r r r< rM r: NTFrL ) r r r r r r Úint32r_ ZiinfoÚmaxr r )r r r Zobj_ar1Zobj_ar2r= Z overflow_ar2r r r r Útest_in1d_errors s( þzTestSetOps.test_in1d_errorsc C s t dddddg¡}t dddddddg¡}t ddddddg¡}t||}t|| t dddgdddgg¡}t dddddg¡}t ddddddg¡}t||}t|| tg tg g d S )Nr r r r r r r )r r r r ) r r r r r rS rT ZezÚzr r r Ú test_union1d3 s
|
|
|
|
zTestSetOps.test_union1dc
|
C sÀ t ddddddddg¡}t dddddddg¡}t ddg¡}t||}t|| t d¡}t d ¡}t d d
|
g¡}t||}t|| tg tg g t d t j¡}tt|g jt j d S ) Nr* r r r r r r é é rX r )r r r
|
r r! Zuint32r r= r6 r r r Útest_setdiff1dE s
|
|
|
|
|
|
zTestSetOps.test_setdiff1dc C sJ t dddg¡}t dddg¡}t ddg¡}t||dd}t|| d S )Nr r r r r Tr )r r r
|
r )r r r rE rI r r r Útest_setdiff1d_uniqueW s
|
z TestSetOps.test_setdiff1d_uniquec C s< t dddg¡}t dddg¡}tt||t dg¡ d S )Nr r r Ús)r r r r
|
r r r r Útest_setdiff1d_char_array^ s z$TestSetOps.test_setdiff1d_char_arrayc C sb t dddddg¡}t dddddddg¡}t||}t||}t||}t||}t|| d S ) Nr r r r r, r0 r r )r r r r r r
|
r )r r r Zc1Zaux1Zaux2Úc2r r r Ú test_manywaysc s
|
|
|
|
zTestSetOps.test_manyways).r$ r% r& r r) r5 r7 r; r} ÚmarkÚ parametrizer r r_ Únanrn r rG r r` r rJ rd ri rk ro rq rs rt r| r Úuint8Zuint16r r r r r r r r r r¢ r£ r¥ r§ r r r r r s° 'ýýýî
|
ýýýýóþ
|
|
<
|
=
|
|
|
|
|
|
|
|
|
|
|
|
úþ
|
|
!"r c @ sn e Zd Zdd Zdd Zdd Zdd Zej d d
|
d g¡d d Z
|
dd Z dd Z dd Z dd Zdd ZdS )Ú
|
TestUniquec C s¸ dd }dddddddgd }ddddg}ddd dg}ddd dd ddgd }t ddddgd¡}g }| t jd
|
¡ | t jd ¡ | d ¡ | d ¡ |D ].}t ||¡} t ||¡}
|
|| |
|
|||| qd}t t||¡} || d d
< t t||¡}
|
||
|
d d
< || |
|
|||| ddg}t tt |||¡} t tt |||¡}
|
|| |
|
|||| dddg} t
|
t | ¡ddg dddg}dddg} d dd dddg} t |} t
|
| | t |dd\}}t
|
|| t
|
|| t d¡}d|d< t j |dd\}}t
|
|t d¡ g }t j |ddd }t j |ddd }t j |ddddd
\}}t|jt j t|jt j t|jt j t|jt j dt jdt jg}ddt jg}dd dg}ddd dg}dddg}tt |¡| tt j |dd||f tt j |dd||f tt j |dd ||f d t jd td t jtdt jg}d d td t jg}dd dg}ddd ddg}dddg}tt |¡| tt j |dd||f tt j |dd||f tt j |dd ||f t d ¡}t d!¡|t d"¡|g}t d"¡t d!¡|g}dd dg}ddd dg}dddg}tt |¡| tt j |dd||f tt j |dd||f tt j |dd ||f t d ¡}t dd#¡|t dd$¡|g}t dd$¡t dd#¡|g}dd dg}ddd dg}dddg}tt |¡| tt j |dd||f tt j |dd||f tt j |dd ||f t jgd% }t jg}d g}d d d d g}d%g}tt |¡| tt j |dd||f tt j |dd||f tt j |dd ||f d S )&Nc S sè 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_indexTFÚreturn_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)rA r r ) r r r2 r3 r r Zbase_msgrF ÚvÚjZj1Zj2Zj3r r r Ú check_allr sN ÿ z,TestUnique.test_unique_1d.<locals>.check_allr r r r r. r r r ÚAllFloatú datetime64[D]útimedelta64[D]ÚO)Ú Úiy ð? y ð? ð¿)r r r T)r¯ r¸ .)r® )r® r¯ g @r> )r° y @ ð¿y ð? ð?ç Únatz
|
2020-12-26z
|
2020-12-24ÚDÚhr )r ÚmultiplyÚextendr rD r ÚemptyÚlenÚlistÚzipr r Z chararrayÚzerosr r= Zintprª ÚcomplexZ
|
datetime64Z timedelta64)r r³ r r r2 r3 r Útypesr ZaaZbbÚunqÚinvZa1Za2Za2_invZa1_idxZa3_idxZa3_invZuaZua_idxZua_invZua_cntr» Zall_nansr r r Útest_unique_1dp sÄ 0
|
|
|
|
|
|
|
|
|
ÿÿ
|
|
|
|
|
|
|
|
|
|
zTestUnique.test_unique_1dc C s\ t t| jt t t| jdtfdtfg t tjtt d¡dd t tjtt d¡dd d S )Nr r r. r ©Úaxiséþÿÿÿ) r rB Ú_run_axis_testsr r r Z AxisErrorr r! r" r r r Útest_unique_axis_errors s ÿz"TestUnique.test_unique_axis_errorsc C s^ d}dddgdddgg}t |¡}tt|ddt|dd| tt|ddt|dd| d S )Nz Unique failed on list of listsr r rÊ )r rN r r )r rF ÚinpZinp_arrr r r Útest_unique_axis_list! s
|
|
z TestUnique.test_unique_axis_listc C s g }| tjd ¡ | tjd ¡ | d¡ | d¡ | dtfdtfg¡ | dtfdtfg¡ |D ]}| |¡ qhd}tjdtjd d
|
d ¡
|
t ¡}tj d d gd d ggt d }t t|dd|| d}t ddgddgddgddgg¡}t ddgg¡}t t|dd|| d S )Nr r´ rµ r¶ r r z&Non-bitwise-equal booleans test failedr. r< r: r FTr rÊ z"Negative zero equality test failedg rº )r¿ r r rD r ÚfloatrÍ r! r« r^ Úviewr] r r r )r rÆ r= rF Údatar r r r Útest_unique_axis( s"
|
|
"zTestUnique.test_unique_axisrË r r: c
|
C s< t ddddddddg¡}t||d}t|ddddg d S )Nr r r r rÊ ©r r r r )r rË rS Úuniqr r r Útest_unique_1d_with_axis> s z#TestUnique.test_unique_1d_with_axisc
|
C sV t jdt jd}t|ddddd\}}}}t|j|j t|t jdd t|t dg¡ t|t ddg¡ t|t dg¡ t|d dddd\}}}}t|j|j t|t jdd t|t g ¡ t|t g ¡ t|t g ¡ d
|
}t j|d}tt |D ]H}t
|
|} || dkr,d| |< nd | |< tt||d t j| d qd S ) N)r r )r1 r= r T©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 ÚrangerÁ r )
|
r Z single_zerorÖ ÚidxrÈ Úcntr1 Zmultiple_zerosrË Zexpected_shaper r r Útest_unique_axis_zerosD s: ÿ ÿ
|
|
ÿz!TestUnique.test_unique_axis_zerosc C s t jdddddddddddddddgdd}t j |d¡}t |¡}t j|d d d
|
\}}}d }t|j|j| t|j|j| d S ) Né@ r r r r rp r« r< T)r® r° z6Unique returned different results when asked for index)r r ÚmaZ masked_equalr r rÓ Úmask)r rS rT r± Zv2r¹ r rF r r r Útest_unique_maskedh s $ÿ
|
z TestUnique.test_unique_maskedc C sD d}dD ]6}t dgdgg|¡}t j|dd}t|||| qd S )Nz*sort order incorrect for integer type '%s'Zbhilqr: r rÊ rÕ )r Úfmtr r r r r r Ú test_unique_sort_order_with_axisu s
|
z+TestUnique.test_unique_sort_order_with_axisc
|
C s t ddddgddddgddddgddddgg¡ |¡}d}t ddg¡}tt|| |¡| d}t ddddgddddgg¡}tt|dd| |¡| d}t dddgdddgdddgdddgg¡}tt|dd| |¡| d}t ddgddggddgddggg¡ |¡}t j|ddgdd}tt|dd|| t|dd d d d
|
\}}}} d }t|| || d }t|| | d }t| t ddg¡| t|dd d d d
|
\}}}} d}t|d d
|f | d}t|d d
|f | d}t| t dddg¡| d S )Nr r z&Unique with 1d array and axis=0 failedz&Unique with 2d array and axis=0 failedrÊ z&Unique with 2d array and axis=1 failedz&Unique with 3d array and axis=2 failedr TrØ 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 rz r r Ztake)
|
r r= rÓ rF r Zdata3drÖ rÚ rÈ rÛ r r r rÍ ~ s`
|
|
|
ýý *
|
ÿÿþý ÿ ÿzTestUnique._run_axis_testsc C sj t ddt jt jt jg¡}t |¡}t j|dd}t|t dt jg¡ t|t dt jt jt jg¡ d S )Nr F)Z equal_nan)r r rª r r )r r rÇ Znot_unqr r r Útest_unique_nanequalsª s
|
|
z TestUnique.test_unique_nanequalsN)r$ r% r& rÉ rÎ rÐ rÔ r} r¨ r© r× rÜ rà râ rÍ rã r r r r r¬ n s *
|
$ ,r¬ )Ú__doc__Únumpyr Z numpy.testingr r r r Znumpy.lib.arraysetopsr r r r r
|
r r r r} r r¬ r r r r Ú<module> s ( d
|
