a
    Pf                     @   s6   d dl Zd dlmZmZ d dlmZ G dd dZdS )    N)	DataFrame
date_rangec                   @   s$   e Zd Zdd Zdd Zdd ZdS )
TestEqualsc                 C   sB   t ddgddgd}t ddgddgd}||du s>J d S )N      sd)abF)r   equals)selfdf1df2 r   _/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/frame/methods/test_equals.pytest_dataframe_not_equal   s    z#TestEquals.test_dataframe_not_equalc                 C   s   t ddgddgddgd}| g d }|sT|sT|jjd j|jjd jksTJ t|| ||snJ ||s|J d S )	Nxyr   r   wz)ABCr   )r   Zreset_indexZ_mgrblocksZdtypetmZassert_frame_equalr   )r   Zusing_array_managerZusing_infer_stringZdf0r   r   r   r   test_equals_different_blocks   s     z'TestEquals.test_equals_different_blocksc                 C   sd  t jdd}tt jdd|dgd}d |d< tdddd	|d
< tdddd	|d< |d |d
  |d< t dd dkt|d< t j	|j
d d d< | }|d |d sJ |d
 |d
 sJ |d |d sJ |d |d sJ |d |d sJ ||s,J |tr<J | }|d d|d< ||rfJ | }||}||rJ | }|jd d d |_||rJ tdddd	}||}| }||sJ |jdgdd}|jdgdd}||sJ |jdgdd}||r0J |jdgdd}|jdgdd}||s`J d S )Nr   
   Zfloats)indexcolumnsz0the sky is so blue. we could use more chocolate.textz2000-1-1min)ZperiodsfreqstartDenddiff   r   boolZfloat32T)append)nprandomZdefault_rngr   splitr   ZarangeZastypeobjectnanloccopyr   Z	set_indexr   )r   r   r   r   Z	differentZdifferent_indexZdf3r   r   r   test_equals   sN    

zTestEquals.test_equalsN)__name__
__module____qualname__r   r   r1   r   r   r   r   r   
   s   r   )	numpyr*   Zpandasr   r   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   