bMh#^ddlmZddlZddlZddlmZddlmZddlm Z m Z ddl m Z GddZGdd eZGd d eZGd d eZGddeZGddeZejdZejdZeZeZeZeZegZeegZejeejjej e dggZ!geee!Z"dgZ#ddgZ$dgZ%ge$e#e%Z&eje"e&dZ'ejee$dZ(ejee#dZ)eje!e%dZ*d Z+eje+d!e+d"d#Z,gd$Z-gd%Z.d&d'gZ/d(e.DZ0eje-)d*Z1eje0)d+Z2ejgd,)d-Z3ejd.Z4ejd/Z5dS)0) annotationsN) HAS_PYARROW)VERSIONS)read_csv read_tablec`eZdZUdZded<dZgZded<dZdZddd ddZ dZ ddddZ dS) BaseParserNz str | NoneengineTzlist[str | None]float_precision_choicescr|}||j|jd|S)N)r low_memory)copyupdater r )selfkwargss _/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/io/parser/conftest.py update_kwargszBaseParser.update_kwargss2 DOLLMMM cD||}t|i|SN)rrrargsrs rrzBaseParser.read_csvs(##F++((((r)raise_on_extra_warningscheck_stacklevel warn_type type[Warning]warn_msgstrrboolc||}tj||||5t|i|cdddS#1swxYwYdS)N)matchrr)rtmassert_produces_warningr)rrrrrrrs rread_csv_check_warningsz"BaseParser.read_csv_check_warnings s##F++  ' $;-     - - T,V,,  - - - - - - - - - - - - - - - - - -s AA A cD||}t|i|Sr)rrrs rrzBaseParser.read_table5s(##F++4*6***r)rc||}tj|||5t|i|cdddS#1swxYwYdS)N)r!r)rr"r#r)rrrrrrs rread_table_check_warningsz$BaseParser.read_table_check_warnings9s##F++  ' X?V    / /t.v.. / / / / / / / / / / / / / / / / / /s AA  A )rrrrrr)rrrr) __name__ __module__ __qualname__r __annotations__r r rrr$rr'rrr r sFJ022222 )))!%!% ------*+++!% ////////rr ceZdZdZgdZdS)CParserc)Nhigh round_tripNr(r)r*r r r,rrr.r.Ks! F:::rr.ceZdZdZdS)CParserHighMemoryFNr(r)r*r r,rrr4r4PsJJJrr4ceZdZdZdS)CParserLowMemoryTNr5r,rrr7r7TsJJJrr7ceZdZdZdgZdS) PythonParserpythonNr2r,rrr9r9Xs F#frr9ceZdZdZdgZdS) PyArrowParserpyarrowNr2r,rrr<r<]s F#frr<c|dddS)zG The directory path to the data files needed for parser tests. ioparserdatar,datapaths r csv_dir_pathrDbs 8D(F + ++rcXtj|ddddS)zH The path to the data file "test1.csv" needed for parser tests. r?rAcsvz test1.csv)ospathjoinrBs rcsv1rJjs( 7<<vu55{ C CCrpyarrow is not installedreasonmarksr:c_highc_lowr=paramsidsc|}|jdkr9tjdtdddl}|d|S)z) Fixture all of the CSV parsers. r=rN)paramr pytest importorskiprr= set_cpu_count)requestr@r=s r all_parsersr\sZ ]]__F } !!Ix ':;;; a   Mrc*|S)z< Fixture all of the CSV parsers using the C engine. rWr[s r c_parser_onlyr` ==??rc*|S)zA Fixture all of the CSV parsers using the Python engine. r^r_s rpython_parser_onlyrcrarc*|S)zB Fixture all of the CSV parsers using the Pyarrow engine. r^r_s rpyarrow_parser_onlyrerarcg}g}tttD]\}}t|dr |jd}|jD]}|jdkr8tjj tj t dgnd}tj ||f|}| || |d|||d S) za Return all allowable parser and float precision combinations and corresponding ids. valuesrr=rKrLr,rN-rR)zip _all_parsers_all_parser_idshasattrrgr r rXmark single_cpuskipifrrWappend)rSrTr@ parser_id precisionrmrWs r,_get_all_parser_float_precision_combinationsrss F C ??33 68 $ $ &]1%F7 3 3I=I-- K*K&&'0J' L&&((I!6dCCCE MM% JJ)11i11 2 2 2 2 3 S ) ))rrSrTc|jS)zR Fixture for all allowable combinations of parser and float precision r^r_s rall_parsers_all_precisionsrus  =r) )rh_utfUTFc0g|]}tD] }||d S)z{0})_encoding_prefixes).0sepprefixs r rsH!BT8>vsr)rSc|jS)zE Fixture for all possible integer values for a UTF encoding. r^r_s r utf_valuer =rc|jS)zD Fixture for all possible string formats of a UTF encoding. r^r_s r encoding_fmtrrr)*)z-1,0)z-1,2e0g333333)z-1e0r)z+1e0?)z+1e+0r)z+1e-1皙?)z+,1e1r)z+1,e0r)z-,1e1r)z-1,e0r)z0,1r)z1,r)z,1r)z-,1g)z1_,r)z1_234,56 ףp=J@)z 1_234,56e0r)rzrz)-_r)-_1r)-_1e0r)_1r)_1,r)_1,_r)_1e0r)1,2e_1r)1,2e1_0r)1,_2r),1__2r),1er)-,1er) 1_000,000_000r)1,e1_2r)e11,2r)1e11,2r)1,2,2r)1,2_1r) 1,2e-10e1r)--1,2r)1a_2,1r)z1,2E-1gQ?)z1,2E1g(@c|jS)z Fixture for all numeric formats which should get recognized. The first entry represents the value to read while the second represents the expected result. r^r_s rnumeric_decimalrs h =rcd|jvr|d}n'd|jvr|dd}ndS|jdkr7tjd}||dSdS)z Fixture that xfails a test if the engine is pyarrow. Use if failure is do to unsupported keywords or inconsistent results. r\rurNr=zpyarrow doesn't support this.rL) fixturenamesgetfixturevaluer rXrmxfail applymarker)r[r@rms r pyarrow_xfailr/s,,,((77 %)= = =(()EFFqI } !!{  (G HHD!!!!!"!rcd|jvr|d}n'd|jvr|dd}ndS|jdkrtjddSdS)z Fixture that skips a test if the engine is pyarrow. Use if failure is do a parsing failure from pyarrow.csv.read_csv r\rurNr=z,https://github.com/apache/arrow/issues/38676rL)rrr rXskip)r[r@s r pyarrow_skiprBs,,,((77 %)= = =(()EFFqI } !! IJJJJJJ"!r)6 __future__rrGrX pandas.compatrpandas.compat._optionalrpandasrrpandas._testing_testingr"r r.r4r7r9r<fixturerDrJ_cParserHighMemory_cParserLowMemory _pythonParser_pyarrowParser_py_parsers_only_c_parsers_onlyrWrmrnro_pyarrow_parsers_onlyrj_py_parser_ids _c_parser_ids_pyarrow_parsers_idsrkr\r`rcrersru _utf_values_encoding_sepsr~_encoding_fmtsrrrrrr,rrrs"""""" %%%%%%,,,,,,7/7/7/7/7/7/7/7/t;;;;;j;;; w%%%%%:%%% %%%%%J%%% ,,,DDD'$ !?%'89FL K " K  ;7Q  R R ML#3L6KL 7# !{JMJNJ5IJ|999  :9 M:::;:'^<<<=<,2FGGGHG***< 7 7 9 9( C4466u= kk U^%3 {###$#~&&&'&, , , ...^_..^"""$KKKKKr