o
    d:                     @   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defddZdS )CustomBlockreturnc                 C   s   dS )NF )selfr   r   ^/app/.heroku/python/lib/python3.10/site-packages/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   sr   t dg d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)         r   Zint64)Zdtyper   r   )Z	placementndimb)pd	DataFrame_mgrblocksr   Zaranger   slicer   r   Indexindex)Zdf1r   valuesbpZcustom_blockZblock_managerr   r   r	   df   s   

r    c                 C   s>   t dg di}t j| |gdd}t|jjd tsJ d S )Nc)g?g?g333333?r   )Zaxis)r   r   concat
isinstancer   r   r   )r    Zdf2resr   r   r	   test_concat_axis1#   s   r%   )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    	
