a
    é´«j¿
  ã                   @   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   ú^/var/www/html/laksyashrimp/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 sˆ0    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Äª¼?ç°ÒrÚ2dÜ?ç�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   