
    bMh                        d dl Zd dlZd dlZd dlmZ d dlm	Z	 ej
                            dej        ej        g          ej
                            d          d                         Zej
                            dej        ej        ej        ej        ej        g          d             Zej
                            dej        ej        g          d             Zd Zej
                            d	d d
gd dgg          d             Zej
                            dddd ifddd
ifddd ifddd
ifdi fdi fdi fg          d             Zd Zd Zd Zej
                            dddg          ej
                            dd dg          d                         Zej
                            dddg          ej
                            dddg          d                          Z ej
                            dddg          ej
                            dd d!g          d"                         Z!ej
                            d#g d$d%fg d&d%fdgd fg          d'             Z"ej
                            d(g d)          d*             Z#dS )+    N)FloatingArrayufuncz7ignore:invalid value encountered in sign:RuntimeWarningc           	         t          j        dddt          j        g          } | |          }t          j         | |                    t
                              d          }t          j        ||           t          j        |          } | |          }t          j        t          j         | |                    t
                              d                    }t          j	        ||           d S )N      Int64dtype)
pdarraynpnanastypefloattmassert_extension_array_equalSeriesassert_series_equalr   aresultexpectedss        i/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/arrays/integer/test_function.pytest_ufuncs_single_intr   	   s     	!QBF#$$AU1XXFxahhuoo..g>>>H#FH555
	!AU1XXFy%%"8"8HHHIIH68,,,,,    c                 <   t          j        dddt          j        g          }t          j        d          5   | |          }t           | |                    t                              |j                  }d d d            n# 1 swxY w Y   t          j
        ||           t          j        |          }t          j        d          5   | |          }d d d            n# 1 swxY w Y   t          j        |          }t          j        ||           d S )Nr   r   r   ignore)invalid)mask)r   r   r   r   errstater   r   r   _maskr   r   r   r   r   s        r   test_ufuncs_single_floatr$      s   
!QBF#$$A	X	&	&	& G Gq qxx!7!7agFFFG G G G G G G G G G G G G G G #FH555
	!A	X	&	&	&  q              y""H68,,,,,s$   ABBBC((C,/C,c                     t          j        dddt          j        g          } | ||          }t          j         | |                    t
                    |                    t
                              d          }t          j        ||           t          j        g d          } | ||          }t          j         | |                    t
                    |          d          }t          j        ||            | ||          }t          j         | ||                    t
                              d          }t          j        ||            | |d          }t          j         | |                    t
                    d          d          }t          j        ||            | d|          }t          j         | d|                    t
                              d          }t          j        ||           d S )Nr   r   r   r	   r
   )r   r         )r   r   r   r   r   r   r   r   )r   r   r   r   arrs        r   test_ufuncs_binary_intr)   '   s    	!QBF#$$AU1a[[Fxahhuooqxx??wOOOH#FH555 (<<<
 
 CU1c]]Fxahhuoos337CCCH#FH555U3]]Fxc188E??337CCCH#FH555 U1a[[Fxahhuooq11AAAH#FH555U1a[[Fxa%11AAAH#FH55555r   c                     t          j        ddt          j        g          } t          j        |           }t          j        |                     t          j        d                    }t          j        |d                   t          j        |d                   f}t          |t                    sJ t          |          dk    sJ t          ||          D ]\  }}t          j        ||           d S )Nr   r   r   )na_valuer   r   )r   r   r   r   modfto_numpy
isinstancetuplelenzipr   r   )r   r   r   xys        r   test_ufunc_binary_outputr4   C   s    
!Q  AWQZZFwqzz26zAABBH!%%rx'<'<=Hfe$$$$$v;;!FH%% . .1
'1----. .r   valuesr   c                     t          j        |           }t          j                            |          }|                    d          }t          j        ||           d S )NFskipna)r   r   r   addreducesumr   assert_almost_equal)r5   r(   resr   s       r   test_ufunc_reduce_raisesr>   P   sP    
(6

C
&--

Cwwew$$H3)))))r   zpandasmethname, kwargsvarddofstdkurtosisskewsemc           
      
   t          j        ddddddt          j        t          j        gd          }t	          ||           } |d
i |}t          j        g d	d          }t	          ||           } |d
i |}||k    sJ d S )Nr   r   r&   r'         r	   )datar   )r   r   r&   r'   rF   rG    )r   r   r   r   getattr)pandasmethnamekwargsr   
pandasmethr   s2r   s          r   test_stat_methodrO   Y   s     		1aAq"&"&9IIIAN++JZ!!&!!F	***'	:	:	:B^,,Jz##F##Hvr   c                      t          j        dddt           j        gd          } |                     d          }t          j        ddt           j        gd          }|j        dk    sJ t          j        g d|dd	          }t          j        ||           |                     d
          }t          j        ddg| d d         dd	          }|j	        j        | j        k    sJ t          j        ||           d S )Nr   r   r	   r
   F)dropna)r   r   r   countindexr   nameT)
r   r   NAvalue_countsIndexr   r   r   r   rT   )r(   r   ex_indexr   s       r   test_value_counts_narZ   o   s    
(Aq!RU#7
3
3
3CU++FxAruW555H>W$$$$y('PPPH68,,,T**Fy!Qs2A2wgGLLLH>39,,,,68,,,,,r   c                     t          j        g d          } |                                 }t          j        g | j                  }|j        | j        k    sJ t          j        g |dd          }t          j        ||           d S )Nr	   r
   rR   rS   )r   r   rW   rX   r   r   r   )serr   idxr   s       r   test_value_counts_emptyr^   }   s    
)Bg
&
&
&CF
(2SY
'
'
'C9	!!!!y3gGDDDH68,,,,,r   c                     t          j        dddt           j        gd          } |                     d          }t          j        ddg| d d         dd	          d
z  }|j        j        | j        k    sJ t          j        ||           d S )Nr   r   r	   r
   T)	normalizeFloat64
proportionrS   r&   )r   r   rV   rW   rT   r   r   r   )r\   r   r   s      r    test_value_counts_with_normalizerc      s    
)Q1be$G
4
4
4C--Fy!Qs2A2wilSSSVWWH>39,,,,68,,,,,r   r8   TF	min_countr'   c                     |}t          j        g d|          }|                    | |          }| r|dk    r
|dk    sJ d S |t           j        u sJ d S )Nr   r   r&   Nr
   r8   rd   r   rG   )r   r   r;   rV   r8   rd   any_int_ea_dtyper   r(   r   s         r   test_integer_array_sumrj      sj     E
(???%
0
0
0CWWFiW88F )q..{{{{{{r   methodminmaxc                     |}t          j        g d|          }t          ||          } ||           }| r||dk    rdndk    sJ d S |t           j        u sJ d S )N)r   r   Nr
   r7   rl   r   r   )r   r   rJ   rV   )r8   rk   ri   r   r(   funcr   s          r   test_integer_array_min_maxrp      s     E
(<<<u
-
-
-C3DT   F v!!A666666r   	   c                     |}t          j        g d|          }|                    | |          }| r|dk    r
|dk    sJ d S |t           j        u sJ d S )N)r   r   Nr
   rg   r   r   )r   r   prodrV   rh   s         r   test_integer_array_prodrt      sj     E
(<<<u
-
-
-CXXVyX99F )q..{{{{{{r   zvalues, expected)r   r   r&   rG   rf   c                 j    t          j        | d          }t          j        |          }||k    sJ d S )Nr	   r
   )r   r   r   r;   )r5   r   r(   r   s       r   test_integer_array_numpy_sumrv      s<     (6
)
)
)CVC[[FXr   op)r;   rs   rl   rm   c                     t          j        dt          j        ddgd          i          }|                                }t	          |d         t
          j                  sJ d S )Nr   r   r   r	   r
   )r   	DataFramer   rm   r.   r   int64)rw   dfr   s      r   test_dataframe_reductionsr|      sZ     
sBHaV7;;;<	=	=BVVXXFfSk28,,,,,,,r   )$numpyr   pytestpandasr   pandas._testing_testingr   pandas.core.arraysr   markparametrizeabssignfilterwarningsr   logexpsincossqrtr$   r9   subtractr)   r4   r>   rO   rZ   r^   rc   rj   rp   rt   rv   r|   rI   r   r   <module>r      s                  , , , , , , 2627"344UVV	- 	- WV 54	- 2626262627"KLL- - ML- 262;"7886 6 9866
. 
. 
. QFQI#677* * 87* 					R		   - - -- - -- - - D%=11q!f--  .- 21 D%=11E5>22  32 21 D%=11q!f--  .- 21 )))Q///1)={K    <<<==- - >=- - -r   