U
    Mf{                     @   s   d Z ddlmZ ddlZddlZddlmZmZ ddl	m
Z ejdZeejdddgd	d
 Zeejdddgdd Zdd ZdS )zy
Tests that work on both the Python and C engines but do not have a
specific classification into the other test modules.
    )StringION)	DataFrameoption_contextZpyarrow_xfail	na_filterTFc                 C   s`   | }d}t dtdtdgd idddd	d
dddddg
d}|jt|d|d}t|| d S )NzE,A
a,inf
b,-inf
c,+Inf
d,-Inf
e,INF
f,-INF
g,+INf
h,-INf
i,inF
j,-inFAinfz-inf   abcdefghijindexr   Z	index_colr   r   floatread_csvr   tmassert_frame_equalall_parsersr   parserdataexpectedresult r!   J/tmp/pip-unpacked-wheel-eb6vo0j3/pandas/tests/io/parser/common/test_inf.pytest_inf_parsing   s    r#   c                 C   sT   | }d}t dtdtdtdgidddgd	}|jt|d
|d}t|| d S )Nz&,A
a,Infinity
b,-Infinity
c,+Infinity
r   Zinfinityz	-infinityz	+infinityr	   r
   r   r   r   r   r   r   r!   r!   r"   test_infinity_parsing+   s    r$   c              	   C   sP   | }d}t dd |jt|d d}W 5 Q R X tdtjdg}t|| d S )Nz1.0
NaN
3.0Zuse_inf_as_naT)headerg      ?g      @)r   r   r   r   npnanr   r   )r   r   r   r    r   r!   r!   r"    test_read_csv_with_use_inf_as_na=   s    r(   )__doc__ior   Znumpyr&   ZpytestZpandasr   r   Zpandas._testingZ_testingr   markZusefixturesZxfail_pyarrowZparametrizer#   r$   r(   r!   r!   r!   r"   <module>   s   