a
    Pf
                     @   s:   d Z ddlZddlmZmZ ddlmZ G dd dZ	dS )z@
Tests for np.foo applied to DataFrame, not necessarily ufuncs.
    N)Categorical	DataFramec                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestAsArrayc                 C   sR   t tddgtddgd}t|}tjddgddggdd}t|| d S )N      )ABobject)Zdtype)r   r   npZasarrayarraytmassert_numpy_array_equal)selfdfresultexpected r   X/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/frame/test_npfuncs.pytest_asarray_homogeneous   s    
z$TestAsArray.test_asarray_homogeneousc                 C   s   t jdd t |}W d    n1 s,0    Y  t|t|sHJ |j|jsZJ |j|jslJ t	||
t j d S )Nignore)all)r
   Zerrstatesqrt
isinstancetypeindexis_columnsr   Zassert_frame_equalapply)r   Zfloat_framer   r   r   r   test_np_sqrt   s    (zTestAsArray.test_np_sqrtc                 C   s   t jdd}t|}d}tjt|dd t |}W d    n1 sN0    Y  tjt|d |jd d}W d    n1 s0    Y  t	|| d S )Nr   )      z:The behavior of DataFrame.sum with axis=None is deprecatedF)matchZcheck_stacklevel)r!   )Zaxis)
r
   randomZdefault_rngZstandard_normalr   r   Zassert_produces_warningFutureWarningsumZassert_series_equal)r   arrr   msgresr   r   r   r   !test_sum_deprecated_axis_behavior   s    (*z-TestAsArray.test_sum_deprecated_axis_behaviorc                 C   s   t g dg dg}t dd |D }t g d}t|| t t|d ddg d	d
}t g d}t|| t dd |D }t g d}t|| d S )N)=w?r2d?B^)Ҫ?T2o灨$пc                 S   s   g | ]}t |d dqS )r   r    r   reshape.0batchr   r   r   
<listcomp>8       z-TestAsArray.test_np_ravel.<locals>.<listcomp>)r)   r*   r+   r,   r-   r.   r   r   r    x1Zx2Zx3r   c                 S   s$   g | ]}t |d dg ddqS )r   r    r6   r8   r/   r1   r   r   r   r4   J   s   )r
   r   Zravelr   r   r   r0   )r   r%   r   r   r   r   r   test_np_ravel/   s,    
"
zTestAsArray.test_np_ravelN)__name__
__module____qualname__r   r   r(   r9   r   r   r   r   r      s   	r   )
__doc__numpyr
   Zpandasr   r   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   