a 鴫j#@sJddlmZddlZddlZddlmZddlmZddlm Z m Z ddl m Z GdddZGdd d eZGd d d eZGd d d eZGdddeZGdddeZejddZejddZeZeZeZeZegZeegZejeejjejj e ddgdgZ!geee!Z"dgZ#ddgZ$dgZ%ge$e#e%Z&eje"e&dddZ'ejee$dd d!Z(ejee#dd"d#Z)eje!e%dd$d%Z*d&d'Z+eje+d(e+d)dd*d+Z,gd,Z-gd-Z.d.d/gZ/d0d1e.DZ0eje-d2d3d4Z1eje0d2d5d6Z2ejgd7d2d8d9Z3ejd:d;Z4ejd) annotationsN) HAS_PYARROW)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 )selfkwargsr^/var/www/html/laksyashrimp/venv/lib/python3.9/site-packages/pandas/tests/io/parser/conftest.py update_kwargsszBaseParser.update_kwargscOs||}t|i|SN)rrr argsrrrrrs 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 rrrrrrrrrread_csv_check_warnings s z"BaseParser.read_csv_check_warningscOs||}t|i|Sr)rrrrrrr5s zBaseParser.read_table)r)rrcOsL||}tj|||dt|i|WdS1s>0YdS)N)rr)rrrr)r rrrrrrrrread_table_check_warnings9s z$BaseParser.read_table_check_warnings) __name__ __module__ __qualname__r__annotations__r r rrrrrrrrrrs     rc@seZdZdZgdZdS)CParserc)NhighZ round_tripNr r!r"rr rrrrr$Ksr$c@seZdZdZdS)CParserHighMemoryFNr r!r"r rrrrr(Psr(c@seZdZdZdS)CParserLowMemoryTNr)rrrrr*Tsr*c@seZdZdZdgZdS) PythonParserpythonNr'rrrrr+Xsr+c@seZdZdZdgZdS) PyArrowParserpyarrowNr'rrrrr-]sr-cCs |dddS)zG The directory path to the data files needed for parser tests. ioparserdatardatapathrrr csv_dir_pathbsr4cCstj|ddddS)zH The path to the data file "test1.csv" needed for parser tests. r/r1csvz test1.csv)ospathjoinr2rrrcsv1jsr9pyarrow is not installedreasonZmarksr,Zc_highZc_lowr.paramsidscCs8|}|jdkr4tdtdddl}|d|S)z) Fixture all of the CSV parsers. r.rN)paramrpytestZ importorskiprr.Z set_cpu_count)requestr0r.rrr all_parserss   rEcCs|S)z< Fixture all of the CSV parsers using the C engine. rBrDrrr c_parser_onlysrHcCs|S)zA Fixture all of the CSV parsers using the Python engine. rFrGrrrpython_parser_onlysrIcCs|S)zB Fixture all of the CSV parsers using the Pyarrow engine. rFrGrrrpyarrow_parser_onlysrJcCsg}g}tttD]\}}t|dr.|jd}|jD]^}|jdkr\tjj tjj t ddgnd}tj ||f|d}| || |d|q4q||d S) za Return all allowable parser and float precision combinations and corresponding ids. valuesrr.r:r;rr=-r>)zip _all_parsers_all_parser_idshasattrrKr rrCmark single_cpuskipifrrBappend)r?r@r0Z parser_id precisionrQrBrrr,_get_all_parser_float_precision_combinationss&     rVr?r@cCs|jS)zR Fixture for all allowable combinations of parser and float precision rFrGrrrall_parsers_all_precisionss rW) )rL_utfZUTFcCs$g|]}tD]}||dq qS)z{0})_encoding_prefixes).0sepprefixrrr srb)r?cCs|jS)zE Fixture for all possible integer values for a UTF encoding. rFrGrrr utf_valuesrccCs|jS)zD Fixture for all possible string formats of a UTF encoding. rFrGrrr encoding_fmtsrd)*)z-1,0)z-1,2e0g333333)z-1e0re)z+1e0?)z+1e+0rf)z+1e-1皙?)z+,1e1rf)z+1,e0rf)z-,1e1re)z-1,e0re)z0,1rg)z1,rf)z,1rg)z-,1g)z1_,rf)z1_234,56 ףp=J@)z 1_234,56e0rh)r\r\)-_ri)-_1rj)-_1e0rk)_1rl)_1,rm)_1,_rn)_1e0ro)1,2e_1rp)1,2e1_0rq)1,_2rr),1__2rs),1ert)-,1eru) 1_000,000_000rv)1,e1_2rw)e11,2rx)1e11,2ry)1,2,2rz)1,2_1r{) 1,2e-10e1r|)--1,2r})1a_2,1r~)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. rFrGrrrnumeric_decimals4rcCsZd|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. rErWrNr.zpyarrow doesn't support this.r;) fixturenamesgetfixturevaluerrCrQZxfailZ applymarker)rDr0rQrrr pyarrow_xfail/s    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 rErWrNr.z,https://github.com/apache/arrow/issues/38676r;)rrrrCskip)rDr0rrr pyarrow_skipBs    r)6 __future__rr6rCZ pandas.compatrZpandas.compat._optionalrZpandasrrZpandas._testingZ_testingrrr$r(r*r+r-Zfixturer4r9Z_cParserHighMemoryZ_cParserLowMemoryZ _pythonParserZ_pyarrowParserZ_py_parsers_onlyZ_c_parsers_onlyrBrQrRrSZ_pyarrow_parsers_onlyrNZ_py_parser_idsZ _c_parser_idsZ_pyarrow_parsers_idsrOrErHrIrJrVrWZ _utf_valuesZ_encoding_sepsr^Z_encoding_fmtsrcrdrrrrrrrs    :                /