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
U
­ý°dþã@szddlZddlZddlmZmZmZmZmZddl    m
Z Gdd„dƒZ dd„Z dd„Zd    d
„Zd d „Zd d„Zdd„ZdS)éN)ÚCategoricalIndexÚIndexÚ
MultiIndexÚ    TimestampÚ
date_rangec@seZdZdd„ZdS)ÚTestGetLevelValuesc    Cshtddd}|ddgg}ddddddddgddddddddgg}t||d}t| d¡dtƒsdt‚dS)    Nz1/1/2000é)Úperiodsrééé©ÚlevelsÚcodes)rrÚ
isinstanceÚget_level_valuesrÚAssertionError)ÚselfÚdatesrrÚindex©rúgd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/indexes/multi/test_get_level_values.pyÚ$test_get_level_values_box_datetime64s
  ( z7TestGetLevelValues.test_get_level_values_box_datetime64N)Ú__name__Ú
__module__Ú __qualname__rrrrrrsrc Csö| d¡}tddddddgdd}t ||¡|jdks<t‚| d¡}| d¡}t ||¡ttdd    gƒtd
d d gƒgt     dddd
d
d
g¡t     dd
d dd
d g¡gd }tdddd    d    d    gƒ}t | d¡|¡td
d d d
d d gƒ}t | d
¡|¡dS)NrZfooÚbarZbazZquxÚfirst©ÚnameÚAÚBr
r r r )
rrÚtmÚassert_index_equalrrrrÚnpÚarray)ÚidxÚresultÚexpectedrÚexprrrÚtest_get_level_valuess
 
 
 *þr*cCs„tjtjtjgdtjdgg}t |¡}| d¡}ttjtjtjgtjd}t ||¡| d¡}tdtjdgt    d}t ||¡dS)NÚar
r©Zdtype)
r$ÚnanrÚ from_arraysrrÚfloat64r"r#Úobject©Zarraysrr'r(rrrÚtest_get_level_values_all_na/s
 
 
r2cCs”dddgdtjdgg}t |¡}| d¡}tdtjdgƒ}t ||¡dddgtjtjdgg}t |¡}| d¡}ttjtjdgƒ}t ||¡dS)Nr+Úbr
r )r$r-rr.rrr"r#r1rrrÚ!test_get_level_values_int_with_na<s
 
 
 
r4cCsütjtjtjgdtjdgg}t |¡}| d¡}ttjtjtjgƒ}t ||¡| d¡}tdtjdgƒ}t ||¡dddgt     ddtj
g¡g}t |¡}| d¡}t     ddtj
g¡}t ||¡ggg}t |¡}| d¡}tgt d}t ||¡dS)Nr+r
rr3r,) r$r-rr.rrr"r#ÚpdÚ DatetimeIndexZNaTr0r1rrrÚtest_get_level_values_naKs$
 
 
 
 
 
 
 r7csjddlm}m}t ||dƒ|dƒgddg¡‰t ‡fdd„tˆjƒDƒ¡}td    d
„|jDƒƒsft    ‚dS) Nr)ÚPeriodÚ PeriodIndexZ2019Q1Z2019Q2r3rcsg|]}ˆ |¡‘qSr)Z_get_level_values)Ú.0Úlevel©r&rrÚ
<listcomp>osz6test_get_level_values_when_periods.<locals>.<listcomp>css|] }|jVqdS)N)Zis_monotonic_increasing)r:ÚxrrrÚ    <genexpr>qsz5test_get_level_values_when_periods.<locals>.<genexpr>)
Úpandasr8r9rr.ÚrangeZnlevelsÚallrr)r8r9Úidx2rr<rÚ"test_get_level_values_when_periodscsÿÿrDcCsrt tdddd¡}|jdd}tdddgƒ}t||gd    ddgd    ddggd
}|j|jdk    sbt‚t     
||¡dS) NZ20200101r ZBM)r    ÚfreqT)Údeepr
r rr ) r5r6rÚcopyrrÚvaluesrErr"r#)r&r(rCZmidxrrrÚ*test_values_loses_freq_of_underlying_indexts  rI)Únumpyr$r@r5rrrrrZpandas._testingZ_testingr"rr*r2r4r7rDrIrrrrÚ<module>s