
    PiO                     H    d dl Zd dlmZmZmZ d dlmZ  G d d          Z	dS )    N)Categorical	DataFrame
date_rangec                   &    e Zd Zd Zd Zd Zd ZdS )
TestEqualsc                     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df2s      z/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/frame/methods/test_equals.pytest_dataframe_not_equalz#TestEquals.test_dataframe_not_equal   s]    q!fC:6677sCj1v6677zz#%''''''    c                 ^   t          ddgddgddgd          }|                                g d         }|s2|j        j        d         j        |j        j        d         j        k    sJ t          j        ||           |                    |          sJ |                    |          sJ d S )	Nxyr	   r
   wz)ABCr   )r   reset_index_mgrblocksdtypetmassert_frame_equalr   )r   using_infer_stringdf0r   s       r   test_equals_different_blocksz'TestEquals.test_equals_different_blocks   s    sCj1vS#JGGHHoo0! 	H 8?1%+sxq/A/GGGGG 	c3'''zz#zz#r   c                    t           j                            d                              d          }t          t           j                            d                              d          |dg          }d                                |d<   t          ddd	          |d
<   t          ddd	          |d<   |d         |d
         z
  |d<   t          j        d          dz  dk                        t                    |d<   t           j	        |j
        d d d<   |                                }|d                             |d                   sJ |d
                             |d
                   sJ |d                             |d                   sJ |d                             |d                   sJ |d                             |d                   sJ |                    |          sJ |                    t                    rJ |                                }|d                             d          |d<   |                    |          rJ | }|                    |          }|                    |          rJ |                                }|j        d d d         |_        |                    |          rJ t          ddd	          }|                    |          }|                                }|                    |          sJ |                    dgd          }|                    dgd          }|                    |          sJ |                    dgd          }|                    |          rJ |                    dgd          }|                    dgd          }|                    |          sJ d S )Nr
   
   floats)indexcolumnsz0the sky is so blue. we could use more chocolate.textz2000-1-1min)periodsfreqstartDenddiff   r   boolfloat32T)append)nprandomdefault_rngr   splitr   arangeastypeobjectnanloccopyr   	set_indexr+   )r   r*   r   r   	differentdifferent_indexdf3s          r   test_equalszTestEquals.test_equals    s   	%%a((//33I!!!$$++B//uxj
 
 
 INNPPF!*buEEEG
BSAAAE
%j3w</Fy}}q(A-55f==Fv!hhjj6{!!#f+.....7|""3w<000005z  U,,,,,6{!!#f+.....6{!!#f+.....zz#::f%%%%% HHJJ	'188CC	(::i((((( !&MM/22	::i((((( HHJJ	K"-	::i((((( :r>>>mmE""hhjjzz# mmVHTm22mmVHTm22zz#mmXJtm44::c??""" mmXJtm44mmXJtm44zz#r   c                 F   t          g dddg          }t          g dddg          }t          d|i          }t          d|i          }|                    |          sJ t          g dg d          }t          d|i          }|                    |          rJ d S )N)r   r   r   r   r   )
categoriesc)r   r   rJ   )r   r   r   )r   cat1cat2r   r   cat3rF   s          r   (test_equals_categorical_categories_orderz3TestEquals.test_equals_categorical_categories_orderX   s    ???SzBBB???SzBBBd$$d$$zz#???GGGd$$::c??"""""r   N)__name__
__module____qualname__r   r&   rG   rN    r   r   r   r      sQ        ( ( (  6 6 6p# # # # #r   r   )
numpyr9   pandasr   r   r   pandas._testing_testingr"   r   rR   r   r   <module>rW      s                
      X# X# X# X# X# X# X# X# X# X#r   