uj<: NdZddlmZddlmZddlZddlZddlmZm Z ddl Z ddl m Zejj!deeddgdZd Zd Zejj!d d dd gie j,gdgdggdfdddgde j,gdggdfddd dgde j,ddggddgfgdZdZejj!d d!gd"gd#gfd$gd%gd&gfd'gd(gd)gfgd*Zejj!d+gd,d-Zd.Zejj!d/id0fdd1gd2d1fgd3Zd4Zd5Zd6Zd7Z d8Z!d9Z"ejj!d:d;dddgfgd?Z#ejj!d:gd@dAZ$ejj!dBdCdDgidEdFigdGZ%ejj!dHdIdgdJZ&dKZ'dLZ(y)Mzj Tests that skipped rows are properly handled during parsing for all of the parsers defined in parsers.py )datetime)StringION)EmptyDataError ParserErrorskiprowsc J|}d}|jdk(rVt|tsFd}tjt |5|j t||ddddddy|j t||ddd}tjtdd d tdd d tdd d gd d }tjtjddjdgd|}tj ||y#1swYyxYw)Nzr#foo,a,b,c #foo,a,b,c #foo,a,b,c #foo,a,b,c #foo,a,b,c #foo,a,b,c 1/1/2000,1.,2.,3. 1/2/2000,4,5,6 1/3/2000,7,8,9 pyarrow@skiprows argument must be an integer when using engine='pyarrow'matchrTrheader index_col parse_datesM8[us]dtypename?$@rrrrrcolumnsindex)engine isinstanceintpytestraises ValueErrorread_csvrpdIndexr DataFramenparangereshapetmassert_frame_equal) all_parsersrparsertextmsgresultr expecteds o/var/www/html/python-recuperauto-other/env/lib/python3.12/site-packages/pandas/tests/io/parser/test_skiprows.pytest_skip_rows_bugr7sF D}} !*Xs*CP ]]:S 1  OO!     __$!QUF HH $1 xa3XdAq5IJ  E || #t$$V,iuH&(+-  s  DD"c|}ddjtdDcgc]7}djt|t|dzt|dzg9c}z}ddjdDcgc]7}djt|t|dzt|dzg9c}z}|jdk(rEd }t j t | 5|jt|d d g dddy|jt|d d g }|jt|}tj||ycc}wcc}w#1swYyxYw)Nza,b,c  ,rr)rrrrr r r r rr=r) joinrangestrr!r$r%r&r'rr.r/)r0r1idatacondensed_datar3r4condensed_results r6test_deep_skip_rowsrG>s5 F tyy=B2YG3q63q1u:s1q5z2 3G D=UV3q63q1u:s1q5z2 3V"N}} !P ]]:S 1 = OOHTNaVO < = __Xd^q!f_ =Fx'?@&"23 H W  =s 1 charr rz\s+)rsepnames)r@r(r*r!r$markxfail applymarkerreplacer%r&r'rr.r/) r0r`requestr1rDr5rqr3r4s r6test_skiprows_lineterminatorrvs F 99  D|| 5 5 5 9 H}} ^t%;{{  (U VD! <<n -D }} !I ]]:S 1  OO>     __ 6 F &(+  s ;!DD%c|}d}tjddgi}|jt|d}t j ||y)Nz a" b" a 1rQrrr?)r(r*r'rr.r/)r0r1rDr5r4s r6test_skiprows_infield_quoterxsF F D||S1#J'H __Xd^a_ 8F&(+rJzkwargs,expected1foo)rrpcj|}d}|jdk(rFd}tjt|5|jt |fddi|dddy|jt |fddi|}t j|dd gi}tj||y#1swYyxYw) N a 1 2 3 4 5r r r rc|dzdk(SNrrxs r6z)test_skip_rows_callable..'sq1uzrJc|dzdk(Sr~rrs r6rz)test_skip_rows_callable..*sA rJrr[)r0rXr5r1rDr3r4s r6test_skip_rows_callablersF D }} !P ]]:S 1 U FOOHTN T5I TV T U V__Xd^ U6J Uf UF||X1v./H&(+  Us B))B2cj|}d}tjddgddgg}|jdk(rEd}tjt |5|j t|dd dddy|j t|dd }tj||y#1swYyxYw) Nz0,a 1,b 2,c 3,d 4,errRrdr r r c |dvSN)rrrrs r6rz0test_skip_rows_callable_not_in..8s rJ)rrc |dvSrrrs r6rz0test_skip_rows_callable_not_in..=s rJ) r(r*r!r$r%r&r'rr.r/)r0r1rDr5r3r4s r6test_skip_rows_callable_not_inr/s F $D||aX3x01H }} !P ]]:S 1  OOt6O     __t.GF&(+  s B))B2ch|}d}d}|jdk(rDd}tjt|5|j t |ddddytjt |5|j t |ddddy#1swYyxYw#1swYyxYw) Nr|No columns to parse from filer r r cyNTrrs r6rz)test_skip_rows_skip_all..JrJr?cyrrrs r6rz)test_skip_rows_skip_all..NrrJ)r!r$r%r&r'rr)r0r1rDr3s r6test_skip_rows_skip_allrBs F D )C }} !P ]]:S 1 E OOHTN^O D E ~S 1A@AA  EAAB5B(B%(B1chd}|}d}|jdk(rDd}tjt|5|j t |ddddytjt |5|j t |ddddy#1swYyxYw#1swYyxYw) Nzby zeror|r r r c ddz SNrrrrs r6rz-test_skip_rows_bad_callable..Ys q1urJr?c ddz Srrrs r6rz-test_skip_rows_bad_callable..]s 1q5rJ)r!r$r%r&r'rZeroDivisionError)r0r3r1rDs r6test_skip_rows_bad_callablerQs C F D }} !P ]]:S 1 F OOHTN_O E F ( 4BABB  FBBrcpd}|}|jdk(rFd}tjt|5|j t |dgddddy|j t |dgd}t jgdgd d }tj||y#1swYyxYw) Nz$a,b 1,a 2,b 3,c 4,d 5,e 6,f 7,g 8,h r z=The 'nrows' option is not supported with the 'pyarrow' enginer r)rr<r)nrowsr)rrrr=)rQrSeghrQrRr[)r0rDr1r3r4r5s r6test_skip_rows_and_n_rowsr`s DF }} !M ]]:S 1 I OOHTN!iO H I __Xd^1y_ IF||/8QRSH&(+  Is B,,B5cd}|}|jdk(rFd}tjt|5|j t ||dddddy|j t ||dd}t |}t |}tj|tjd gd itj|tjd gd igd  y#1swYyxYw)Nz%col_a 10 20 30 40 50 60 70 80 90 100 r zAThe 'chunksize' option is not supported with the 'pyarrow' enginer c |dvSN)rr<rrrs r6rz,test_skip_rows_with_chunks..s 1 >rJr<)r!r chunksizec |dvSrrrs r6rz,test_skip_rows_with_chunks..s !y.rJcol_a)<F)PZd)r<rr)r ) r!r$r%r&r'rnextr.r/r(r*)r0rDr1r3readerdf1df2s r6test_skip_rows_with_chunksrxs DF }} !Q ]]:S 1  OO1     __v0HTUF v,C v,C#r||W6F,GHI#r||Wm,DIVW!  s C..C7z!j0,j0b j1,j1b j2,j2b a,b 1,2 3,4 zskiprows,header)rinferrrrrrrc|}|jtt||}tjddgddgd}t j ||y)Nrrrrrr<rr'rDATA_WITH_PREAMBLEr(r*r.r/r0rrr1r4r5s r6test_skiprows_with_headerrsV F __#$xF||1a&1v67H&(+rJ)rrrc|}|jtt||ddg}tjddgddgd}t j ||y) Nry)rrrprrrr<)rrrrs r6#test_skiprows_with_header_and_namesrs]F __#$xsTWjF||1a&1v67H&(+rJrXusecolsrRrrQc|}|jttfddi|}tjdddgi}d|vrtj ddgd |_tj||y) NrrrRrr<rrrQ)r) r'rrr(r*r)r r.r/)r0rXr1r4r5s r6*test_skiprows_with_header_column_selectionrsmF V__X&89 PA P PF||S1a&M*Hf1a&s3&(+rJrrc|}d}|jt|d|}|jt||}tj||y)Na,b 1,2 r)r)r'rr.r/)r0rr1rDr4r5s r6test_skiprows_negativersLF D __Xd^b_ HFx~f=H&(+rJc\|}d}|jdk(rAtjtd5|j t |ddddytjt d5|j t |ddddy#1swYyxYw#1swYyxYw)Nrr zCould not skip initial 5 rowsr rr?r)r!r$r%rr'rr)r0r1rDs r6test_skiprows_past_end_of_filersF D }} ! ]];.M N 8 OOHTNQO 7 8 ~-L M4344  844sB0B"B"B+cb|}d}|jdk(rAtjtd5|j t |ddddy|j t |d}t jddgt }tj||y#1swYyxYw) Nzj0 a,br rr rr?rQrR)rr) r!r$r%rr'rr(r*objectr.r/)r0r1rDr4r5s r6+test_skiprows_to_final_line_without_newlinersF D }} ! ]]>1P Q 8 OOHTNQO 7 8 __Xd^a_ 8F||S#Jf=H&(+  8s B%%B.))__doc__riornumpyr+r$ pandas.errorsrrpandasr(pandas._testing_testingr.rq parametrizelistrAr7rGrIr*rYr]r_rvrxrrrrrrrrrrrrrrrJr6rs   d58na%89%,:%,P4**8  !  BLL+->?3   AA3 / BLL56 P 3 Aq6 2 BLL,f56  '"F ,G"F ,,* +,I J   +,I J   !/0Q R  4 ,54 ,+, +,\, S  (%0 , ,,& A B,0!XH? fffq1#hG,,*,DE,F,Y$6c8J#KL,M,GT?3,4, 4 ,rJ