o
    vho                     @   sn   d dl Z 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
mZ d dlZd dlmZ G dd dZdS )    N)np_version_gt2)'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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   c/var/www/html/hyperkenya/venv/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_interfaceC   s   

z'BaseInterfaceTests.test_array_interfacec           	      C   sD  t j|dd}t j|dd}t ||rJ tsd S d}d}tjdd-}td t j|dd}t|dks9J t|rJd}|t|d j	v sJJ W d    n1 sTw   Y  tjdd-}td t j|dd}t|dkstJ t|rd}|t|d j	v sJ W d    n1 sw   Y  |st ||sJ d S d S )	NT)copyFzGStarting with NumPy 2.0, the behavior of the 'copy' keyword has changed)recordalwaysr   r   )
r+   r,   may_share_memoryr   warningscatch_warningssimplefilterr	   strmessage)	r   r   result_copy1result_copy2warning_raisedmsgwresult_nocopy1result_nocopy2r   r   r   test_array_interface_copyN   s:   

z,BaseInterfaceTests.test_array_interface_copyc                 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_dtypem   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_attributes   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$   rI   _mgrrK   
is_numericr   _is_numericr   r   r(   r   r   r   test_is_numeric_honoredy   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   _reducerQ   rR   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)r3   r   _is_immutablepytestskiprO   r   r   r   	test_copy   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 rW   r*   )viewr   r   rX   rY   rZ   rO   r   r   r   	test_view   s   zBaseInterfaceTests.test_viewc                 C   s.   |  }t|}t|tsJ ||ksJ d S rD   )tolistr.   rE   r0   r   r   r   test_tolist   s   zBaseInterfaceTests.test_tolistN)__name__
__module____qualname____doc__r   r   r   r   r!   r)   r2   rC   rF   rJ   rP   rV   r[   r]   r_   r   r   r   r   r      s"    r   )r7   numpyr+   rY   pandas.compat.numpyr   pandas.core.dtypes.castr   pandas.core.dtypes.commonr   pandas.core.dtypes.dtypesr   pandasr#   pandas._testing_testingr   r   r   r   r   r   <module>   s    