o
    dU                     @   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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   d/var/www/html/visualizacion-main/env/lib/python3.10/site-packages/pandas/tests/frame/test_npfuncs.pytest_asarray_homogenous   s   
z#TestAsArray.test_asarray_homogenousc                 C   s   t jdd t |}W d    n1 sw   Y  t|t|s$J |j|ju s,J |j|ju s4J t||	t j d S )Nignore)all)
r   errstatesqrt
isinstancetypeindexcolumnsr   assert_frame_equalapply)r   float_framer   r   r   r   test_np_sqrt   s   zTestAsArray.test_np_sqrtN)__name__
__module____qualname__r   r"   r   r   r   r   r      s    r   )
__doc__numpyr   pandasr   r   pandas._testing_testingr   r   r   r   r   r   <module>   s
    