o
    ŀg[                     @   sZ   d dl Zd dlZd dlmZ d dlmZ d dlmZ d dl	Z
d dlmZ G dd dZdS )    N)'construct_1d_object_array_from_listlike)is_extension_array_dtype)ExtensionDtypec                   @   s   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd ZdS )BaseInterfaceTestsz,Tests that the basic interface is satisfied.c                 C   s   t |dksJ d S Nd   )lenselfdata r   Y/var/www/html/myenv/lib/python3.10/site-packages/pandas/tests/extension/base/interface.pytest_len   s   zBaseInterfaceTests.test_lenc                 C      |j dksJ d S r   )sizer	   r   r   r   	test_size      zBaseInterfaceTests.test_sizec                 C   r   )N   )ndimr	   r   r   r   	test_ndim   r   zBaseInterfaceTests.test_ndimc                 C   s   |j du sJ d S )NT)_can_hold_nar	   r   r   r   test_can_hold_na_valid   s   z)BaseInterfaceTests.test_can_hold_na_validc                 C   s   |j j}||   }|d |v sJ |d |v sJ ||v s!J ||vs'J tjD ]}||u s8t|t|kr9q*||vs?J ||vsEJ q*d S )Nr   )dtypena_valueisnatmNULL_OBJECTStype)r
   r   data_missingr   na_value_objr   r   r   test_contains    s   
z BaseInterfaceTests.test_containsc                 C   s(   t |}|jdd}||jksJ d S )NF)index)pdSeriesmemory_usagenbytes)r
   r   sresultr   r   r   test_memory_usage:   s   
z$BaseInterfaceTests.test_memory_usagec                 C   sd   t |}|d |d ksJ t j|td}t jt|td}|jdkr*tt|}t|| d S )Nr   r   r   )nparrayobjectlistr   r   r   assert_numpy_array_equalr
   r   r'   expectedr   r   r   test_array_interface?   s   

z'BaseInterfaceTests.test_array_interfacec                 C   s@   t |sJ t |jsJ t t|sJ t|jtsJ d S N)r   r   r"   r#   
isinstancer   r	   r   r   r   test_is_extension_array_dtypeJ   s   z0BaseInterfaceTests.test_is_extension_array_dtypec                 C   s    t |drJ t |drJ d S )Nvalues_values)hasattrr	   r   r   r   test_no_values_attributeP   s   z+BaseInterfaceTests.test_no_values_attributec                 C   s8   t |}t|jdr|jjd j|jju sJ d S d S )Nblocksr   )r"   r#   r7   _mgrr9   
is_numericr   _is_numericr
   r   r'   r   r   r   test_is_numeric_honoredV   s
   
z*BaseInterfaceTests.test_is_numeric_honoredc                 C   sX   |  }t|r(|dsJ | sJ |drJ | r"J |jjs*J d S d S )Nanyall)r   r   _reducer?   r@   r   _is_boolean)r
   r   nar   r   r   test_isna_extension_array[   s   z,BaseInterfaceTests.test_isna_extension_arrayc                 C   s\   |d |d ks
J |  }|jjrtd|j d |d |d< |d |d ks,J d S )Nr   r   z!test_copy assumes mutability and  is immutable)copyr   _is_immutablepytestskipr=   r   r   r   	test_copyh   s   zBaseInterfaceTests.test_copyc                 C   s   |d |d ks
J |  }||usJ t|t|ksJ |jjr,td|j d |d |d< |d |d ks<J |j d d d S )Nr   r   z!test_view assumes mutability and rE   r)   )viewr   r   rG   rH   rI   r=   r   r   r   	test_views   s   zBaseInterfaceTests.test_viewc                 C   s.   |  }t|}t|tsJ ||ksJ d S r2   )tolistr-   r3   r/   r   r   r   test_tolist   s   zBaseInterfaceTests.test_tolistN)__name__
__module____qualname____doc__r   r   r   r   r    r(   r1   r4   r8   r>   rD   rJ   rL   rN   r   r   r   r   r      s     r   )numpyr*   rH   pandas.core.dtypes.castr   pandas.core.dtypes.commonr   pandas.core.dtypes.dtypesr   pandasr"   pandas._testing_testingr   r   r   r   r   r   <module>   s    