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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
| U
| ý°do) ã @ s d Z ddlmZ ddlZddlZddlmZmZm Z ddl
| m Z ej d¡Zej dddg¡d d
| Zd d Zd d Zedd Zeej ddddddgifdddddgifdddgeg dddfdddgeg dddfdddgeg dddfdddgeg dddfddgdge jg gd ddgddfddgdge jg gd ddgddfddgdge jg gd ddgddfddgdge jg gd ddgddfg
| ¡dd Zed d Zeej d d d gd!d gd d"gd!d"gd#d$gg¡d%d& Zed'd( Zed)d* Zed+d, Zej j d-d. Z ed/d0 Z ed1d2 Z ed3d4 Z ed5d6 Z!ed7d8 Z"eej d9e#d:fd;g¡d<d= Z$ed>d? Z%dS )@z
| Tests that the specified index column (a.k.a "index_col")
| is properly handled or inferred during parsing for all of
| the parsers defined in parsers.py
| é )ÚStringION)Ú DataFrameÚIndexÚ
| MultiIndexZ pyarrow_skipÚ with_headerTFc C s | }d}d}|rP|| }|j t|dd}|j t|dd d¡}t ||¡ n4|}d}tjt|d |j t|dd W 5 Q R X d S ) NaÑ KORD1,19990127, 19:00:00, 18:56:00, 0.8100, 2.8100, 7.2000, 0.0000, 280.0000
| KORD2,19990127, 20:00:00, 19:56:00, 0.0100, 2.2100, 7.2000, 0.0000, 260.0000
| KORD3,19990127, 21:00:00, 20:56:00, -0.5900, 2.2100, 5.7000, 0.0000, 280.0000
| KORD4,19990127, 21:00:00, 21:18:00, -0.9900, 2.0100, 3.6000, 0.0000, 270.0000
| KORD5,19990127, 22:00:00, 21:56:00, -0.5900, 1.7100, 5.1000, 0.0000, 290.0000
| KORD6,19990127, 23:00:00, 22:56:00, -0.5900, 1.7100, 4.6000, 0.0000, 280.0000zBID,date,NominalTime,ActualTime,TDew,TAir,Windspeed,Precip,WindDir
| ZID©Ú index_colr ©ÚheaderzIndex ID invalid©Úmatch)Úread_csvr Ú set_indexÚtmÚassert_frame_equalÚpytestÚraisesÚ
| ValueError) Ú all_parsersr ÚparserZ no_headerr
| ÚdataÚresultÚexpectedÚmsg© r ú\d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/parser/test_index_col.pyÚtest_index_col_named s r c C sx | }d}t dddgdddgdd d
| gd d d gdtdddgddd}dddddg}|jt||dgd}t ||¡ d S )Nz+1,2,3,4,hello
| 5,6,7,8,world
| 9,10,11,12,foo
| é é é é é é
| é é é é é é )ÚaÚbÚcÚdZhelloZworldÚfooÚmessage©Úname©Úindexr) r* r+ r, )Únamesr )r r r r r r )r r r r r3 r r r r Útest_index_col_named20 s $þr4 c C s< d}| }d}t jt|d |jt|dd W 5 Q R X d S )Nza,b
| 1,2z)The value of index_col couldn't be 'True'r Tr )r r r r r )r r r r r r r Útest_index_col_is_trueB s
| r5 c C sX d}| }| t|¡}tdddgdddgdd d
| ggd d d gdddgd}t ||¡ d S )Nz$A,B,C
| foo,1,2,3
| bar,4,5,6
| baz,7,8,9
| r r r# r& r r! r$ r' r r- ÚbarZbazÚAÚBÚC©r2 Úcolumns©r r r r r ©r r r r r r r r Útest_infer_index_colL s ýr> zindex_col,kwargsr; ÚxÚyÚzr/ ©r; r2 r r ©r3 c C s4 d}| }|j t||d}tf |}t ||¡ d S )Nzx,y,zr r< )r r Úkwargsr r r r r r r Útest_index_col_empty_data^ s
| )
| rE c C s8 d}| }|j t|dd}tddgd}t ||¡ d S )Nzx,yFr r? r@ ©r; r< r= r r r Ú test_empty_with_index_col_false s
| rG Ú index_namesÚ r- r6 ZNotReallyUnnamedz
| Unnamed: 0c C sv | }d |dg ¡}|jt|ddgd}tddddd git d
| d gd d gg¡d}dd |D |j_t ||¡ d S )Nú,zcol
| a,c,1
| a,d,2
| b,c,3
| b,d,4r r r Úcolr r# r& r) r* r+ r, r1 c S s g | ]}|r|nd qS )Nr )Ú.0r0 r r r Ú
| <listcomp>¯ s z+test_multi_index_naming.<locals>.<listcomp>)
| Újoinr r r r Z from_productr2 r3 r r )r rH r r r r r r r Útest_multi_index_naming s ÿrO c C sr | }d}|j t|ddgd}tdddddgitdd gd
| dd d ggddd
| d
| gdd
| dd ggd d}t ||¡ d S )Nz$,Unnamed: 2,
| a,c,1
| a,d,2
| b,c,3
| b,d,4r r r z
| Unnamed: 2r+ r, r) r* r r# r& )ÚlevelsÚcodesr1 )r r r r r r )r r r r r r r r Ú,test_multi_index_naming_not_all_at_beginning³ s ÿþrR c C st | }t dddg¡}ttj dd¡|dddgd}t ¡ $}| |¡ |j |dd d
| gd }W 5 Q R X t
| ||¡ d S ) N)r7 r r )r8 r r r# r? r@ rA r: r r r r ) r Ú from_tuplesr ÚnpÚrandomÚrandnr Ú ensure_cleanÚto_csvr r )r r Úmidxr Úpathr r r r Ú%test_no_multi_index_level_names_empty s
|
| r[ c C sº | }d}t jddgddgd}tdg}tdd gg||d
| }|jt|d d dgd }t ||¡ td dg}tddgdd}tddgddgg||d
| }|jt|dd d }t ||¡ d S )Nz
| I11,A,A
| I12,B,B
| I2,1,3
| )r7 r8 )r7 zB.1ZI11ZI12rC ZI2r r# r: r )r r
| r7 zA.1r/ r8 Ú1Ú3)r rS r r r r r r )r r r rY Úidxr r Zcol_idxr r r Útest_header_with_index_colÎ s
| r_ c C sj | }d}t t|tj |¡d}t ¡ $}|j|dd |j|dgd}W 5 Q R X t ||
| d¡¡ d S )NiAB ©r) r* Fr1 r r r) ) r ÚrangerT rU rV r rW rX r r r )r r ÚNZdfrZ r r r r Útest_index_col_large_csvæ s
| rc c C sX | }|j tdddgdd}tg tg tjddgddggd d
| gd d }t ||¡ d S ) Núa0,a1,a2
| b0,b1,b2
| r r ©r
| r Úa1Úa2Úb1Úb2Úa0Úb0rC r: )r r r r r Ú from_arraysr r ©r r r r r r r Ú)test_index_col_multiindex_columns_no_dataõ s ÿ ÿýrn c C sD | }|j tddgdd}tg ddgtg ddd}t ||¡ d S ) Nz a0,a1,a2
| r re rf rg rj r/ rB ©r r r r r r rm r r r Ú test_index_col_header_no_data s
| ýrp c C sL | }|j tdddgd}tg t dddgdd d
| gg¡d }t ||¡ d S ) Nrd r r r rj rf rg rk rh ri rF )r r r r rl r r rm r r r Ú test_multiindex_columns_no_data s ÿrq c C s` | }|j tdddgdd}tddggtjddgdd ggd
| d gd tdgd }t ||¡ d S )Nz a0,a1,a2
| b0,b1,b2
| data,data,datar r re r rf rg rh ri rj rk rC rB )r r r r rl r r r rm r r r Ú+test_multiindex_columns_index_col_with_data s ÿ ÿûrr c C sR | }|j tdddgdgddid}tdgdgdd d¡}tj||d
| d d S ) Nz0,1r) r* ZUInt8)r3 r Údtyper r r` )r F)Zcheck_index_type)r r r r r r rm r r r Ú test_infer_types_boolean_sum/ s üÿüÿù rt z
| dtype, valZ01)Úint64r c C sL d}| }|j t|dd|id}tddgit|gddd}t ||¡ d S )Nza,b
| 01,2r) )r rs r* r r/ r1 ro )r rs Úvalr r r r r r r Ú test_specify_dtype_for_index_colG s
| rw c C s` | }d}|j t|ddgdd}tjdddgdd gd
| }td dd gg|d gd}t ||¡ d S )Nza,b,c,d
| e,f,g,h
| x,y,1,2
| r r re )r) Úe)r+ Úg)r, Úhr* ÚfrC r? r r@ rB )r r r rS r r r )r r r r Úcolsr r r r Ú-test_multiindex_columns_not_leading_index_colR s ý ÿr} )&Ú__doc__Úior ÚnumpyrT r Zpandasr r r Zpandas._testingZ_testingr ÚmarkZ usefixturesZ skip_pyarrowZ parametrizer r4 r5 r> rl rE rG rO rR r[ r_ Zslowrc rn rp rq rr rt Úobjectrw r} r r r r Ú<module> s
|
|
| þþþþþþþþäþ'
|
| ûþ
|
|
|
|
|
|
|
|
|
|
|
|
|
