zmc
2023-12-22 9fdbf60165db0400c2e8e6be2dc6e88138ac719a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
U
­ý°dvã@sJddlZddlZddlZddlmZmZmZddlm    Z
Gdd„dƒZ dS)éN)Ú    DataFrameÚ
MultiIndexÚSeriesc@sÆeZdZdd„Zej dgeede     
g¡g¡dd„ƒZ dd„Z d    d
„Z d d „Zd d„Zdd„Zdd„Zdd„Zdd„Zdd„Zej deddddgiddedddgddg¡dd „ƒZd!d"„Zd#S)$ÚTestDataFrameIsIncsrtddddgddddgddd    dgd
œd d d dgd‰ddd    g‰ˆ ˆ¡}t‡‡fdd„ˆjDƒƒ}t ||¡dS)NééééÚaÚbÚfÚnÚc©ÚvalsÚidsZids2ÚfooÚbarÚbazÚqux©Úindexcsg|]}ˆj| ˆ¡‘qS©)ÚlocÚisin)Ú.0Ús©ÚdfÚotherrú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/methods/test_isin.pyÚ
<listcomp>sz/TestDataFrameIsIn.test_isin.<locals>.<listcomp>)rrrÚtmÚassert_frame_equal)ÚselfÚresultÚexpectedrrr Ú    test_isins
 
 
ý
ú
 
zTestDataFrameIsIn.test_isinÚempty)ZdtypecCsDtdddgdddgdœƒ}td|j|jƒ}| |¡}t ||¡dS)Nr
r rÚer ©ÚAÚBF)rrÚcolumnsrr"r#)r$r(rr&r%rrr Útest_isin_emptys
z!TestDataFrameIsIn.test_isin_emptycCs¬tdddgdddgdœƒ}ddgi}td|j|jƒ}d    |jd
<| |¡}t ||¡tdddgdddgdœƒ}ddg|_td|j|jƒ}d    |jd
<| |¡}t ||¡dS) Nr
r rr)r r*r+FT)rr+)rrr-rrr"r#)r$rÚdr&r%rrr Útest_isin_dict's
 
 
 
 
 
z TestDataFrameIsIn.test_isin_dictc    Cs„tddddgddddgddd    dgd
œd d d dgd}d}tjt|d| d¡W5QRXtjt|d| d¡W5QRXdS)Nrrrr    r
r r r rrrrrrrzfonly list-like or dict-like objects are allowed to be passed to DataFrame.isin\(\), you passed a 'str'©ÚmatchZaaa)rÚpytestÚraisesÚ    TypeErrorr)r$rÚmsgrrr Útest_isin_with_string_scalar9s
 
 
ý
ú    ÿz.TestDataFrameIsIn.test_isin_with_string_scalarcCs¶tddddgdtjddgdœƒ}tddddgdtjddgdœƒ}td    |j|jƒ}| |¡}d
|jddgd f<d
|jddgd f<t ||¡d d g|_| |¡}d    |d <t ||¡dS)Nrrrr    r*ré éFTr+r,ÚC)    rÚnpÚnanrr-rrr"r#)r$Údf1Údf2r&r%rrr Ú test_isin_dfMs  
 
 
zTestDataFrameIsIn.test_isin_dfcCsbtdddgdddgdœƒ}tt|d|d    ƒƒ|d
<|d
 d g¡}t |td d d gd
d¡dS)Nrrrr
r r r*r+r,r:)rr
TF)Úname)rÚlistÚziprr"Zassert_series_equalr)r$rr%rrr Útest_isin_tuples\sz"TestDataFrameIsIn.test_isin_tuplesc    Csütddddgdtjddgdœƒ}tddgddgdtjgddggd    d    gd
}d }tjt|d | |¡W5QRXtddgddgdtjgddggd d    gddddgd}tjt|d | |¡W5QRXd    d    g|_tjt|d | |¡W5QRXdS)Nrrrr    r*rr8r9r,©r-z+cannot compute isin with a duplicate axis\.r1r+©r-r)rr;r<r3r4Ú
ValueErrorrr-)r$r=r>r6rrr Útest_isin_df_dupe_valuescs *
ý
z*TestDataFrameIsIn.test_isin_df_dupe_valuescCs€tddddgddddgdœƒ}tddgddgddggddgd}| |¡}td|j|jd}d|jd<d|jd    <t ||¡dS)
Nrrr*r+rDF©rr-T)rr)rrrr-rZilocr"r#)r$rrr%r&rrr Útest_isin_dupe_selfys"
 
 
z%TestDataFrameIsIn.test_isin_dupe_selfcCsˆtddddgdtjddgdœdddd    gd
}tddd dgdddd    gd
}td |j|jd }d|jd<d|jd    <| |¡}t     ||¡dS)Nrrrr    r*r
r rr/ré FrHT)r
r+)
rr;r<rrr-rrr"r#)r$rrr&r%rrr Útest_isin_against_series‚s
ÿ
 
 
z*TestDataFrameIsIn.test_isin_against_seriescCst ddddddddd    d
d d g ¡}tt d ¡t d ¡dœ|d}tddddddddddddg ddddddddddddg dœƒ}td|j|jd}| |¡}t     
||¡||_|j   t ¡}|dd…df|dd…df<t|ddg|d}| |¡}t     
||¡dS)N)rr
r)rr
r)rr r)rr r)rr
r)rr
r)rrr)rrr)rr r)rr r)rrr)rrrr8r*rrrFrHr+r,rE)rÚ from_tuplesrr;ZonesÚzerosrr-rr"r#ÚvaluesZastypeÚbool)r$Úidxr=r>r&r%rrr Útest_isin_multiIndexs<ôÿþÿ
 
z&TestDataFrameIsIn.test_isin_multiIndexcCs´tdt ddg¡iƒ}tdt dd¡t dd¡giƒ}tdgiƒ}tƒ}tdddgiƒ}| |¡}t ||¡| |¡}t ||¡| |¡}t ||¡| |¡}t ||¡dS)NÚdatez
2014-01-01z
2014-01-02rrrF)rÚpdÚ to_datetimeZ    Timedeltarr"r#)r$Zdf1_tsZdf1_tdr>Zdf3r&r%rrr Útest_isin_empty_datetimelike²s  
 
 
 
z.TestDataFrameIsIn.test_isin_empty_datetimelikerNr
rrrÚcategorycCsNtdddgdddgdœƒ}tdddgd    d    d    gdœƒ}| |¡}t ||¡dS)
Nrrrr    r9é)r
r TF)rrr"r#)r$rNrr&r%rrr Útest_isin_category_frameÅs    
z*TestDataFrameIsIn.test_isin_category_framecCsRt dddg¡}|jddtdddgƒ}| |¡}tdddgƒ}t ||¡dS)NrrrF)ÚwriteT)r;ÚarrayZsetflagsrrr"r#)r$Zarrrr%r&rrr Útest_isin_read_onlyÔs  
z%TestDataFrameIsIn.test_isin_read_onlyN)Ú__name__Ú
__module__Ú __qualname__r'r3ÚmarkZ parametrizerÚobjectr;rZr.r0r7r?rCrGrIrKrQrUrrXr[rrrr r s(
     %þþ
r) Únumpyr;r3ZpandasrSrrrZpandas._testingZ_testingr"rrrrr Ú<module>s