
    bMh                     b   d dl Zd dlZd dlZd dlmZ ej        	                    dej
        ej        ej        ej        g          d             Zej        	                    dej        g          d             Zd Zej        	                    dddgddgg          d	             Zd
 Zd Zd ZdS )    Nufuncc                 Z   t          j        g dd          } | ||          }t          j         | |j        |j                  d          }t          j        ||j        <   t          j        ||           t          j        |          } | ||          }t          j         | |j        |j                  d          }t          j        ||j        <   t          j	        ||           t          j        g d          } | ||          }t          j         | |j        |          d          }t          j        ||j        <   t          j        ||            | ||          }t          j         | ||j                  d          }t          j        ||j        <   t          j        ||            | |d          }t          j         | |j        d          d          }t          j        ||j        <   t          j        ||            | d|          }t          j         | d|j                  d          }t          j        ||j        <   t          j        ||           d}t          j        t          |          5   | |d           d d d            d S # 1 swxY w Y   d S )	NTFNbooleandtype)TTFTzBoperand type\(s\) all returned NotImplemented from __array_ufunc__)matchtest)pdarray_datanpnan_masktmassert_extension_array_equalSeriesassert_series_equalpytestraises	TypeError)r   aresultexpectedsarrmsgs          i/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/arrays/boolean/test_function.pytest_ufuncs_binaryr      s   
 	$$$I666AU1a[[Fxagqw//yAAAHHQW#FH555
	!AU1a[[Fyqw00	BBBHHQW68,,, (&&&
'
'CU1c]]Fxags++9===HHQW#FH555U3]]Fxc17++9===HHQW#FH555 U1d^^Fxagt,,I>>>HHQW#FH555U4^^FxdAG,,I>>>HHQW#FH555 PC	y	,	,	,  a                 s   J  J$'J$c                    t          j        g dd          } | |          }t          j         | |j                  d          }t          j        ||j        <   t          j        ||           t          j        |          } | |          }t          j         | |j                  d          }t          j        ||j        <   t          j	        ||           d S )Nr   r   r   )
r   r   r   r   r   r   r   r   r   r   )r   r   r   r   sers        r   test_ufuncs_unaryr"   6   s    
$$$I666AU1XXFxagi888HHQW#FH555
)A,,CU3ZZFyqwy999HHQW68,,,,,    c                      t          j        g dd          } t          j        |           }t          j        g dd          }t	          j        ||           d S )Nr   r   r   )   r   NFloat32)r   r   r   sqrtr   r   )r   resr   s      r   test_ufunc_numericr)   E   s]     (&&&i
8
8
8C
'#,,CxI666H#C22222r#   valuesTFc                    t          j        | d          }t          j                            |          }|d         t           j        u rt           j        }n|j                                        }t          j	        ||           d S )Nr   r   )
r   r   r   addreduceNAr   sumr   assert_almost_equal)r*   r   r(   r   s       r   test_ufunc_reduce_raisesr2   P   sk    
(6
+
+
+C
&--

C
2w"%59==??3)))))r#   c                     t          j        ddt           j        gd          } |                     d          }t          j        g d| dd	          }|j        j        | j        k    sJ t          j        ||           |                     d          }t          j        d
d
g| d d         dd	          }|j        j        | j        k    sJ t          j        ||           d S )NTFr   r   )dropna)r%   r%   r%   Int64countindexr   namer%   r,   )	r   r   r/   value_countsr   r8   r   r   r   )r   r   r   s      r   test_value_counts_nar;   \   s    
(D%'y
9
9
9CU++Fy#W7KKKH>39,,,,68,,,T**Fy!Qs3B3xwWMMMH>39,,,,68,,,,,r#   c                     t          j        ddt           j        gd          } |                     d          }t          j        ddg| d d         dd	
          dz  }|j        j        dk    sJ t          j        ||           d S )NTFr   r   )	normalizer%   r,   Float64
proportionr7      )r   r   r/   r:   r8   r   r   r   )r!   r   r   s      r    test_value_counts_with_normalizerA   i   s    
)T5"%(	
:
:
:C--Fy!Qs3B3xy|TTTWXXH>9,,,,68,,,,,r#   c                  |   t          j        g dd          } t           j        j                            | d          }t          j        g dd          }t          j        ||           t          j        |           }|                                }t          j        |          }t          j        ||           d S )N)	TTFFTNTNFr   r   r%   )	NFTFTNNNN)	r   r   core
algorithmsdiffr   r   r   r   )r   r   r   r!   s       r   	test_diffrF   q   s    
AAA	 	 	A W$$Q**Fx@@@	  H #FH555
)A,,CXXZZFy""H68,,,,,r#   )numpyr   r   pandasr   pandas._testing_testingr   markparametrizer-   
logical_orlogical_andlogical_xorr   logical_notr"   r)   r2   r;   rA   rF    r#   r   <module>rR      s4                  bfbmR^R^D ( ( (V 2>"233- - 43-3 3 3 T5MD$<#@AA* * BA*
- 
- 
-- - -- - - - -r#   