
    Pi                         d dl Zd dlZd dlZd dlmZ d Zej	        
                    dg d          d             Zd Zd Zd Zd	 Zd
 Zd ZdS )    Nc                    t          j        g d          }t          j        g d          }|                    ||           }| dv r`t          j        ddg          }t          j        ddg          }t          j        d	d
gddgg||          }t          j        ||           d S t           j                            ddgddgg          }t          j        g d|          }t          j	        ||           d S )Nabcxr   z
align_axis)   columnsr      selfotherr   r	   r   r
   indexr   r   r	   r   r
   r   )
pdSeriescompareIndex	DataFrametmassert_frame_equal
MultiIndexfrom_productassert_series_equal)axiss1s2resultindicesr   expecteds          |/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/series/methods/test_compare.pytest_compare_axisr'      s   	???	#	#B	???	#	#BZZtZ,,F~(Aq6""(FG,--<3Z#s$GW
 
 
 	fh/////-,,q!fvw6G-HII9111AAA
vx00000    zkeep_shape, keep_equal))TF)FT)TTc                 V   t          j        g d          }t          j        g d          }|                    || |          }| rt          j        g d          }t          j        ddg          }|r!t          j        ddgd	d	gd
dgg||          }n~t          j        ddgt
          j        t
          j        gd
dgg||          }nIt          j        ddg          }t          j        ddg          }t          j        ddgd
dgg||          }t          j        ||           d S )Nr   r   )
keep_shape
keep_equalr   r   r   r   r   r   r	   r   r   r
   r   r   r   )	r   r   r   r   r   npnanr   r   )r*   r+   r!   r"   r#   r$   r   r%   s           r&   test_compare_various_formatsr/      sS    
???	#	#B	???	#	#BZZzjZIIF 
(999%%(FG,-- 		|sc3Z#s4GW  HH |sbfbf-Sz:  HH (Aq6""(FG,--<3Z#s$GW
 
 
 &(+++++r(   c                     t          j        ddt          j        g          } t          j        ddt          j        g          }|                     |          }t          j        ddggddg          }t          j        ||           d S )Nr   r   r	   r   r   )r   )r   r   r-   r.   r   r   r   r   )r!   r"   r#   r%   s       r&   test_compare_with_equal_nullsr1   A   s{     
Cbf%	&	&B	Cbf%	&	&BZZ^^F|c3ZL672CDDDH&(+++++r(   c                  ^   t          j        g d          } t          j        ddt          j        g          }|                     |d          }t           j                            ddgddgg          }t          j        d	dd
t          j        g|          }t          j        ||           d S )Nr   r	   r   r   r   r   r   r   r   r   r   )	r   r   r-   r.   r   r   r   r   r   )r!   r"   r#   r$   r%   s        r&   !test_compare_with_non_equal_nullsr3   L   s    	???	#	#B	Cbf%	&	&BZZqZ))Fm((1a&672C)DEEGy#sC0@@@H68,,,,,r(   c                     t           j                            g dg dg          } t          j        g d|           }t          j        g d|           }|                    |d          }t           j                            g dg d	g d
g          }t          j        g d|          }t          j        ||           d S )N)r   r   r   r,   r   r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   r   r   )r   r   from_arraysr   r   r   r   )r   r!   r"   r#   r$   r%   s         r&   test_compare_multi_indexr6   X   s    M%%yyy)))&<==E	???%	0	0	0B	???%	0	0	0BZZqZ))Fm''	|||%G%G%GH G y---W===H68,,,,,r(   c                     d} t          j        g dg d          }t          j        g dg d          }t          j        t          |           5  |                    |           d d d            d S # 1 swxY w Y   d S )N3Can only compare identically-labeled Series objectsr   r      r   r   )r   r   dmatchr   r   pytestraises
ValueErrorr   msgser1ser2s      r&   test_compare_different_indicesrF   f   s    
?C9YYYooo666D9YYYooo666D	z	-	-	-  T                 s   A55A9<A9c                      d} t          j        g d          }t          j        g d          }t          j        t          |           5  |                    |           d d d            d S # 1 swxY w Y   d S )Nr8   r9   )r   r   r:      r<   r>   rB   s      r&   test_compare_different_lengthsrI   n   s    
?C9YYYD9\\\""D	z	-	-	-  T                 s   
A--A14A1c                     ddddddddddd	dd
ddg} ddd}t          j        |                               |          }|d                             |d                   }|d         |d         k    }|d         |d         k    }t          j        dgdz            }t          j        dgdz            }t          j        ||           t          j        ||           t          j        ||           d S )Nz
2015-07-0108335394550)r   r   z
2015-07-02z+49 (0) 0345 300033z
2015-07-03z+49(0)2598 04457z
2015-07-04
0741470003z
2015-07-05z04181 83668zdatetime64[ns]string)datar   r   F   T)r   r   astypeeqr   r   r   )rN   dtypesdf
result_eq1
result_eq2
result_neqexpected_eqexpected_neqs           r&   "test_compare_datetime64_and_stringrY   v   s    //!677!344..//D $(33F	4	 	 	 	'	'	/	/BCBsG$$JCBsG#JCBsG#J)UGaK((K9dVaZ((L:{333:{333:|44444r(   )numpyr-   r?   pandasr   pandas._testing_testingr   r'   markparametrizer/   r1   r3   r6   rF   rI   rY    r(   r&   <module>ra      s                  1 1 1(    , , ,8, , ,	- 	- 	-- - -    5 5 5 5 5r(   