U
|
ý°du ã @ sÎ d Z ddlmZ ddlZddlmZmZmZ ddlm Z
|
dZ dZ ej d¡Zej ddd d
|
gd
|
dd gg¡d d Zd d Zdd Zdd Zej ddd d
|
gd
|
dd gg¡ej dededg¡dd ZdS )z]
|
Tests the usecols functionality during parsing
|
for all of the parsers defined in parsers.py
|
é )ÚStringION)Ú DataFrameÚIndexÚ Timestampz['usecols' must either be list-like of all strings, all unicode, all integers or a callable.zAUsecols do not match columns, columns expected but not found: {0}Z pyarrow_skipÚusecolsé é c C s` d}| }ddgg}ddgt dt dgd}t|dd gd
|
}|jt|||d }t ||¡ d S ) Nz7a,b,c,d,e
|
0,1,2014-01-01,09:00,4
|
0,1,2014-01-02,10:00,4é r r ú2014-01-01 09:00:00ú2014-01-02 10:00:00©ÚaÚc_dr r ©Úcolumns©r Ú parse_dates©r r Úread_csvr ÚtmÚassert_frame_equal)Ú all_parsersr ÚdataÚparserr ÚcolsÚexpectedÚresult© r úfd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/parser/usecols/test_parse_dates.pyÚ test_usecols_with_parse_dates s
|
þr c
|
C s | }d}ddg}|d d
}dg}t tdtdtdgdd}dd d
|
d gi}t||d }|jt||d|d |d } t | |¡ d S )NzJ2008-02-07 09:40,1032.43
|
2008-02-07 09:50,1042.54
|
2008-02-07 10:00,1051.65ÚdateÚvaluesr z2008-02-07 09:40z2008-02-07 09:50z2008-02-07 10:00)Únameg
ëQ¸!@g\Âõ(J@gn@)Úindex)r Z index_colr ÚheaderÚnames)r r r r r r r )
|
r r r r% r r r# r r r r r r Ú test_usecols_with_parse_dates20 s. ýú úr& c C sv | }d}t d}dg}tddgdgdgdgdgd gd
|
gd gd gd
|
}t||d}|jt|||d}t ||¡ d S )Nú0a,b,c,d,e,f,g,h,i,j
|
2016/09/21,1,1,2,3,4,5,6,7,8Ú
|
abcdefghijr z
|
2016-09-21r r r é é é é é )
|
r ÚbÚcÚdÚeÚfÚgÚhÚiÚjr r )Úlistr r r r r r )r r r r r r r r r r r Ú test_usecols_with_parse_dates3Q s$ ö r8 c
|
C s| d}t d}ddgg}| }ddgdgdgdgd gd
|
gd gd gd }t|dgt d d}|jt|||d}t ||¡ d S )Nr' r( r r z 2016/09/21 1r r r) r* r+ r, r- ) Úa_br/ r0 r1 r2 r3 r4 r5 r6 r9 Zcdefghijr r )r7 r r r r r )r r r r r r r r r r r Ú test_usecols_with_parse_dates4l s*
|
÷ ýr: r% ZabcdeZacdc C sb d}ddgg}| }ddgt dt dgd}t|dd gd
|
}|jt||||d }t ||¡ d S ) Nz-0,1,2014-01-01,09:00,4
|
0,1,2014-01-02,10:00,4r r r r
|
r r r r r )r% r r r ) r r r% Úsr r r r r r r r Ú'test_usecols_with_parse_dates_and_names s
|
|
þ ÿr< )Ú__doc__Úior ZpytestZpandasr r r Zpandas._testingZ_testingr Z_msg_validate_usecols_argZ_msg_validate_usecols_namesÚmarkZ usefixturesZ
|
pytestmarkZ parametrizer r& r8 r: r7 r< r r r r Ú<module> s, ÿÿ
|
! þþ
|
