
    bMh                     N   d dl Zd dlZd dlmZ d dl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ej
                            dd d	gd dgg          d
             Zej
                            e d          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g dd fg d!d fdgd"fg          d#             Z ej
                            d$g d%          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#dS )-    N)IS64ufuncz7ignore:invalid value encountered in sign:RuntimeWarningc                    t          j        dddt          j        gd          } | |          }t          j         | |                    t
                              d          }t          j        ||           t          j        |          } | |          }t          j        |          }t          j	        ||           d S )N      Float64dtype)
pdarraynpnanastypefloattmassert_extension_array_equalSeriesassert_series_equalr   aresultexpectedss        j/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pandas/tests/arrays/floating/test_function.pytest_ufuncs_singler   
   s     	!QBF#9555AU1XXFxahhuoo..i@@@H#FH555
	!AU1XXFy""H68,,,,,    c                    t          j        dddt          j        gd          }t          j        d          5   | |          }t          j         | |                    t                              d          }d d d            n# 1 swxY w Y   t          j        ||           t          j	        |          }t          j        d          5   | |          }t          j	         | |                    t                              d          }d d d            n# 1 swxY w Y   t          j
        ||           d S )N      ?皙?      @r	   r
   ignore)invalid)r   r   r   r   errstater   r   r   r   r   r   r   s        r   test_ufuncs_single_floatr%      s   
#sC(	:::A	X	&	&	& E Eq8EE!((5//22)DDDE E E E E E E E E E E E E E E #FH555
	!A	X	&	&	& F Fq9UU188E??339EEEF F F F F F F F F F F F F F F 68,,,,,s%   AB		BBAD!!D%(D%c                 $   t          j        dddt          j        gd          } | ||          }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_floatr*   (   s    	!S"bf%Y777AU1a[[Fxahhuooqxx??yQQQH#FH555 (<<<
 
 CU1c]]Fxahhuoos339EEEH#FH555U3]]Fxc188E??339EEEH#FH555 U1a[[Fxahhuooq11CCCH#FH555U1a[[Fxa%11CCCH#FH55555r   valuesr   c                     t          j        | d          }t          j                            |          }|                    d          }t          j        ||           d S )Nr	   r
   Fskipna)r   r   r   addreducesumr   assert_almost_equal)r+   r)   resr   s       r   test_ufunc_reduce_raisesr4   D   sU    
(6
+
+
+C
&--

Cwwew$$H3)))))r   zGH 36579: fail on 32-bit system)reasonz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 )N皙?r    333333?皙?      ?333333?r	   )datar   )r=   r    r>   r?   r@   rA   float64 )r   r   r   r   getattr)pandasmethnamekwargsr   
pandasmethr   s2r   s          r   test_stat_methodrJ   M   s     		S#sCbfbfEYWWWAN++JZ!!&!!F	666i	H	H	HB^,,Jz##F##Hvr   c                     t          j        dddt           j        gd          } |                     d          }t          j        ddt           j        g| j                  }|j        | j        k    sJ t          j        g d|dd	
          }t          j        ||           |                     d          }t          j        ddg|d d         dd	
          }t          j        ||           d S )Nr=   r    r	   r
   F)dropna)r   r   r   Int64countindexr   nameTr   r   )	r   r   NAvalue_countsIndexr   r   r   r   )r)   r   idxr   s       r   test_value_counts_narW   d   s    
(Cc25)
;
;
;CU++F
(Cbe$CI
6
6
6C9	!!!!y#W7KKKH68,,,T**Fy!Qs3B3xwWMMMH68,,,,,r   c                      t          j        g d          } |                                 }t          j        g d          }|j        dk    sJ t          j        g |dd          }t          j        ||           d S )Nr	   r
   rM   rN   rO   )r   r   rT   rU   r   r   r   )serr   rV   r   s       r   test_value_counts_emptyrZ   q   s    
)Bi
(
(
(CF
(2Y
'
'
'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)	normalizer   r   
proportionrO   r'   )r   r   rS   rT   rP   r   r   r   )rY   r   r   s      r    test_value_counts_with_normalizer^   z   s    
)S#sBE*)
<
<
<C--Fy!Qs2A2wilSSSVWWH>39,,,,68,,,,,r   r.   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
   r.   r_   r         @)r   r   r1   rS   r.   r_   r   r)   r   s        r   test_floating_array_sumre      se     (???%
0
0
0CWWFiW88F )q..}}}}}}r   zvalues, expected)r   r   r'   rc   ra           c                 j    t          j        | d          }t          j        |          }||k    sJ d S )Nr	   r
   )r   r   r   r1   )r+   r   r)   r   s       r   test_floating_array_numpy_sumrh      s<     (6
+
+
+CVC[[FXr   op)r1   minmaxprodc                    t          j        g dg dt          j        g dd          d          } t          |j        |                       }t          |t          j                  sJ  t          |                    d          |                       }t          j        t          j        dd	g          t          j        d
dgd          dt          j	        ddgd                    }t          j        ||           d S )N)r   brn   )r   Nr'   )r=   Nr!   r	   r
   )ABCro   r   r!   r=   r'   )rp   rq   r   rn   )rQ   )rP   )r   	DataFramer   rE   rq   
isinstancer   rC   groupbyrU   r   assert_frame_equal)ri   dfr   r   s       r   test_preserve_dtypesrw      s   	 ***)<<<	
 	

 
B WRT2  Ffbj))))) *WRZZ__b))++F|hSz""3())L)L)LMMhSz,,,  H &(+++++r   methodrj   rk   c                     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)rf   r   Nr
   r-   rj   r   r   )r   r   rE   rS   )r.   rx   r   r)   funcr   s         r   test_floating_array_min_maxr{      s~     (###5
1
1
1C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   g       @Nr
   rb   r   r   )r   r   rl   rS   rd   s        r   test_floating_array_prodr~      sh     (###5
1
1
1CXXVyX99F )q..{{{{{{r   )$numpyr   pytestpandas.compatr   pandasr   pandas._testing_testingr   markparametrizeabssignfilterwarningsr   logexpsincossqrtr%   r/   subtractr*   r4   skipifrJ   rW   rZ   r^   re   rh   rw   r{   r~   rD   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* H%FGG					R		    HG
- 
- 
-- - -- - - D%=11q!f--  .- 21 )))S)OOOS+ATFC=Q    <<<==, , >=,. D%=11E5>22  32 21 D%=11q!f--  .- 21  r   