U
|
ý°da ã @ s~ d dl Zd dlZd dlmZ d dlm mZ d dl Z
|
d dl m Z d dl mZ ejZG dd deZejdd Zd d
|
ZdS ) é N)ÚBlockPlacement)Ú BlockManager)ÚExtensionBlockc @ s$ e Zd ZejZeedddZdS )Ú CustomBlock)Úreturnc C s dS )NF© )Úselfr r úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tests/extension/test_external_block.pyÚ _can_hold_na s zCustomBlock._can_hold_naN) Ú__name__Ú
|
__module__Ú __qualname__ÚnpZndarrayZ_holderÚpropertyÚboolr
|
r r r r r s r c C st t ddddgi¡} | jj}tjddd}ttdd}t||dd}||f }t |t
|
ddg¡| j g}t |¡S ) NÚaé é é Úint64)Zdtype)Z placementÚndimÚb) ÚpdÚ DataFrameÚ_mgrÚblocksr Zaranger Úslicer r ZIndexÚindex)Zdf1r ÚvaluesZbpZ custom_blockZ block_managerr r r Údf s
|
r c C s@ t ddddgi¡}t j| |gdd}t|jjd ts<td S )NÚcg¹?gÉ?g333333Ó?r )Zaxis)r r ÚconcatÚ
|
isinstancer r r ÚAssertionError)r Zdf2Úresr r r Útest_concat_axis1# s r% )Únumpyr ZpytestZpandas._libs.internalsr Z pandas.util._test_decoratorsÚutilZ_test_decoratorsÚtdZpandasr Zpandas.core.internalsr Z pandas.core.internals.blocksr Z skip_array_manager_invalid_testZ
|
pytestmarkr Zfixturer r% r r r r Ú<module> s
|
|
