U
|
ý°dq ã @ s: d Z ddlZddlmZmZ ddlmZ G dd dZ dS )z@
|
Tests for np.foo applied to DataFrame, not necessarily ufuncs.
|
é N)Ú CategoricalÚ DataFramec @ s e Zd Zdd Zdd ZdS )Ú TestAsArrayc C sR t tddgtddgd}t |¡}tjddgddggdd}t ||¡ d S )Né é )ÚAÚBÚobject)Zdtype)r r ÚnpZasarrayÚarrayÚtmZassert_numpy_array_equal)ÚselfZdfÚresultÚexpected© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/frame/test_npfuncs.pyÚtest_asarray_homogenous s
|
z#TestAsArray.test_asarray_homogenousc C sl t jdd t |¡}W 5 Q R X t|t|s4t|j|jksDt|j|jksTtt ||
|
t j¡¡ d S )NÚignore)Úall) r
|
ZerrstateÚsqrtÚ
|
isinstanceÚtypeÚAssertionErrorÚindexÚcolumnsr Zassert_frame_equalÚapply)r Z float_framer r r r Ú test_np_sqrt s zTestAsArray.test_np_sqrtN)Ú__name__Ú
|
__module__Ú __qualname__r r r r r r r s r )
|
Ú__doc__Únumpyr
|
Zpandasr r Zpandas._testingZ_testingr r r r r r Ú<module> s
|
