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
U
®ý°d|ã    @srddlmZddlZddlZddlZddlmZmZm    Z    ddl
m Z ej  dddgddggddgddggddgddggddgddggg¡ej  dddg¡d    d
„ƒƒZej  dddgddggddgddggddgddggddgddggg¡ej  d d d g¡ej  dddg¡dd„ƒƒƒZdd„Zdd„Zdd„Zej  d dd d ddg¡dd„ƒZdd„Zdd„Zd d!„Zd"d#„ZdS)$é)ÚtimezoneN)ÚSeriesÚ
date_rangeÚ period_rangezfirst_slice,second_sliceééûÿÿÿÚfilléÿÿÿÿcCs,|t|Ž}|t|Ž}|j|||d\}}|jj|j|d}    |dk    r¢|j |    ¡}
|j |    ¡} t|
ƒdkr€| |
¡|k ¡s€t‚t| ƒdkr¢| | ¡|k ¡s¢t‚| |    ¡} | |    ¡} |dk    rÒ|      |¡} |      |¡} t
  || ¡t
  || ¡|j dksøt‚| j dkst‚|j dkst‚| j dks(t‚dS)N)ÚjoinZ
fill_value©ÚhowrÚts) ÚsliceÚalignÚindexr
Ú
differenceÚlenÚreindexÚallÚAssertionErrorÚfillnaÚtmÚassert_series_equalÚname)Údatetime_seriesÚ first_sliceÚ second_sliceÚ    join_typerÚaÚbÚaaÚabÚ
join_indexZdiff_aZdiff_bÚeaÚeb©r%ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/series/methods/test_align.pyÚ
test_aligns,      
 
 
 
  r'ÚmethodÚpadZbfillÚlimitéc CsŒ|t|Ž}|t|Ž}|j||||d\}}    |jj|j|d}
| |
¡} | |
¡} | j||d} | j||d} t || ¡t |    | ¡dS)N)r
r(r*r )r(r*)rrrr
rrrr) rrrrr(r*rrr r!r"r#r$r%r%r&Útest_align_fill_method7s  
 
 r,cCsr|dd… ¡}| ¡}|j|dd\}}d|dd…<|dd…dk ¡rNt‚| ¡}|j|ddd\}}d|dd…<|r”|dd…dk ¡r¬t‚n|dd…dk ¡s¬t‚| ¡}|dd… ¡}|j|dd\}}d|dd…<|dd…dk ¡rút‚| ¡}|dd… ¡}|j|ddd\}}d|dd…<|rT|dd…dk ¡rnt‚n|dd…dk ¡snt‚dS)    NéÚleft©r
F)r
ÚcopyÚrightér)r0rÚanyrr)rÚusing_copy_on_writerrÚraÚ_Úrbr%r%r&Útest_align_nocopyUs.    r8cCsŽ|j|dd\}}|j|jks"t‚|j|jks2t‚|j|dd\}}|sj|j|jk    sXt‚|j|jk    sŠt‚n |j|jkszt‚|j|jksŠt‚dS)NF)r0T)rrr)rr4rrr%r%r&Útest_align_same_indexysr9c     Cs†tjjtdƒtdƒtdƒgdd}tjtdƒdd}ttjddd    |d
}ttjddd    |d
}|j|d d \}}|j|d d \}}|}t     
||¡t     
||¡tddddtj tj gd|d
}    t     
|    |¡t     
|    |¡|j|d d \}}|j|d d \}}tjjtdƒtdƒtdƒgdd}
tddddddddg|
d
}t     
||¡t     
||¡tddddgd|
d
}    t     
|    |¡t     
|    |¡dS)Nrr2)rrÚc©Únamesr)ré Úint64)Zdtype©rr.r/r1rr+éééé    ) ÚpdÚ
MultiIndexZ from_productÚrangeZIndexrÚnpZarangerrrÚnan) ZmidxÚidxÚs1Ús2Zres1lZres1rZres2lZres2rZexplÚexprZexp_idxr%r%r&Útest_align_multiindex‡s6ÿ     ÿ   rMZbackfillZffillcCsZttdƒtdƒd}tjdtdƒtdƒd}|j||d\}}t ||¡t ||¡dS)Nr2r?g)rÚcolumns)r()rrFrDZ    DataFramerrrZassert_frame_equal)r(ÚserZdfZ
result_serZ    result_dfr%r%r&Ú test_align_with_dataframe_methodªs
 rPcCshtddddd}ttj t|ƒ¡|d}| d¡}| |¡\}}|jj    t
j ksRt ‚|jj    t
j ksdt ‚dS)NZ2001r-ÚHz
US/Eastern)ZperiodsÚfreqÚtzr?z
US/Central) rrrGÚrandomÚrandnrZ
tz_convertrrrSrÚutcr)Zidx1rOZ ser_centralZnew1Znew2r%r%r&Ú%test_align_dt64tzindex_mismatched_tzsµs 
rWcCsBtdddd}ttj t|ƒ¡|d}|j|ddd…|ddS)Nz1/1/2000z1/1/2010ÚA)rRr?rr/)rrrGrTrUrr)rÚrngr r%r%r&Útest_align_periodindexÀsrZcCs°tdgtjjdgddgdd}tdgtjjdgdd    dgdd}| |¡\}}tdgtjjd
gddd    gdd}tdgtjjd
gddd    gdd}t ||¡t ||¡dS) Nr)r+r2rr:r;r?r+©r+rr2r)r+r2r©rrDrEÚ from_tuplesrrr)r.r1Ú result_leftÚ result_rightÚexpected_rightÚ expected_leftr%r%r&Útest_align_left_fewer_levelsÈs  ÿÿÿ rbcCs¶tdgtjjdgdddgdd}tdgtjjd    gdd
dgdd}| |¡\}}tdgtjjd gdddd
gdd}tdgtjjd gdddd
gdd}t ||¡t ||¡dS) Nr)r+ér2rÚdr:r;r?r+r[r)r+r2rcrr\)r.r1r^r_rar`r%r%r&Ú&test_align_left_different_named_levelsÚs&ÿÿÿÿ re)ÚdatetimerÚnumpyrGZpytestZpandasrDrrrZpandas._testingZ_testingrÚmarkZ parametrizer'r,r8r9rMrPrWrZrbrer%r%r%r&Ú<module>sF  üþ     üþ    $#