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
| U
| ý°dB ã @ sÞ d Z ddlmZ ddlmZ ddlZddlZddlZ ddlm
| Z
| m Z ddl m Z ej d¡Zdd Zej d d
| d g¡ej d ed d g¡dd Zdd Zdd Zdd Zej ddd eg¡dd Zdd Zd d ZdS ) zc
| Tests column conversion functionality during parsing
| for all of the parsers defined in parsers.py
| é )ÚStringIO)ÚparseN)Ú DataFrameÚIndexZ pyarrow_skipc C s8 | }d}t jtdd |jt|dd W 5 Q R X d S )Nzindex,A,B,C,D
| foo,2,3,4,5
| zType converters.+)Úmatchr ©Ú
| converters)ÚpytestZraisesÚ TypeErrorÚread_csvr )Ú all_parsersÚparserÚdata© r ú]d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/io/parser/test_converters.pyÚ!test_converters_type_must_be_dict s r Úcolumné ÚDÚ converterc C s t | d¡d S )Nú/é )ÚintÚsplit©Úxr r r Ú<lambda>! ó r c C sN | }d}|j t|||id}| t|¡}|d |¡|d< t ||¡ d S )Nz;A,B,C,D
| a,1,2,01/01/2009
| b,3,4,01/02/2009
| c,4,5,01/03/2009
| r r )r r ÚmapÚtmÚassert_frame_equal)r r r r r ÚresultÚexpectedr r r Útest_converters s r# c C sP | }d}ddd i}|j t|d |d}tdddgd d
| d gg}t ||¡ d S ) Nz000102,1.2,A
| 001245,2,Br c S s | ¡ S ©N)Ústripr r r r r 7 r z2test_converters_no_implicit_conv.<locals>.<lambda>©Úheaderr Z000102g333333ó?ÚAZ001245r ÚB©r r r r r )r r r r r! r" r r r Ú test_converters_no_implicit_conv2 s r+ c C s i }| }d}dd |d< |d< |d< |j t|d|d}td d
| d d d dgddddddgddddddggdddd d dgd }t ||¡ d S ) NzId;Number1;Number2;Text1;Text2;Number3
| 1;1521,1541;187101,9543;ABC;poi;4,7387
| 2;121,12;14897,76;DEF;uyt;0,3773
| 3;878,158;108013,434;GHI;rez;2,7356c S s t | dd¡S )Nú,Ú.)ÚfloatÚreplacer r r r r J r z5test_converters_euro_decimal_format.<locals>.<lambda>ZNumber1ZNumber2ZNumber3ú;)Úsepr é gAñcÌÄ@gÕ h¢ïÖAÚABCZpoig8ÖÅmô@r gHáz®G^@g{®GáÍ@ZDEFZuytg¡ø1æ®%Ø?r g%Cq@gçû©ñÖ^ú@ZGHIZrezg4·@â@ZIdZText1ZText2)Úcolumnsr* )r r r r r! r" r r r Ú#test_converters_euro_decimal_format? s þÿýúr5 c C s | }d}dd }dd }dd }g }||fD ]B}|j t|||dd d gd
| }t |d d ¡sdt| |¡ q,t |d |d ¡ d S )Nz+id,score,days
| 1,2,12
| 2,2-5,
| 3,,14+
| 4,6-12,2c S sB | ¡ } | stjS | d¡}|r6t| d d
d } nt| } | S ©Nú+éÿÿÿÿr2 ©r% ÚnpÚnanÚendswithr ©r Zis_plusr r r Ú convert_daysa s
| z6test_converters_corner_with_nans.<locals>.convert_daysc S sB | ¡ } | stjS | d¡}|r6t| d d
d } nt| } | S r6 r9 r= r r r Úconvert_days_sentinelp s
| z?test_converters_corner_with_nans.<locals>.convert_days_sentinelc S sN | ¡ } | stjS | d¡dkrBtt| d¡\}}d|| }nt| }|S )Nú-r g à?)r% r: r; Úfindr r r r. )r Zval_minZval_maxÚvalr r r Ú convert_score s z7test_converters_corner_with_nans.<locals>.convert_score)ZscoreÚdaysÚ )r Z na_valuesrD r2 r )r r ÚpdZisnaÚAssertionErrorÚappendr r ) r r r r> r? rC ÚresultsZ day_converterr! r r r Ú test_converters_corner_with_nansX s ý rJ Úconv_fc C s | S r$ r r r r r r r c C sT | }d}|j t|ddd|id}tdddgitdd gdd
| d d }t ||¡ d S ) Nz A;B
| 1;2
| 3;4r0 r( )r1 Z index_colr r) r é Ú1Ú3Úobject)ÚnameZdtype)Úindex)r r r r r r )r rK r r ÚrsÚxpr r r Ú test_converter_index_col_bug s ÿ"rT c C sH | }d}|j t|ddd id}tddgdd gd
| }t ||¡ d S ) Nz A,B
| 1,2
| 3,4r( c S s | S r$ r r r r r r ® r z0test_converter_identity_object.<locals>.<lambda>r rM rN r rL )r( r) r* )r r r rR rS r r r Ú test_converter_identity_object© s
| rU c C sj | }d}|j t|ttdtjtjtjdd}tt dg¡t dg¡t dg¡d}t ||¡ d S )NzA,B,B
| X,Y,Z
| 1,2,3r ))r( ÚX)r) ÚY)r) ÚZr& r2 r )
| r r ÚlistÚranger: Úint32Úfloat32r r r )r r r r! r" r r r Útest_converter_multi_index´ s"
| ýý
|
|
|
| ýÿr] ) Ú__doc__Úior Zdateutil.parserr Únumpyr: r ZpandasrF r r Zpandas._testingZ_testingr ÚmarkZ usefixturesZ
| pytestmarkr Z parametrizer# r+ r5 rJ ÚstrrT rU r] r r r r Ú<module> s,
|
| ÿ C
|
|
|
