o
    թZh                     @   s  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d Zej
dejejejejejgdd Zej
dejejgd	d
 Zej
dd dgd dggdd Zej
je d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d Zdd Zdd Zdd Zej
d d!d"gej
d#d d$gd%d& Zej
d'g d(d)fg d*d)fdgd+fgd,d- Z ej
d.g d/d0d1 Z!ej
d d!d"gej
d2d3d4gd5d6 Z"ej
d d!d"gej
d#d d7gd8d9 Z#dS ):    N)IS64ufuncz7ignore:invalid value encountered in sign:RuntimeWarningc                 C   sp   t jdddtjgdd}| |}t j| |tdd}t|| t |}| |}t |}t	|| d S )N      Float64dtype)
pdarraynpnanastypefloattmassert_extension_array_equalSeriesassert_series_equalr   aresultexpecteds r   a/var/www/html/lang_env/lib/python3.10/site-packages/pandas/tests/arrays/floating/test_function.pytest_ufuncs_single
   s   

r   c                 C   s   t jdddtjgdd}tjdd | |}t j| |tdd}W d    n1 s-w   Y  t|| t 	|}tjdd | |}t j	| |tdd}W d    n1 s^w   Y  t
|| d S )N      ?皙?      @r   r   ignore)invalid)r
   r   r   r   Zerrstater   r   r   r   r   r   r   r   r   r   test_ufuncs_single_float   s   
r!   c                 C   s   t jdddtjgdd}| ||}t j| |t|tdd}t|| tg d}| ||}t j| |t|dd}t|| | ||}t j| ||tdd}t|| | |d}t j| |tddd}t|| | d|}t j| d|tdd}t|| d S )Nr   r   r   r   r   )r   r         )r
   r   r   r   r   r   r   r   )r   r   r   r   arrr   r   r   test_ufuncs_binary_float(   s"   
 



r%   valuesr   c                 C   s6   t j| dd}tj|}|jdd}t|| d S )Nr   r   Fskipna)r
   r   r   addreducesumr   Zassert_almost_equal)r&   r$   resr   r   r   r   test_ufunc_reduce_raisesD   s   r-   zGH 36579: fail on 32-bit system)reasonzpandasmethname, kwargsvarZddofZstdZkurtosisZskewZsemc              	   C   st   t jddddddtjtjgdd}t|| }|di |}t jg d	d
d}t|| }|di |}||ks8J d S )N皙?r   333333?皙?      ?333333?r   )datar	   )r0   r   r1   r2   r3   r4   float64r   )r
   r   r   r   getattr)Zpandasmethnamekwargsr   Z
pandasmethr   s2r   r   r   r   test_stat_methodM   s   "

r:   c                  C   s   t jdddt jgdd} | jdd}t jddt jg| jd}|j| jks&J t jg d|dd	d
}t|| | jdd}t jddg|d d dd	d
}t|| d S )Nr0   r   r   r   F)Zdropna)r   r   r   Int64countindexr	   nameTr   r   )	r
   r   NAvalue_countsIndexr	   r   r   r   )r$   r   idxr   r   r   r   test_value_counts_nad   s   rE   c                  C   sT   t jg dd} |  }t jg dd}|jdksJ t jg |ddd}t|| d S )Nr   r   r;   r<   r=   )r
   r   rB   rC   r	   r   r   )serr   rD   r   r   r   r   test_value_counts_emptyq   s   rG   c                  C   sh   t jdddt jgdd} | jdd}t jddg| d d dd	d
d }|jj| jks,J t|| d S )Nr0   r   r   r   T)	normalizer   r   Z
proportionr=   r"   )r
   r   rA   rB   r>   r	   r   r   )rF   r   r   r   r   r    test_value_counts_with_normalizez   s
   "rI   r(   TF	min_countr#   c                 C   N   t jg d|d}|j| |d}| r|dkr|dksJ d S |t ju s%J d S )Nr   r   r"   Nr   r(   rJ   r         @)r
   r   r+   rA   r(   rJ   r	   r$   r   r   r   r   test_floating_array_sum   
   rP   zvalues, expected)r   r   r"   rN   rL           c                 C   s(   t j| dd}t|}||ksJ d S )Nr   r   )r
   r   r   r+   )r&   r   r$   r   r   r   r   test_floating_array_numpy_sum   s   
rS   op)r+   minmaxprodc                 C   s   t g dg dt jg dddd}t|j|  }t|tjs#J t|d|  }t jtdd	gt jd
dgdddt j	ddgddd}t
|| d S )N)r   brX   )r   Nr"   )r0   Nr   r   r   )ABCrY   r   r   r0   r"   )rZ   r[   r   rX   )r?   )r>   )r
   Z	DataFramer   r7   r[   
isinstancer   r6   groupbyrC   r   Zassert_frame_equal)rT   dfr   r   r   r   r   test_preserve_dtypes   s   	 r_   methodrU   rV   c                 C   sX   t jg d|d}t||}|| d}| r#||dkrdndks!J d S |t ju s*J d S )N)rR   r   Nr   r'   rU   r   r   )r
   r   r7   rA   )r(   r`   r	   r$   funcr   r   r   r   test_floating_array_min_max   s   

rb   	   c                 C   rK   )N)r   g       @Nr   rM   r   r   )r
   r   rW   rA   rO   r   r   r   test_floating_array_prod   rQ   rd   )$numpyr   ZpytestZpandas.compatr   Zpandasr
   Zpandas._testingZ_testingr   markZparametrizeabssignfilterwarningsr   logexpsincossqrtr!   r)   subtractr%   r-   Zskipifr:   rE   rG   rI   rP   rS   r_   rb   rd   r   r   r   r   <module>   sZ    
 







		 


