
    .iO                     @    d dl Zd dlmZmZmZ d dlmZ  G d d      Z	y)    N)Categorical	DataFrame
date_rangec                   $    e Zd Zd Zd Zd Zd Zy)
TestEqualsc                 v    t        ddgddgd      }t        ddgddgd      }|j                  |      du sJ y )N      sd)abF)r   equals)selfdf1df2s      d/var/www/app/trading-bot/venv/lib/python3.12/site-packages/pandas/tests/frame/methods/test_equals.pytest_dataframe_not_equalz#TestEquals.test_dataframe_not_equal   sF    q!fC:67sCj1v67zz#%'''    c                 f   t        ddgddgddgd      }|j                         g d   }|sI|j                  j                  d   j                  |j                  j                  d   j                  k7  sJ t        j                  ||       |j                  |      sJ |j                  |      sJ y )	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Hoo0! 88??1%++sxxq/A/G/GGGG 	c3'zz#zz#r   c                 `   t         j                  j                  d      j                  d      }t        t         j                  j                  d      j                  d      |dg      }dj	                         |d<   t        ddd	      |d
<   t        ddd	      |d<   |d   |d
   z
  |d<   t        j                  d      dz  dk(  j                  t              |d<   t         j                  |j                  d d d<   |j                         }|d   j                  |d         sJ |d
   j                  |d
         sJ |d   j                  |d         sJ |d   j                  |d         sJ |d   j                  |d         sJ |j                  |      sJ |j                  t              rJ |j                         }|d   j                  d      |d<   |j                  |      rJ | }|j                  |      }|j                  |      rJ |j                         }|j                  d d d   |_        |j                  |      rJ t        ddd	      }|j                  |      }|j                         }|j                  |      sJ |j                  dgd      }|j                  dgd      }|j                  |      sJ |j                  dgd      }|j                  |      rJ |j                  dgd      }|j                  dgd      }|j                  |      sJ y )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II!!!$++B/uxj
 INNPF!*buEG
BSAE
%j3w</Fyy}q(A-55f=Fvv!hhj6{!!#f+...7|""3w<0005z  U,,,6{!!#f+...6{!!#f+...zz#::f%%% HHJ	'188C	(::i((( !&MM/2	::i((( HHJ	KK"-	::i((( :r>mmE"hhjzz# mmVHTm2mmVHTm2zz#mmXJtm4::c?"" mmXJtm4mmXJtm4zz#r   c                    t        g dddg      }t        g dddg      }t        d|i      }t        d|i      }|j                  |      sJ t        g dg d      }t        d|i      }|j                  |      rJ y )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?SzBd$d$zz#?Gd$::c?""?r   N)__name__
__module____qualname__r   r&   rG   rN    r   r   r   r      s    (6p#r   r   )
numpyr9   pandasr   r   r   pandas._testing_testingr"   r   rR   r   r   <module>rW      s#     
 X# X#r   