
    Pi                     |   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          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          d'             Z"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                              }|                    t
                    }t           j        ||                                <   t          j        |d          }t          j        ||           t          j	        |          } | |          }t          j	        |          }t          j
        ||           d S )N      Float64dtype)pdarrayNAastypefloatobjectisnatmassert_extension_array_equalSeriesassert_series_equalr   using_nan_is_naaresultnp_resexpectedss          ~/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/pandas/tests/arrays/floating/test_function.pytest_ufuncs_singler   
   s     	!QBE")444AU1XXFU188E??##F]]6""FuF16688xi000H#FH555
	!AU1XXFy""H68,,,,,    c                 z   t          j        dddt           j        gd          }t          j        d          5   | |          } | |                    t                              }|                    t                    }t           j        ||                                <   t          j        |d          }d d d            n# 1 swxY w Y   t          j
        ||           t          j        |          }t          j        d          5   | |          } | |                    t                              }|                    t                    }t           j        ||                                <   t          j        |d          }d d d            n# 1 swxY w Y   t          j        ||           d S )N      ?皙?      @r	   r
   ignore)invalid)r   r   r   nperrstater   r   r   r   r   r   r   r   r   s          r   test_ufuncs_single_floatr)      s   
#sC'y999A	X	&	&	& 5 5qqxx''v&&5qvvxx8F)4445 5 5 5 5 5 5 5 5 5 5 5 5 5 5 #FH555
	!A	X	&	&	& 6 6qqxx''v&&5qvvxx9V95556 6 6 6 6 6 6 6 6 6 6 6 6 6 6 68,,,,,s%   B CC
C
B FF"Fc                    t          j        dddt           j        gd          } | ||          } | |                    t                    |                    t                              }|                    t
                    }t           j        ||                                <   t          j        |d          }t          j        ||           t          j        g d          } | ||          } | |                    t                    |          }|                    t
                    }t           j        ||                                <   t          j        |d          }t          j        ||            | ||          } | ||                    t                              }|                    t
                    }t           j        ||                                <   t          j        |d          }t          j        ||            | |d          } | |                    t                    d          }|                    t
                    }t           j        ||                                <   t          j        |d          }t          j        ||            | d|          } | d|                    t                              }|                    t
                    }t           j        ||                                <   t          j        |d          }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   arrs          r   test_ufuncs_binary_floatr.   1   s    	!S"be$I666AU1a[[FU188E??AHHUOO44F]]6""FuF16688xi000H#FH555 (<<<
 
 CU1c]]FU188E??C((F]]6""FuF16688xi000H#FH555U3]]FU3((F]]6""FuF16688xi000H#FH555 U1a[[FU188E??A&&F]]6""FuF16688xi000H#FH555U1a[[FU1ahhuoo&&F]]6""FuF16688xi000H#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_raisesr8   \   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   )rA   r#   rB   rC   rD   rE   float64 )r   r   r   getattr)pandasmethnamekwargsr   
pandasmethr   s2r   s          r   test_stat_methodrN   e   s     		S#sCbeRUC9UUUA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 )NrA   r#   r	   r
   F)dropna)r   r   r   Int64countindexr   nameTr   r   )	r   r   r   value_countsIndexr   r   r   r   )r-   r   idxr   s       r   test_value_counts_narZ   |   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
   rQ   rR   rS   )r   r   rW   rX   r   r   r   )serr   rY   r   s       r   test_value_counts_emptyr]      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 )NrA   r#   r	   r
   T)	normalizer   r   
proportionrS   r+   )r   r   r   rW   rT   r   r   r   )r\   r   r   s      r    test_value_counts_with_normalizera      s    
)S#sBE*)
<
<
<C--Fy!Qs2A2wilSSSVWWH>39,,,,68,,,,,r    	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
   r2   rb   r         @)r   r   r5   r   r2   rb   r   r-   r   s        r   test_floating_array_sumrh      sc    
(???%
0
0
0CWWFiW88F )q..}}}}}}r    zvalues, expected)r   r   r+   rf   rd           c                 j    t          j        | d          }t          j        |          }||k    sJ d S )Nr	   r
   )r   r   r'   r5   )r/   r   r-   r   s       r   test_floating_array_numpy_sumrk      s<     (6
+
+
+CVC[[FXr    op)r5   minmaxprodc                 H   t          j        g dg dt          j        g dd          d          } t          |j        |                       }|rt          |          t          k    sJ n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   brq   )r   Nr+   )rA   Nr$   r	   r
   )ABCrr   r"   r$   rA   r+   )rs   rt   r   rq   )rU   )rT   )r   	DataFramer   rI   rt   typer   
isinstancer'   rG   groupbyrX   r   assert_frame_equal)rl   using_python_scalarsdfr   r   s        r   test_preserve_dtypesr|      s/   	 ***)<<<	
 	

 
B WRT2  F .F||u$$$$$&"*----- *WRZZ__b))++F|hSz""3())L)L)LMMhSz,,,  H &(+++++r    methodrm   rn   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)ri   r"   Nr
   r1   rm   r   r   )r   r   rI   r   )r2   r}   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
   re   r   r   )r   r   ro   r   rg   s        r   test_floating_array_prodr      sf    
(###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)   r3   subtractr.   r8   skipifrN   rZ   r]   ra   rh   rk   r|   r   r   rH   r    r   <module>r      sM                        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'6T QFQI#677* * 87* H%FGG					R		    HG
- 
- 
-- - -- - - q!f--  .- )))S)OOOS+ATFC=Q    <<<==, , >=,4 E5>22  32 q!f--  .-  r    