eh""ddlmZddlZddlZddlmZddlmZmZddl m Z GddZ Gdde Z Gd d e ZGd d e ZGd de ZGdde Zej$dZej$dZeZeZeZeZegZeegZej6eej8j:gZgeeeZdgZ ddgZ!dgZ"ge!e e"Z#ej$ee#dZ$ej$ee!dZ%ej$ee dZ&ej$ee"dZ'dZ(ej$e(de(dd Z)gd!Z*gd"Z+d#d$gZ,e+Dcgc]}e,D] }||d% c}}Z-ej$e*&d'Z.ej$e-&d(Z/ej$gd)&d*Z0ej$d+Z1ej$d,Z2ycc}}w)-) annotationsN)VERSIONS)read_csv read_tablecneZdZUdZded<dZgZded<dZdZddd  dd Z d Z dd  dd Z y) BaseParserNz str | NoneengineTzlist[str | None]float_precision_choicescv|j}|j|j|jd|S)N)r low_memory)copyupdater r )selfkwargss ]/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/tests/io/parser/conftest.py update_kwargszBaseParser.update_kwargss+ DOOLM c:|j|}t|i|SN)rrrargsrs rrzBaseParser.read_csvs"##F+(((r)raise_on_extra_warningscheck_stacklevelc|j|}tj||||5t|i|cdddS#1swYyxYw)N)matchrr)rtmassert_produces_warningr)r warn_typewarn_msgrrrrs rread_csv_check_warningsz"BaseParser.read_csv_check_warningssR##F+  ' ' $;-   - T,V,  - - -s AA c:|j|}t|i|Sr)rrrs rrzBaseParser.read_table4s"##F+4*6**r)rc|j|}tj|||5t|i|cdddS#1swYyxYw)N)rr)rrrr)rrrrrrs rread_table_check_warningsz$BaseParser.read_table_check_warnings8sN##F+  ' ' X?V  /t.v. / / /s ?A)r type[Warning]rstrrbool)rr$rr%) __name__ __module__ __qualname__r __annotations__r r rrr rr#rrrrsqFJJ02-2 )!%!% - --  -*+!% / //rrceZdZdZgdZy)CParserc)Nhigh round_tripNr'r(r)r r r+rrr-r-Js F:rr-ceZdZdZy)CParserHighMemoryFNr'r(r)r r+rrr3r3OsJrr3ceZdZdZy)CParserLowMemoryTNr4r+rrr6r6SsJrr6ceZdZdZdgZy) PythonParserpythonNr1r+rrr8r8Ws F#frr8ceZdZdZdgZy) PyArrowParserpyarrowNr1r+rrr;r;\s F#frr;c|dddS)zG The directory path to the data files needed for parser tests. ioparserdatar+datapaths r csv_dir_pathrCas D(F ++rcRtjj|ddddS)zH The path to the data file "test1.csv" needed for parser tests. r>r@csvz test1.csv)ospathjoinrAs rcsv1rIis" 77<<vu5{ CCrmarksr9c_highc_lowr<paramsidsc|j}|jdk(r2tjdtdddl}|j d|S)z) Fixture all of the CSV parsers. r<rN)paramr pytest importorskiprr< set_cpu_count)requestr?r<s r all_parsersrXsH ]]_F }} !Ix ':; a Mrc"|jS)z< Fixture all of the CSV parsers using the C engine. rSrWs r c_parser_onlyr\ ==?rc"|jS)zA Fixture all of the CSV parsers using the Python engine. rZr[s rpython_parser_onlyr_r]rc"|jS)zB Fixture all of the CSV parsers using the Pyarrow engine. rZr[s rpyarrow_parser_onlyrar]rcg}g}tttD]\}}t|dr|jd}|j D]r}|j dk(rtjjnd}tj||f|}|j||j|d|t||dS)za Return all allowable parser and float precision combinations and corresponding ids. valuesrr<r+rJ-rN) zip _all_parsers_all_parser_idshasattrrcr r rTmark single_cpurSappend)rOrPr? parser_id precisionrirSs r,_get_all_parser_float_precision_combinationsrns F C ? 3 68 $]]1%F77 3I-3]]i-G6;;))RDLL&(I!6dCE MM% JJ)Ai[1 2  3 3S ))rrOrPc|jS)zR Fixture for all allowable combinations of parser and float precision rZr[s rall_parsers_all_precisionsrps ==r) )rd_utfUTFz{0})rOc|jS)zE Fixture for all possible integer values for a UTF encoding. rZr[s r utf_valuery ==rc|jS)zD Fixture for all possible string formats of a UTF encoding. rZr[s r encoding_fmtr|rzr)*)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)ruru)-_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. rZr[s rnumeric_decimalrsh ==rcd|jvr|jd}n$d|jvr|jdd}ny|jdk(r2tjj d}|j |yy)z Fixture that xfails a test if the engine is pyarrow. Use if failure is do to unsupported keywords or inconsistent results. rXrprNr<zpyarrow doesn't support this.reason) fixturenamesgetfixturevaluer rTrixfail applymarker)rWr?ris r pyarrow_xfailrs,,,((7 %)=)= =(()EFqI }} !{{  (G HD!"rcd|jvr|jd}n$d|jvr|jdd}ny|jdk(rtjdyy)z Fixture that skips a test if the engine is pyarrow. Use if failure is do a parsing failure from pyarrow.csv.read_csv rXrprNr<z,https://github.com/apache/arrow/issues/38676r)rrr rTskip)rWr?s r pyarrow_skipr0sg,,,((7 %)=)= =(()EFqI }} ! IJ"r)3 __future__rrFrTpandas.compat._optionalrpandasrrpandas._testing_testingrrr-r3r6r8r;fixturerCrI_cParserHighMemory_cParserLowMemory _pythonParser_pyarrowParser_py_parsers_only_c_parsers_onlyrSrirj_pyarrow_parsers_onlyrf_py_parser_ids _c_parser_ids_pyarrow_parsers_idsrgrXr\r_rarnrp _utf_values_encoding_seps_encoding_prefixes_encoding_fmtsryr|rrr)sepprefixs00rrs" ,7/7/t;j; w%:% %J% ,,DD'$ !?%'89%nFKKvhse5 {#$~&', .^_.^""$KKAsH