o
    d                     @  s,  d dl mZ d dlZd dlZd dlmZ d dlmZmZ d dl	m
Z G dd dZG dd deZG d	d
 d
eZG dd deZG dd deZG dd deZejdd Zejdd ZeZeZeZeZegZeegZejeejjdgZg eeeZdgZ ddgZ!dgZ"g e!e e"Z#ejee#ddd Z$ejee!ddd Z%ejee ddd  Z&ejee"dd!d" Z'd#d$ Z(eje( d% e( d& dd'd( Z)g d)Z*g d*Z+d+d,gZ,d-d. e+D Z-eje*d/d0d1 Z.eje-d/d2d3 Z/ejg d4d/d5d6 Z0ejd7d8 Z1ejd9d: Z2dS );    )annotationsN)VERSIONS)read_csv
read_tablec                   @  sV   e Zd ZU dZded< dZg Zded< dd Zd	d
 ZdddZ	dd Z
dddZdS )
BaseParserNz
str | NoneengineTzlist[str | None]float_precision_choicesc                 C  s    |  }|| j| jd |S )N)r   
low_memory)copyupdater   r	   )selfkwargs r   d/var/www/html/visualizacion-main/env/lib/python3.10/site-packages/pandas/tests/io/parser/conftest.pyupdate_kwargs   s   zBaseParser.update_kwargsc                 O     |  |}t|i |S N)r   r   r   argsr   r   r   r   r         
zBaseParser.read_csv	warn_typetype[Warning]warn_msgstrc                 O  J   |  |}tj||d t|i |W  d    S 1 sw   Y  d S N)match)r   tmassert_produces_warningr   r   r   r   r   r   r   r   r   read_csv_check_warnings      
$z"BaseParser.read_csv_check_warningsc                 O  r   r   )r   r   r   r   r   r   r   )   r   zBaseParser.read_tablec                 O  r   r   )r   r   r   r   r   r   r   r   read_table_check_warnings-   r!   z$BaseParser.read_table_check_warnings)r   r   r   r   )__name__
__module____qualname__r   __annotations__r	   r   r   r   r    r   r"   r   r   r   r   r      s   
 

r   c                   @  s   e Zd ZdZg dZdS )CParserc)Nhigh
round_tripNr#   r$   r%   r   r   r   r   r   r   r'   8   s    r'   c                   @     e Zd ZdZdS )CParserHighMemoryFNr#   r$   r%   r	   r   r   r   r   r-   =       r-   c                   @  r,   )CParserLowMemoryTNr.   r   r   r   r   r0   A   r/   r0   c                   @     e Zd ZdZdgZdS )PythonParserpythonNr+   r   r   r   r   r2   E       
r2   c                   @  r1   )PyArrowParserpyarrowNr+   r   r   r   r   r5   J   r4   r5   c                 C  s   | dddS )zG
    The directory path to the data files needed for parser tests.
    ioparserdatar   datapathr   r   r   csv_dir_pathO   s   r<   c                 C  s   t j| ddddS )zH
    The path to the data file "test1.csv" needed for parser tests.
    r7   r9   csvz	test1.csv)ospathjoinr:   r   r   r   csv1W   s   rA   marksr3   c_highc_lowr6   paramsidsc                 C  s8   |   }|jdkrtdtd  ddl}|d |S )z)
    Fixture all of the CSV parsers.
    r6   r   N   )paramr   pytestimportorskipr   r6   set_cpu_count)requestr8   r6   r   r   r   all_parsersq   s   

rO   c                 C     |   S )z<
    Fixture all of the CSV parsers using the C engine.
    rJ   rN   r   r   r   c_parser_only      rS   c                 C  rP   )zA
    Fixture all of the CSV parsers using the Python engine.
    rQ   rR   r   r   r   python_parser_only   rT   rU   c                 C  rP   )zB
    Fixture all of the CSV parsers using the Pyarrow engine.
    rQ   rR   r   r   r   pyarrow_parser_only   rT   rV   c                  C  s   g } g }t ttD ]8\}}t|dr|jd }|jD ]&}|jdkr%tjj	nd}tj
| |f|d}| | || d|  qq	| |dS )za
    Return all allowable parser and float precision
    combinations and corresponding ids.
    valuesr   r6   r   rB   -rF   )zip_all_parsers_all_parser_idshasattrrW   r   r   rK   mark
single_cpurJ   append)rG   rH   r8   	parser_id	precisionr]   rJ   r   r   r   ,_get_all_parser_float_precision_combinations   s   




rb   rG   rH   c                 C     | j S )zR
    Fixture for all allowable combinations of parser
    and float precision
    rQ   rR   r   r   r   all_parsers_all_precisions   s   	rd   )          ) rX   _utfUTFc                 C  s&   g | ]}t D ]
}| | d  qqS )z{0})_encoding_prefixes).0sepprefixr   r   r   
<listcomp>   s
    rp   )rG   c                 C  rc   )zE
    Fixture for all possible integer values for a UTF encoding.
    rQ   rR   r   r   r   	utf_value      rq   c                 C  rc   )zD
    Fixture for all possible string formats of a UTF encoding.
    rQ   rR   r   r   r   encoding_fmt   rr   rs   )*)z-1,0      )z-1,2e0g333333)z-1e0rt   )z+1e0      ?)z+1e+0ru   )z+1e-1皙?)z+,1e1ru   )z+1,e0ru   )z-,1e1rt   )z-1,e0rt   )z0,1rv   )z1,ru   )z,1rv   )z-,1g)z1_,ru   )z1_234,56
ףp=J@)z
1_234,56e0rw   )ri   ri   )-_rx   )-_1ry   )-_1e0rz   )_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   )z1,2E-1gQ?)z1,2E1g      (@c                 C  rc   )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.
    rQ   rR   r   r   r   numeric_decimal   s   4r   c                 C  s`   d| j v r| d}nd| j v r| dd }ndS |jdkr.tjjdd}| j| dS dS )z>
    Fixture that xfails a test if the engine is pyarrow.
    rO   rd   r   Nr6   pyarrow doesn't support this.)reason)fixturenamesgetfixturevaluer   rK   r]   xfailnode
add_marker)rN   r8   r]   r   r   r   pyarrow_xfail  s   


r   c                 C  sP   d| j v r| d}nd| j v r| dd }ndS |jdkr&td dS dS )z=
    Fixture that skips a test if the engine is pyarrow.
    rO   rd   r   Nr6   r   )r   r   r   rK   skip)rN   r8   r   r   r   pyarrow_skip  s   


r   )3
__future__r   r>   rK   pandas.compat._optionalr   pandasr   r   pandas._testing_testingr   r   r'   r-   r0   r2   r5   fixturer<   rA   _cParserHighMemory_cParserLowMemory_pythonParser_pyarrowParser_py_parsers_only_c_parsers_onlyrJ   r]   r^   _pyarrow_parsers_onlyrZ   _py_parser_ids_c_parser_ids_pyarrow_parsers_idsr[   rO   rS   rU   rV   rb   rd   _utf_values_encoding_sepsrl   _encoding_fmtsrq   rs   r   r   r   r   r   r   r   <module>   sx    (











/
