o
    ŀg                     @   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   ^/var/www/html/myenv/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   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   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varddofstdkurtosisskew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	   )r7   r   r8   r9   r:   r;   float64r   )r
   r   r   r   getattr)pandasmethnamekwargsr   
pandasmethr   s2r   r   r   r   test_stat_methodM   s   "

rC   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 )Nr7   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   r   r   r   test_value_counts_nad   s   rO   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   rE   rF   rG   )r
   r   rL   rM   r	   r   r   )serr   rN   r   r   r   r   test_value_counts_emptyq   s   rQ   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 )Nr7   r   r   r   T)	normalizer   r   
proportionrG   r#   )r
   r   rK   rL   rH   r	   r   r   )rP   r   r   r   r   r    test_value_counts_with_normalizez   s
   "rT   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)   rU   r         @)r
   r   r,   rK   r)   rU   r	   r%   r   r   r   r   test_floating_array_sum   
   r[   zvalues, expected)r   r   r#   rY   rW           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   
r^   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   brc   )r   Nr#   )r7   Nr   r   r   )ABCrd   r   r   r7   r#   )re   rf   r   rc   )rI   )rH   )r
   	DataFramer   r>   rf   
isinstancer   r=   groupbyrM   r   assert_frame_equal)r_   dfr   r   r   r   r   test_preserve_dtypes   s   	 rl   methodr`   ra   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)r]   r   Nr   r(   r`   r   r   )r
   r   r>   rK   )r)   rm   r	   r%   funcr   r   r   r   test_floating_array_min_max   s   

ro   	   c                 C   rV   )N)r   g       @Nr   rX   r   r   )r
   r   rb   rK   rZ   r   r   r   test_floating_array_prod   r\   rq   )$numpyr   pytestpandas.compatr   pandasr
   pandas._testing_testingr   markparametrizeabssignfilterwarningsr   logexpsincossqrtr"   r*   subtractr&   r/   skipifrC   rO   rQ   rT   r[   r^   rl   ro   rq   r   r   r   r   <module>   sZ    
 







		 


