a Pf"@s,ddlmZddlZddlZddlmZddlmZmZddl m Z GdddZ Gddde Z Gd d d e ZGd d d e ZGd dde ZGddde ZejddZejddZeZeZeZeZegZeegZejeejjdgZgeeeZdgZ ddgZ!dgZ"ge!e e"Z#ejee#dddZ$ejee!dddZ%ejee ddd Z&ejee"dd!d"Z'd#d$Z(eje(d%e(d&dd'd(Z)gd)Z*gd*Z+d+d,gZ,d-d.e+DZ-eje*d/d0d1Z.eje-d/d2d3Z/ejgd4d/d5d6Z0ejd7d8Z1ejd9d:Z2dS);) annotationsN)VERSIONS)read_csv read_tablec@sreZdZUdZded<dZgZded<ddZd d Zddd d d ddddZ ddZ ddd d dddZ dS) BaseParserNz str | NoneengineTzlist[str | None]float_precision_choicescCs |}||j|jd|S)N)r low_memory)copyupdaterr )selfkwargsrX/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/io/parser/conftest.py update_kwargsszBaseParser.update_kwargscOs||}t|i|SN)rrr argsr rrrrs zBaseParser.read_csv)raise_on_extra_warningscheck_stacklevelz type[Warning]strbool) warn_typewarn_msgrcOsN||}tj||||dt|i|WdS1s@0YdS)N)matchrr)rtmassert_produces_warningr)r rrrrrr rrrread_csv_check_warningss z"BaseParser.read_csv_check_warningscOs||}t|i|Sr)rrrrrrr4s zBaseParser.read_table)r)rrcOsL||}tj|||dt|i|WdS1s>0YdS)N)rr)rrrr)r rrrrr rrrread_table_check_warnings8s z$BaseParser.read_table_check_warnings) __name__ __module__ __qualname__r__annotations__r rrrrrrrrrrrs     rc@seZdZdZgdZdS)CParserc)NhighZ round_tripNrr r!rrrrrrr#Jsr#c@seZdZdZdS)CParserHighMemoryFNrr r!r rrrrr'Osr'c@seZdZdZdS)CParserLowMemoryTNr(rrrrr)Ssr)c@seZdZdZdgZdS) PythonParserpythonNr&rrrrr*Wsr*c@seZdZdZdgZdS) PyArrowParserpyarrowNr&rrrrr,\sr,cCs |dddS)zG The directory path to the data files needed for parser tests. ioparserdatardatapathrrr csv_dir_pathasr3cCstj|ddddS)zH The path to the data file "test1.csv" needed for parser tests. r.r0csvz test1.csv)ospathjoinr1rrrcsv1isr8Zmarksr+Zc_highZc_lowr-paramsidscCs8|}|jdkr4tdtdddl}|d|S)z) Fixture all of the CSV parsers. r-rN)paramrpytestZ importorskiprr-Z set_cpu_count)requestr/r-rrr all_parserss   rAcCs|S)z< Fixture all of the CSV parsers using the C engine. r>r@rrr c_parser_onlysrDcCs|S)zA Fixture all of the CSV parsers using the Python engine. rBrCrrrpython_parser_onlysrEcCs|S)zB Fixture all of the CSV parsers using the Pyarrow engine. rBrCrrrpyarrow_parser_onlysrFcCsg}g}tttD]p\}}t|dr.|jd}|jD]L}|jdkrJtjj nd}tj ||f|d}| || |d|q4q||dS)za Return all allowable parser and float precision combinations and corresponding ids. valuesrr-rr9-r:) zip _all_parsers_all_parser_idshasattrrGrrr?mark single_cpur>append)r;r<r/Z parser_id precisionrMr>rrr,_get_all_parser_float_precision_combinationss    rQr;r<cCs|jS)zR Fixture for all allowable combinations of parser and float precision rBrCrrrall_parsers_all_precisionss rR) )rH_utfZUTFcCs$g|]}tD]}||dq qS)z{0})_encoding_prefixes).0sepprefixrrr sr])r;cCs|jS)zE Fixture for all possible integer values for a UTF encoding. rBrCrrr utf_valuesr^cCs|jS)zD Fixture for all possible string formats of a UTF encoding. rBrCrrr encoding_fmtsr_)*)z-1,0)z-1,2e0g333333)z-1e0r`)z+1e0?)z+1e+0ra)z+1e-1皙?)z+,1e1ra)z+1,e0ra)z-,1e1r`)z-1,e0r`)z0,1rb)z1,ra)z,1rb)z-,1g)z1_,ra)z1_234,56 ףp=J@)z 1_234,56e0rc)rWrW)-_rd)-_1re)-_1e0rf)_1rg)_1,rh)_1,_ri)_1e0rj)1,2e_1rk)1,2e1_0rl)1,_2rm),1__2rn),1ero)-,1erp) 1_000,000_000rq)1,e1_2rr)e11,2rs)1e11,2rt)1,2,2ru)1,2_1rv) 1,2e-10e1rw)--1,2rx)1a_2,1ry)z1,2E-1gQ?)z1,2E1g(@cCs|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. rBrCrrrnumeric_decimals4rzcCsZd|jvr|d}nd|jvr0|dd}ndS|jdkrVtjjdd}||dS)z Fixture that xfails a test if the engine is pyarrow. Use if failure is do to unsupported keywords or inconsistent results. rArRrNr-zpyarrow doesn't support this.reason) fixturenamesgetfixturevaluerr?rMZxfailZ applymarker)r@r/rMrrr pyarrow_xfails    rcCsNd|jvr|d}nd|jvr0|dd}ndS|jdkrJtjdddS)z Fixture that skips a test if the engine is pyarrow. Use if failure is do a parsing failure from pyarrow.csv.read_csv rArRrNr-z,https://github.com/apache/arrow/issues/38676r{)r}r~rr?skip)r@r/rrr pyarrow_skip0s    r)3 __future__rr5r?Zpandas.compat._optionalrZpandasrrZpandas._testingZ_testingrrr#r'r)r*r,Zfixturer3r8Z_cParserHighMemoryZ_cParserLowMemoryZ _pythonParserZ_pyarrowParserZ_py_parsers_onlyZ_c_parsers_onlyr>rMrNZ_pyarrow_parsers_onlyrJZ_py_parser_idsZ _c_parser_idsZ_pyarrow_parsers_idsrKrArDrErFrQrRZ _utf_valuesZ_encoding_sepsrYZ_encoding_fmtsr^r_rzrrrrrrsv   :               /