U
|
ý°de ã @ s~ d dl Zd dlZd dlZd dlmZmZmZmZm Z m
|
Z
|
d dl m Z d dlmZmZ ej dddg¡ej ddd d
|
d g¡d d Zej de eje
|
eg¡dd Zejjej de eje
|
eg¡dd Zdd Zdd Zdd Zdd Z dd Z d d Z d d Z d!d" Z d#d$ Z!d%d& Z"d'd( Z#d)d* Z$ej ddd d
|
d g¡d+d, Z%d-d. Z&d/d0 Z'd1d2 Z(ej d3e)de)d g¡d4d5 Z*d6d7 Z+d8d9 Z,d:d; Z-d<d= Z.d>d? Z/d@dA Z0ej dd dd
|
d g¡dBdC Z1ej dDej2dEg¡dFdG Z3ej dDej2dHg¡dIdJ Z4ej dKdLgdLgfdLdLgdLgfdMdLgdMdLgfg¡dNdO Z5ej dPdQdddgfdQdRgddgfdQdSgddgfdTdUdVgddgfdTdSdRgddgfdWdRdSdUgddgfdTdRdUgdRdgfdTdUdSgddSgfdTdRdSgdRdSgfdTddSgddSgfg
|
¡dXdY Z6dZd[ Z7d\d] Z8d^d_ Z9d`da Z:ej dbdcddg¡ej deddfg¡dgdh Z;ej dDdidjg¡dkdl Z<ej dmdnej2g¡dodp Z=dqdr Z>dsdt Z?dudv Z@ej dwdRdSdUgd dxggdUdVdSgdxggd djdygdjdjdjggd djdygd d d ggdzd{gfg¡d|d} ZAej dRejBdRdSgdRdSgd~dRdSgg¡ej dSejBdRdSgdSdRgddejBdRdSgdSdRgd~g¡dd ZCej dDej2dEg¡dd ZDd
d ZEdS )é N)ÚCategoricalIndexÚ DataFrameÚIndexÚ IntervalIndexÚ
|
MultiIndexÚSeries)Úis_float_dtypeÚis_unsigned_integer_dtypeÚcaseg à?ZxxxÚmethodÚ intersectionÚunionÚ
|
differenceÚsymmetric_differencec C s4 d}t jt|d t| |||d W 5 Q R X d S )Nz!Input must be Index or array-like©Úmatch©Úsort)ÚpytestÚraisesÚ TypeErrorÚgetattr)Úidxr
|
r r Úmsg© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_setops.pyÚtest_set_ops_error_cases s r Úklassc C s | dd d
}| d d
}|t k r,||j}|j||d}|d krL| ¡ }n|}t ||¡ d}tjt|d |jdddg|d W 5 Q R X d S ) Né éÿÿÿÿé r ú.other must be a MultiIndex or a list of tuplesr é é ) r Úvaluesr Ú sort_valuesÚtmÚassert_index_equalr r r )r r r ÚfirstÚsecondZ intersectÚexpectedr r r r Útest_intersection_base s
|
|
r+ c C s | d d d
}| d d
}|t k r,||j}|j||d}|d krL| ¡ }n|}t ||¡ d}tjt|d |jdddg|d W 5 Q R X d S ) Nr r r r! r r" r r# ) r r$ r r% r& r' r r r )r r r r( r) r r* r r r r Útest_union_base3 s
|
|
r, c sÄ | dd
| d d
}| j |d}|d kr6| ¡ }| |¡sDtt ||¡ fddtjtt fD }|D ] }| j ||d}t ||¡ qnd}t
|
j t |d | j ddd g|d W 5 Q R X d S )
|
Né r c s g | ]}| j qS r ©r$ ©Ú.0r ©r) r r Ú
|
<listcomp>T s z(test_difference_base.<locals>.<listcomp>r! r r" r r# ) r r% ÚequalsÚAssertionErrorr& r' ÚnpÚarrayr Úlistr r r )r r ÚanswerÚresultÚcasesr
|
r r r1 r Útest_difference_baseH s r; c s | dd
}| d d
| ddg }|j |d}|d krB| ¡ }t ||¡ fddtjttfD }|D ] }|j ||d}t ||¡ qld}tj t
|
|d |j dd d
|
g|d W 5 Q R X d S ) Nr" r r r c s g | ]}| j qS r r. r/ r1 r r r2 j s z-test_symmetric_difference.<locals>.<listcomp>r! r r r# ) r r% r& r' r5 r6 r r7 r r r )r r r( r8 r9 r: r
|
r r r1 r Útest_symmetric_difference^ s r< c C sj t jddgddggddgd} | | ¡}|j| jks8t| ¡ ddg¡}| |¡}|jd d gksftd S )NÚaÚbÚAÚB©Únames)r Ú from_productr rB r4 ÚcopyÚrename)r r9 Úidx2r r r Ú$test_multiindex_symmetric_differencet s
|
|
rG c C s | j r
|
t| d d
j s td S )Nr )Úemptyr4 ©r r r r Ú
|
test_empty s
|
rJ c C sl | }|j | dd
|d}| d d
j}|d kr8t|}tj|d| jd}t|tsXt| |¡sft|j| jksvtt
|
||¡ | j | |d}| d d
}| |¡sªt|j| jksºt| dd
j | |d}| d d
}| |¡sêt|j| jksút| d d
j | |d}| d d
}| |¡s,t|j| jks>t| dd
}ddg|_|j ||d}|jdksrt| j | d¡d |d}t |dkst|j |j|d}| |d d
¡sÄt|j g |d}| |¡sât|j|jksôt|j d g|d}t d
|
d d d dg¡}|j|_|j|jks2td}t jt|d |j dddddg|d W 5 Q R X d S )Néýÿÿÿr r )Z sortorderrB ÚfooÚbaz©NNr" ©rL Úone)ÚbarrP )rM Útwo)rL rR )ÚquxrP )rS rR r! r r r# r- r )r r$ Úsortedr Ú from_tuplesrB Ú
|
isinstancer4 r3 r& r' Z sortlevelÚlenr r r )r r r( r9 Úvalsr* Zchunkletr r r r Útest_difference
sT
|
ÿrY c C s0 t ddgddgg¡} | g ¡}t || ¡ d S )Nr" r r= r> ©r rC r r& r' ©r r9 r r r Ú test_difference_sort_specialÆ s
|
r\ c C sJ t ddgddgg¡} | jg dd}t ddgddgg¡}t ||¡ d S ©Nr" r r= r> Tr rZ )r r9 r* r r r Ú!test_difference_sort_special_trueÎ s r^ c C s t dt d¡dgddgg¡} t dt d¡dgdd gg¡}d
|
}tjt|d | |¡}W 5 Q R X t || ¡ | j|d d }t || ¡ d S )Nr" Ú2000r r= r> r# r- ÚcÚdz0sort order is undefined for incomparable objectsr Fr ) r rC ÚpdÚ Timestampr& Úassert_produces_warningÚRuntimeWarningr r' )r Úotherr r9 r r r Ú!test_difference_sort_incomparableÖ s rg c C sl t dt d¡dgddgg¡} t dt d¡dgdd gg¡}d
|
}tjt|d | j|d d W 5 Q R X d S )Nr" r_ r r= r> r# r- r` ra zH'values' is not ordered, please explicitly specify the categories order r Tr )r rC rb rc r r r r )r rf r r r r Ú&test_difference_sort_incomparable_trueæ s
|
rh c C sà | d d
d d d
}| dd
}|j ||d}|d krHt || ¡ ¡ t || ¡sXt| j | |d}t || ¡ | j | d d
|d}t || ¡ | j}| d d
j |dd
|d}|d krÎt || ¡ n| | ¡sÜtd S )Nr r r# r r r- )r r& r' r% Ú equalContentsr4 r$ r3 )r r Úpiece1Úpiece2Z the_unionÚtuplesr9 r r r Ú
|
test_unionñ s rm c C sj t dddg}| | ¡}d|ks$td|ks0td}tjt|d | |¡}W 5 Q R X | |¡rftd S )Nr? r@ ÚCrO z'The values in the array are unorderabler )r r r4 r& rd re r3 )r rf r9 r Zresult2r r r Ú test_union_with_regular_index s
|
ro c C sØ | d d
d d d
}| dd
}|j ||d}|d krLt || dd
¡ t || dd
¡sdt| j | |d}t || ¡ | d d
j | dd
|d}| d d
}| |¡s¶t| j}| |¡}| | ¡sÔtd S )Nr r r# r r r )r r& r' ri r4 r3 r$ ) r r rj rk Zthe_intrH r* rl r9 r r r Útest_intersection s
|
rp c C s | ¡ d¡}d g| j }t| |||d}t| || |d |¡}t ||¡ t| ||d d
|d}t| || d d
|d |¡}t ||¡ d S )NÚcategoryr r )Z to_flat_indexÚastypeÚnlevelsr rE r& r' )r r r rf Z res_namesr9 r* r r r Útest_setop_with_categorical2 s rt c C s t tddd}| j||d}t| jg g| j d d}tj||dd | jt |¡d d
|d}t| jg g| j | j
|
d}tj||dd d }t j t |d
|
| jt |¡|d W 5 Q R X d S ) Nr# rL )Únamer ©ÚlevelsÚcodesrB T)Úexactr r! r )r Úranger r rw rs r& r' r5 ZasarrayrB r r r )r r rf r9 r* r r r r Ú test_intersection_non_objectB s r{ c C sF t ddgddgg¡} t | j| dd| ¡ t | j| d d| ¡ d S ©Nr" r r= r> Fr )r rC r& r' r rI r r r Útest_intersect_equal_sortT s r} c C sJ t ddgddgg¡} t ddgddgg¡}| j| dd}t ||¡ d S r] )r rC r r& r' )r r* r9 r r r Ú test_intersect_equal_sort_true[ s r~ Úslice_c C s\ t ddgddgg¡}|| }t | |¡|¡ t | |¡|¡ t |j|dd|¡ d S r| )r rC r& r' r )r r rf r r r Útest_union_sort_other_emptyb s
|
r c C sV t ddgddgg¡} | d d
}| j|dd}t ddgddgg¡}t ||¡ d S r] )r rC r r& r' )r rf r9 r* r r r Ú test_union_sort_other_empty_sortp s
|
r c C sv t dt d¡gddgg¡} t t¡ | | d d
¡}W 5 Q R X t || ¡ | j| d d
dd}t || ¡ d S )Nr" r_ r= r> Fr ) r rC rb rc r& rd re r r' r[ r r r Ú"test_union_sort_other_incomparablez s r c C sT t dt d¡gddgg¡} d}tjt|d | j| d d
dd W 5 Q R X d S ) Nr" r_ r= r> z<'<' not supported between instances of 'Timestamp' and 'int'r Tr )r rC rb rc r r r r )r r r r r Ú'test_union_sort_other_incomparable_sort s r c C sL t ddgddgg¡} | jd }d}tjt|d | |¡ W 5 Q R X d S )Nr= r> r" r z<Can only union MultiIndex with MultiIndex or Index of tuplesr )r rC rw r r ÚNotImplementedErrorr )Úmir r r r r Ú"test_union_non_object_dtype_raises s
|
|
r c C sZ t g g¡} t jddgddggddgd}| |¡}t ddgddgg¡}t ||¡ d S )Nr" r r# r- r= r> rA ©r Ú from_arraysr r& r' )r
Úmi2r9 r* r r r Ú%test_union_empty_self_different_names s
|
|
r c C s` t jddgddggddgd} t d¡}| |¡}tj| |d d
|
| | ¡}tj| |d d
|
d S ) Nr" r r# r- r= r> rA r F)Z check_names)r r rb Z
|
RangeIndexr r& r' )r
ÚriZ result_leftZ result_rightr r r Ú&test_union_multiindex_empty_rangeindex£ s
|
|
|
r c C sn t ddgddgg¡}t ddgddgg¡}tjtdd t|| |dd W 5 Q R X t|| |d d d S )
|
Nr= r> r" r r` z The 'sort' keyword only takesr r T)r rC r r Ú
|
ValueErrorr )r Zidx1rF r r r Útest_setops_sort_validation¯ s
|
r Úvaléd c C s t jtddg| dddggdd gd}t tdd|g| ddddgg¡}| |¡}t tdg| ddgg¡}t ||¡ | |jdd¡}t jtg | dtg tjdgdd gd}t ||¡ d S ) Nr" r ©Údtyper= rA r# F)Z ascending) r r r r r& r' r% r5 Úint64©Úany_numeric_ea_dtyper ÚmidxÚmidx2r9 r* r r r Ú test_difference_keep_ea_dtypes½ s ÿÿ
|
þr r c C s t jtddg| dddggdd gd}t tdd|g| ddddgg¡}| |¡}t tdd|g| ddddgg¡}t ||¡ d S ©Nr" r r r= rA r# )r r r r r& r' r r r r Ú*test_symmetric_difference_keeping_ea_dtypeÒ s ÿÿ
|
ÿr )rl Ú
|
exp_tuples©Zval1Ztest1©Zval2Ztest2c C sV t j| ddgd}t jdddgddgd}| |¡}t j|ddgd}t ||¡ d S )Nr( r) rA r r )r rU r r& r' )rl r ÚleftÚrightr9 r* r r r Ú test_intersect_with_duplicatesâ s þ
|
r zdata, names, expected©r" r= r> )r" r r` ra )r" r r# c C s< t jg ddgd}t j| g|d}| |¡}||ks8td S )Nr= r> rA )r rU Z_maybe_match_namesr4 )ÚdatarB r* r
r r9 r r r Útest_maybe_match_namesù s
|
r£ c C st t jddgddggddgd} t jddgddggddgd}| |¡}t jddgddggd dgd}t ||¡ d S ) Nr" r r# r- r` r> rA r= ©r r r r& r' )Úmi1r r9 r* r r r Ú'test_intersection_equal_different_names s
|
|
r¦ c C sF t jdgdggddgd} t dgdgg¡}| |¡}t ||¡ d S )Nr" r# r` r> rA r¤ )r
r r9 r r r Ú!test_intersection_different_names s
|
r§ c C sh t d| d| gddddgg¡}t d| dgdddgg¡}| |¡}t d| gddgg¡}t ||¡ d S )Nr# r- r" r r¤ ©Z nulls_fixturer¥ r r9 r* r r r Ú3test_intersection_with_missing_values_on_both_sides" s
|
|
r© c C sN t d| gg¡}t d| dgg¡}| |¡}t dd| gg¡}t ||¡ d S )Nr" r# r r¨ r r r Ú,test_union_with_missing_values_on_both_sides+ s
|
|
rª r Úfloat64ZFloat64r Fc C s t tjdtjg| dddgg¡}t tjdtjdg| ddddgg¡}|j||d}|d krt tjddtjg| ddddgg¡}n|}t ||¡ d S )Nç ð?r r r# g @r- r ) r r rb r6 r5 Únanr r& r' )r r r¥ r r9 r* r r r Ú test_union_nan_got_duplicated4 s "& ÿr® r- r" c C s° t |dg| d}t ddg| d}tj|ddggdd gd}t |ddgg¡}| |¡}|dkrt t dddg| ddddgg¡}n t t ddg| dddgg¡}t ||¡ d S )Nr r r" r= rA r- ©r r r r r& r' )r r Úarr1Úarr2r r r9 r* r r r Útest_union_keep_ea_dtypeD s
|
ÿÿr² Údupe_valr# c C s t td| dg|dtd| dg|dg¡}t td| | g|dtd| | g|dg¡}| |¡}t tdd| | g|dtdd| | g|dg¡}t ||¡ d S )Nr" r r )r r r r r& r' )r³ r r¥ r r9 r* r r r Ú(test_union_with_duplicates_keep_ea_dtypeX s" þÿþÿ
|
þÿr´ c C s | j st| ttfrd S | ¡ j ¡ }t |dgt | g¡}t |d g| dgt |d g¡}|
|
|¡}| ¡ }t ||¡ t|jd rÆ| d¡dk ¡ rÆ| |jd tj¡|jd g¡}n.t|jd rô| |jd t¡|jd g¡}|
|
|¡}t ||¡ d S )Nr" r l )rH rV r r Úuniquer$ Útolistr r rW r r% r& r' r rw Zget_level_valuesÚallZ
|
set_levelsrr r5 r r Úfloat)ÚindexÚrequestr$ r¥ r r9 r* r r r Útest_union_duplicatesq s* &
|
ÿþÿÿ
|
r» c C s t dddg| d}t ddg| d}tj|dddggdd gd}tj|ddggdd gd}| |¡}tjt dddg| ddddggdd gd}t ||¡ d S )Nr- r" r r r= rA r¯ )Zany_real_numeric_dtyper° r± r r r9 r* r r r Ú test_union_keep_dtype_precision s
|
þr¼ c C s t dtjg| d}t dtjg| d}tj|ddggdd gd}t |ddgg¡}| |¡}t t ddtjtjg| dddddgg¡}t ||¡ d S )Nr- r r" r r= rA )r rb ÚNAr r r r& r' )r r° r± r r r9 r* r r r Ú test_union_keep_ea_dtype_with_na£ s
|
"ÿr¾ z'levels1, levels2, codes1, codes2, namesÚ r Zname1Zname2c C s8 t | ||d}t |||d}| |¡}|jdks4td S )Nrv r )r r Z_lexsort_depthr4 )Zlevels1Zlevels2Zcodes1Zcodes2rB r¥ r Zmi_intr r r Ú test_intersection_lexsort_depth° s
|
rÀ )Ú
|
categoriesT)rÁ Zorderedc
|
C sî ddg}t | |d}t ||d}tj| |gddgd}t |¡ t | ddg¡¡¡}t |¡ t |¡¡}t | ddg¡¡ t |¡¡}t | ddg¡¡ t | ddg¡¡¡} t ||¡ t ||¡ t ||¡ t | |¡ d S )NÚ1Ú2)ÚxÚyrÄ rÅ rA )r r r Z
|
from_framer r% r& r' )
|
r= r> rf Zdf1Zdf2r* Zres1Zres2Zres3Zres4r r r Ú0test_intersection_with_non_lex_sorted_categoriesÄ s$
|
ÿÿÿ rÆ c C s| t jtddg|dddggdd gd}t tdd| g|ddddgg¡}| |¡}t tdg|ddgg¡}t ||¡ d S r )r r r r r& r' )r r r r r9 r* r r r Ú test_intersection_keep_ea_dtypesé s ÿÿ
|
rÇ c C sb t dt d¡d} t dt d¡d}t| |g}ttjtjfddgdtjd gi}t ||¡ d S )
|
Nr¡ )rN )r¹ )é
|
é )rN ©r= r> r¬ g $@rÊ g 4@)r r rU r r5 r r& Zassert_frame_equal)Zseries1Zseries2r9 r* r r r Ú.test_union_with_na_when_constructing_dataframe÷ s
|
"rË )FÚnumpyr5 r Zpandasrb r r r r r r Zpandas._testingZ_testingr& Zpandas.api.typesr r ÚmarkZ parametrizer r6 r7 r+ Zarm_slowr, r; r< rG rJ rY r\ r^ rg rh rm ro rp rt r{ r} r~ Úslicer r r r r r r r r½ r r r r£ r¦ r§ r© rª r® r² r´ r» r¼ r¾ rÀ Z CategoricalrÆ rÇ rË r r r r Ú<module> sê
|
ÿ
|
A
|
ÿ
|
|
|
|
ÿ
|
|
|
|
þýþ
|
öþ
|
|
|
|
# ûÿþ
|
þþþ
|
|
