zmc
2023-08-08 e792e9a60d958b93aef96050644f369feb25d61b
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
U
®ý°dÈ5ã @shdZddlmZddlZddlZddlZddlmZddl    m
Z
m Z m Z m Z mZmZmZmZmZmZmZmZddlmZdd„Zdd    „Zd
d „Zd d „Zdd„Zdd„Zej  dedddddgdœƒeddiƒeddie!dgedddgddddgdœƒeddddgiƒeddddgiƒgg¡dd„ƒZ"dd„Z#dd„Z$d d!„Z%d"d#„Z&d$d%„Z'd&d'„Z(d(d)„Z)d*d+„Z*d,d-„Z+d.d/„Z,d0d1„Z-d2d3„Z.d4d5„Z/d6d7„Z0ej  d8ed9d:d;d<ed=d:d>d<ed?d:d@d<g¡dAdB„ƒZ1dCdD„Z2dEdF„Z3dGdH„Z4dIdJ„Z5dKdL„Z6dMdN„Z7dOdP„Z8dQdR„Z9ej  dSidTdUig¡ej  dVgdWdWge:ddXƒe ;g¡g¡dYdZ„ƒƒZ<Gd[d\„d\ƒZ=dS)]z test get/set & misc é)Ú    timedeltaN)Ú IndexingError) ÚNAÚ    DataFrameÚIndexÚ
IndexSliceÚ
MultiIndexÚSeriesÚ    TimedeltaÚ    TimestampÚconcatÚ
date_rangeÚ period_rangeÚtimedelta_rangec    CsÜttj d¡dddddgd}d}tjt|d|dW5QRXtjt|dd|d<W5QRXtjtdd|d    W5QRX| ¡}tjt|d|dW5QRXd
}tjt|dd|d<W5QRXdS) NéÚaÚb©Úindexz/index 5 is out of bounds for axis 0 with size 5©Úmatchrz^'c'$Úcz7index 5 is out of bounds for axis (0|1) with size 5|^5$)    r    ÚnpÚrandomÚrandnÚpytestÚraisesÚ
IndexErrorÚKeyErrorZ
sort_index)ÚsÚmsg©r!úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/indexing/test_indexing.pyÚtest_basic_indexingsr#cCs^|}tdddg|d}ttdƒ|d}|d}tddgtddg|dd}tj||dddS)    Nér©ÚdtypeéréT©Z check_exact©rr    ÚrangeÚtmÚassert_series_equal)Úany_numeric_dtyper&ÚidxÚserÚresultÚexpectedr!r!r"Ú6test_getitem_numeric_should_not_fallback_to_positional0s r3cCsT|}tdddg|d}ttdƒ|d}d|d<tdddg|d}tj||dddS)    Nr$rr%r'ré
Tr)r*)r.r&r/r0r2r!r!r"Ú6test_setitem_numeric_should_not_fallback_to_positional:s r5cCsh|jdddg}||}| |¡}t ||¡||d|d…}|j|d|d…}t ||¡dS)Nrr4érr()rÚreindexr,r-Úloc)Údatetime_seriesÚindicesr1r2r!r!r"Útest_basic_getitem_with_labelsDs
 r;cCslttdddddddgd}tddd    }|jd}||ks>t‚|jd
}||ksTt‚|d}||ksht‚dS) Nz
2011-01-01r'z
US/Eastern)ÚperiodsÚtzrrrr)r=r)r    r r r8ÚAssertionErrorÚiloc)r0r2r1r!r!r"Ú test_basic_getitem_dt64tz_valuesPs ÿ 
 
 r@cCs@ttj d¡ƒ}|d}t ||¡d|d<|dk ¡s<t‚dS)Nr4.r)r    rrrr,r-Úallr>)rr1r!r!r"Útest_getitem_setitem_ellipsis_s
 rBz$result_1, duplicate_item, expected_1é r$r(r'©r$r(i9r%cCs<| |¡}| |¡}t |d|¡|d|dks8t‚dS)Nr$r()Ú_appendr,r-r>)Zresult_1Zduplicate_itemZ
expected_1r1r2r!r!r"Ú$test_getitem_with_duplicates_indicesis
 
rFcCsJtdddgdddgƒ}|jd|dks,t‚d|jd<t |dd¡dS)    Nr$r(r'rrrrr)r    r?r>r,Zassert_almost_equal©rr!r!r"Útest_getitem_setitem_integers€s
rHcCsòtdddd}t|ƒ}t|dtƒs(t‚t|jdtƒs<t‚t|jdtƒsPt‚t|jdtƒsdt‚t|jd    tƒsxt‚t||d
}t|dtƒs–t‚t|j|dtƒs®t‚t|jdtƒsÂt‚t|j|dtƒsÚt‚t|jd    tƒsît‚dS) NZ20090415Z20090519ÚB)Úfreqrr$r(r'ér)    r r    Ú
isinstancer r>ÚatÚiatr8r?©Úrngr0r!r!r"Útest_series_box_timestamp‰s rQcCs|tdddd}t|ƒ}t|dtƒs(t‚t|jdtƒs<t‚t|jdtƒsPt‚t|jdtƒsdt‚t|jd    tƒsxt‚dS)
Nz    1 day 1 srÚh©r<rJrr$r(r'rK)    rr    rLr
r>rMrNr8r?rOr!r!r"Útest_series_box_timedeltašsrTc    CsFttdƒttdddƒƒd}tjtdd||ƒdW5QRXdS)    Nr4rér(rz^1$rr$)r    r+Úlistrrr©Ú
indexer_slr0r!r!r"Útest_getitem_ambiguous_keyerror¤srYc    CsRtddddgddddgƒ}tjtt d¡d    ||ƒdddd
gW5QRXdS) Nr$r(r'rKÚfooÚbarZbahz['bam'] not in indexrZbam)r    rrrÚreÚescaperWr!r!r"Útest_getitem_dups_with_missingªsr^cCsXttdƒttdddƒƒd}| ¡}d||ƒd<t|tdgdgdgƒ}t ||¡dS)Nr4rrUr(rrr$)r    r+rVÚcopyr r,r-)rXrÚs2r2r!r!r"Útest_setitem_ambiguous_keyerror²s
 racCsrtj||jd<tj|dddg<tj|d<t |d¡s<t‚t |d¡sNt‚d|t |¡<t |d¡rnt‚dS)Nrr$r(éé)rÚNaNrÚisnanr>)r9r!r!r"Ú test_setitem¼s
rfcCs6|dd…}t|ƒt|jƒks"t‚|jjdks2t‚dS)NrrUT)Úlenrr>Z    is_unique)r9Úslr!r!r"Ú test_setsliceÆs ric    Csd}tjt|d|dd…dfW5QRXtjt|dd|dd…df<W5QRXd}tjt|d|tddƒgW5QRX|tddƒf}|dd…}t ||¡d}tjt|d|dtddƒgW5QRXtjt|dd|dtddƒg<W5QRXdS)Nz0key of type tuple not found and not a MultiIndexrr(z Indexing with a single-item listrzunhashable type(: 'slice')?)rrrÚ
ValueErrorÚslicer,r-Ú    TypeError)r9r r1r2r!r!r"Ú!test_basic_getitem_setitem_cornerÌs   rmcCs| ¡}|dd…}|dd…}|dd…}|jd|jks@t‚|jd|jksTt‚t|ƒt|jƒksjt‚||jd||jdksŠt‚|jd|jdks¢t‚t |t |¡dd…¡sÀt‚|dd…}d|dd…<|rêt ||¡n|dd…dk     ¡st‚dS)Nr4rUiöÿÿÿé    rr$é )
r_rr>rgr,Z equalContentsrÚarrayr-rA)Z string_seriesZ object_seriesÚusing_copy_on_writeÚoriginalZnumSliceZ numSliceEndZobjSlicerhr!r!r"Ú
test_sliceçs      rscCsœtgtd}tdƒ|jd<t |ttdƒdgd¡| |j     dd¡¡}t |tt
j tdƒgddgd¡tdƒ|jd<ttdƒddgd}t ||¡dS)Nr%r$rIz1 daysrrÚA) r    Úobjectrr8r,r-r
r7rÚinsertrÚnan)rr2r!r!r"Útest_timedelta_assignments "rxcCsÈtdd„dDƒƒ}|jdddgdd}|dks2t‚td    gd
gd }d |d <| ¡}||d  |¡|rn|}nJtd    ddgd    ddgd    ddgd d    d gdœƒ}|jdddgdd}|dks¸t‚t ||¡dS)NcSsi|]}|dddg“qS)r$r(r'r!)Ú.0rr!r!r"Ú
<dictcomp>sz3test_underlying_data_conversion.<locals>.<dictcomp>)rrrrrrT)Zinplacer$)r(r(r(rrÚvalr(r')rrrr{)rZ    set_indexr>r    r_Úupdater,Úassert_frame_equal)rqÚdfZ return_valuerZ df_originalr2r!r!r"Útest_underlying_data_conversions  $ÿ rcCs.|dddg}tj|d<t |d¡r*t‚dS)Nrr4r6r$)rrdrer>)r9Úseqr!r!r"Útest_preserve_refs%s
rcCs>|}ttj t|ƒ¡|dd}||ƒd}|j|jks:t‚dS)NZsth)rÚnamerZ)r    rrrrgr‚r>)Z%lexsorted_two_level_string_multiindexrXrr0r1r!r!r"Útest_multilevel_preserve_name+s rƒrz
2014-01-01rUZMSrSz2014-01ÚMÚ0ÚHc
CsÜt|dƒ}t|dƒ}tt d¡|ƒ}t}||dfD]ž}t |||dd…|ddd…¡t ||d|d…|ddd…¡||dfD]H}t ||||d…|ddd…¡t ||||d…|ddd…¡qŒq8dS)Nrné rUéÿÿÿÿér)Ústrr    rZarangerr,Z!assert_indexing_slices_equivalent)rZkeystr1Zkeystr2r0ZSLCÚkeyÚkey2r!r!r"Útest_slice_with_negative_step6s      """rcCsPtddgddgd}|ddks$t‚|ddks4t‚d|d<|ddksLt‚dS)Nr$r()r)rrr')r    r>rGr!r!r"Útest_tuple_indexNs
rŽcCsbtdƒtdƒ}}tddg||gd}||dks6t‚||dksFt‚d||<||dks^t‚dS)Nrrr$r(rr')Ú    frozensetr    r>)Zidx0Zidx1rr!r!r"Útest_frozenset_indexWs rcCsNtddgddgd}| ¡}tddgddgd}||j|dk<t ||¡dS)    Nr$r(rrrrcééd)r    r_r8r,r-)r0r2Úrhsr!r!r"Ú"test_loc_setitem_all_false_indexeras
r”cCs<tdgƒ}|jttdgdd}tgdd}t ||¡dS)Nr$FÚbooleanr%Úint64)r    r8rr,r-©r0r1r2r!r!r"Ú+test_loc_boolean_indexer_non_matching_indexjs
 r˜c    CsHtdgƒ}ttdgdddgd}tjtdd|j|W5QRXdS)Nr$Fr•r()r&rZ Unalignabler)r    rrrrr8)r0Úindexerr!r!r"Ú,test_loc_boolean_indexer_miss_matching_indexrs
ršcCsBtddgiƒ}td|iƒ}||jd<t||dœƒ}t ||¡dS)Nrr$ÚlabelÚ    new_label)r›rœ)rr    r8r,r-)r~r0r2r!r!r"Ú(test_loc_setitem_nested_data_enlargementzs
 
rcCsNtdtdddgddd}|jdd…}tdtdgddd}t ||¡dS)Nr$rr(ZInt64r%rr')r    rr8r,r-r—r!r!r"Ú'test_loc_ea_numeric_index_oob_slice_endƒsržc    Cs6tdddœƒ}tjtdd|jdW5QRXdS)Nr$r)TFr…r)r    rrrr8)r0r!r!r"Útest_getitem_bool_int_key‹srŸr{rÚxr™FrˆcCs6tdddgi|–ƒ}| ¡}d|j|<t ||¡dS)Nrr$r(gø?)rr_r8r,r})r™r{r~r2r!r!r"Útest_setitem_empty_indexer’s
r¡c@s°eZdZej ddhddig¡dd„ƒZej ddhddidhdfddidfg¡dd„ƒZej ddhddig¡dd    „ƒZej ddhddidhdfddidfg¡d
d „ƒZ    d S) ÚTestDeprecatedIndexersr‹r$c    Cs4tddgƒ}tjtdd|j|W5QRXdS©Nr$r(úas an indexer is not supportedr©r    rrrlr8©Úselfr‹r0r!r!r"Ú$test_getitem_dict_and_set_deprecateds z;TestDeprecatedIndexers.test_getitem_dict_and_set_deprecatedr(c    CsBtddgt ddg¡d}tjtdd|j|W5QRXdS©Nr$r(rD)r'rKrr¤r©r    rÚ from_tuplesrrrlr8r¦r!r!r"Ú/test_getitem_dict_and_set_deprecated_multiindex¤szFTestDeprecatedIndexers.test_getitem_dict_and_set_deprecated_multiindexc    Cs4tddgƒ}tjtddd|j|<W5QRXdSr£r¥r¦r!r!r"Ú$test_setitem_dict_and_set_disallowed«s z;TestDeprecatedIndexers.test_setitem_dict_and_set_disallowedc    CsBtddgt ddg¡d}tjtddd|j|<W5QRXdSr©rªr¦r!r!r"Ú/test_setitem_dict_and_set_disallowed_multiindex²szFTestDeprecatedIndexers.test_setitem_dict_and_set_disallowed_multiindexN)
Ú__name__Ú
__module__Ú __qualname__rÚmarkÚ parametrizer¨r¬r­r®r!r!r!r"r¢œs
(
 
(r¢)>Ú__doc__Údatetimerr\ÚnumpyrrZ pandas.errorsrZpandasrrrrrr    r
r r r rrZpandas._testingZ_testingr,r#r3r5r;r@rBr²r³rurFrHrQrTrYr^rarfrirmrsrxrrrƒrrŽrr”r˜ršrržrŸrkrpr¡r¢r!r!r!r"Ú<module>sx  8 
 
 
 
ýýúþ
    
 
 
    ýþ
    
        $