U
    sVc;                     @   s~   d dl Zd dlZd dlmZ d dlm  mZ d dl	Z
d dlmZ d dlmZ ejZG dd deZejdd Zd	d
 ZdS )    N)BlockPlacement)BlockManager)ExtensionBlockc                   @   s$   e Zd ZejZeedddZdS )CustomBlock)returnc                 C   s   dS )NF )selfr   r   N/tmp/pip-unpacked-wheel-xj8nt62q/pandas/tests/extension/test_external_block.py_can_hold_na   s    zCustomBlock._can_hold_naN)	__name__
__module____qualname__npZndarrayZ_holderpropertyboolr
   r   r   r   r	   r      s   r   c                  C   st   t ddddgi} | jj}tjddd}ttdd}t||dd}||f }t	|t 
ddg| jg}t |S )	Na         Zint64)Zdtype)Z	placementndimb)pd	DataFrame_mgrblocksr   Zaranger   slicer   r   ZIndexindex)Zdf1r   valuesZbpZcustom_blockZblock_managerr   r   r	   df   s    
r   c                 C   s@   t ddddgi}t j| |gdd}t|jjd ts<td S )Ncg?g?g333333?r   )Zaxis)r   r   concat
isinstancer   r   r   AssertionError)r   Zdf2resr   r   r	   test_concat_axis1$   s    r$   )Znumpyr   ZpytestZpandas._libs.internalsr   Zpandas.util._test_decoratorsutilZ_test_decoratorstdZpandasr   Zpandas.core.internalsr   Zpandas.core.internals.blocksr   Zskip_array_manager_invalid_testZ
pytestmarkr   Zfixturer   r$   r   r   r   r	   <module>   s   

