a
    Pf                     @   s6   d dl m Z  d dlmZ d dlmZ G dd dZdS )    )datetime)	DataFrameNc                   @   s   e Zd Zdd ZdS )TestInferObjectsc              
   C   s   t g dg ddtdddtdddtdddgg dd	g d	d
}|jdd   }|d jdksjJ |d jdks|J |d jdksJ |d jdksJ t g dg dtdddtdddtdddgg dd	g d	d
}|jdd}t|| d S )N)a         )b       @      @ffffff@ci  r   r   r   )r   r   r   d)r   r	   r   r   )columnsr   Zint64r	   Zfloat64zM8[ns]r   object)r   r   r   )r
   r   r   )r   r   r   T)Zdrop)r   r   ZilocZinfer_objectsZdtypeZreset_indextmZassert_frame_equal)selfZdfexpectedresult r   f/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/frame/methods/test_infer_objects.pytest_infer_objects   s6    


 
z#TestInferObjects.test_infer_objectsN)__name__
__module____qualname__r   r   r   r   r   r      s   r   )r   Zpandasr   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   