U
    sVcU                     @   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)Zdtype)r   r   npZasarrayarraytmZassert_numpy_array_equal)selfZdfresultexpected r   C/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/frame/test_npfuncs.pytest_asarray_homogenous   s    
z#TestAsArray.test_asarray_homogenousc              	   C   sl   t jdd t |}W 5 Q R X t|t|s4t|j|jksDt|j|jksTtt	||
t j d S )Nignore)all)r
   Zerrstatesqrt
isinstancetypeAssertionErrorindexcolumnsr   Zassert_frame_equalapply)r   Z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__Znumpyr
   Zpandasr   r   Zpandas._testingZ_testingr   r   r   r   r   r   <module>   s   