o
    Û­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)Údtype)r   r   ÚnpÚasarrayÚarrayÚtmÚassert_numpy_array_equal)ÚselfÚdfÚresultÚexpected© r   ú\/var/www/html/CropPilot/venv/lib/python3.10/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w   Y  t|t|ƒƒs$J ‚|j |j¡s-J ‚|j |j¡s6J ‚t 	|| 
t j¡¡ d S )NÚignore)Úall)r   ÚerrstateÚsqrtÚ
isinstanceÚtypeÚindexÚis_Úcolumnsr   Úassert_frame_equalÚapply)r   Ú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 s'w   Y  tjt|d�� |jd d�}W d   ƒ n1 sDw   Y  t 	||¡ d S )Nr   )é   é   z:The behavior of DataFrame.sum with axis=None is deprecatedF)ÚmatchÚcheck_stacklevel)r&   )Úaxis)
r   ÚrandomÚdefault_rngÚstandard_normalr   r   Úassert_produces_warningÚFutureWarningÚsumÚ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   s    z-TestAsArray.test_np_ravel.<locals>.<listcomp>)r4   r5   r6   r7   r8   r9   r   r   r%   ©Úx1Úx2Úx3©r   c                 S   s$   g | ]}t | d d¡g d¢d�‘qS )r   r%   r@   rD   r:   r<   r   r   r   r?   J   s    ÿÿ)r   r   Úravelr   r   r   r;   )r   r0   r   r   r   r   r   Útest_np_ravel/   s,   þÿÿ
"þÿÿ
zTestAsArray.test_np_ravelN)Ú__name__Ú
__module__Ú__qualname__r   r#   r3   rF   r   r   r   r   r      s
    	r   )
Ú__doc__Únumpyr   Úpandasr   r   Úpandas._testingÚ_testingr   r   r   r   r   r   Ú<module>   s
    