a
    j=ic4                     @   s   d Z 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 Zejdd Zejd	d
 Zejdd Zejdd Zejdd Zejdd Zejdd Zejdd Zejdd ZG dd dejZG dd dejZG dd dejZG dd  d ejZG d!d" d"ej Z!G d#d$ d$ej"Z#G d%d& d&ej$Z%G d'd( d(ej&Z'G d)d* d*ej(Z)G d+d, d,ej*Z+G d-d. d.ej,Z-G d/d0 d0ej.Z/G d1d2 d2ej0Z1G d3d4 d4ej2Z3G d5d6 d6ej4Z5G d7d8 d8ej6Z7G d9d: d:ej8Z9G d;d< d<ej:Z;G d=d> d>ej<Z=dS )?aj  
This file contains a minimal set of tests for compliance with the extension
array interface test suite, and should contain no other tests.
The test suite for the full functionality of the array is located in
`pandas/tests/arrays/`.

The tests in this file are inherited from the BaseExtensionTests, and only
minimal tweaks should be applied to get the tests passing (by overwriting a
parent method).

Additional tests should either be added to one of the BaseExtensionTests
classes (if they are relevant for the extension interface for all dtypes), or
be added to the array-specific tests in `pandas/tests/arrays/`.

    NBooleanDtype)basec                   C   s0   ddgd t jg ddgd  t jg ddg S )NTF   ,   )npnan r	   r	   t/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pandas/tests/extension/test_boolean.py	make_data   s    r   c                   C   s   t  S Nr   r	   r	   r	   r
   dtype   s    r   c                 C   s   t jt | dS )Nr   )pdarrayr   r   r	   r	   r
   data"   s    r   c                 C   s   t jtd| dS )Nd   r   )r   r   r   Zonesr   r	   r	   r
   data_for_twos'   s    r   c                 C   s   t jtjdg| dS )NTr   r   r   r   r   r   r	   r	   r
   data_missing,   s    r   c                 C   s   t jg d| dS )N)TTFr   )r   r   r   r	   r	   r
   data_for_sorting1   s    r   c                 C   s   t jdtjdg| dS NTFr   r   r   r	   r	   r
   data_missing_for_sorting6   s    r   c                   C   s   dd S )Nc                 S   s   | t ju o|t ju S r   r   NA)xyr	   r	   r
   <lambda>>       zna_cmp.<locals>.<lambda>r	   r	   r	   r	   r
   na_cmp;   s    r   c                   C   s   t jS r   r   r	   r	   r	   r
   na_valueA   s    r    c                 C   s*   d}d}t j}tj|||||||g| dS r   )r   r   r   r   )r   banar	   r	   r
   data_for_groupingF   s    r$   c                   @   s   e Zd ZdS )	TestDtypeN__name__
__module____qualname__r	   r	   r	   r
   r%   N   s   r%   c                   @   s   e Zd ZdS )TestInterfaceNr&   r	   r	   r	   r
   r*   R   s   r*   c                   @   s   e Zd ZdS )TestConstructorsNr&   r	   r	   r	   r
   r+   V   s   r+   c                   @   s   e Zd ZdS )TestGetitemNr&   r	   r	   r	   r
   r,   Z   s   r,   c                   @   s   e Zd ZdS )TestSetitemNr&   r	   r	   r	   r
   r-   ^   s   r-   c                   @   s   e Zd ZdS )	TestIndexNr&   r	   r	   r	   r
   r.   b   s   r.   c                   @   s   e Zd ZdS )TestMissingNr&   r	   r	   r	   r
   r/   f   s   r/   c                       sf   e Zd ZddhZd fdd	ZefddZejj	dd	 fd
dZ
ejj	dd	 fddZ  ZS )TestArithmeticOps__sub____rsub__Nc                    s4   d }| dddv rt}t j||||d d S )N_r)powtruedivfloordivexc)striplstripNotImplementedErrorsupercheck_opnameselfsop_nameotherr9   	__class__r	   r
   r>   m   s    zTestArithmeticOps.check_opnamec           	      C   s   |d u r|| j v rRd}tjt|d ||| W d    n1 sD0    Y  d S |||}| |||}|dv r~|d}n&|dv r| |t||}|d}|dkrtj||	 < | 
|| n4t| ||| W d    n1 s0    Y  d S )Nznumpy boolean subtractmatch)__floordiv____rfloordiv____pow____rpow____mod____rmod__ZInt8)__truediv____rtruediv__ZFloat64rK   )
implementspytestZraises	TypeErrorZ_combineastypefloatr   r   isnaZassert_equal)	r@   objoprC   rB   r9   msgresultexpectedr	   r	   r
   	_check_opu   s$    
(
	
zTestArithmeticOps._check_opzInconsistency between floordiv and divmod; we raise for floordiv but not for divmod. This matches what we do for non-masked bool dtype.)reasonc                    s   t  || d S r   )r=   test_divmod_series_array)r@   r   r   rD   r	   r
   r]      s    z*TestArithmeticOps.test_divmod_series_arrayc                    s   t  | d S r   )r=   test_divmod)r@   r   rD   r	   r
   r^      s    zTestArithmeticOps.test_divmod)N)r'   r(   r)   rP   r>   r<   r[   rQ   markZxfailr]   r^   __classcell__r	   r	   rD   r
   r0   j   s   "r0   c                       s   e Zd Zd fdd	Z  ZS )TestComparisonOpsNc                    s   t  j|||d d d S )Nr8   )r=   r>   r?   rD   r	   r
   r>      s    zTestComparisonOps.check_opname)N)r'   r(   r)   r>   r`   r	   r	   rD   r
   ra      s   ra   c                   @   s   e Zd ZdS )TestReshapingNr&   r	   r	   r	   r
   rb      s   rb   c                   @   s>   e Zd Zejdddgdd Zdd Zdd	 Zd
d Z	dS )TestMethodsna_sentinelc                 C   s   |dkrd}nd}t jt|d" tj||d\}}W d    n1 sH0    Y  tjdd||dddgtjd}|dd	g}t 	|| | 
|| d S )
Nre   z)Specifying `na_sentinel=-1` is deprecatedzDSpecifying the specific value to use for `na_sentinel` is deprecatedrF   )rd   r      r   r   )tmZassert_produces_warningFutureWarningr   	factorizer   r   intptakeassert_numpy_array_equalassert_extension_array_equal)r@   r$   rd   rX   labelsuniquesZexpected_labelsZexpected_uniquesr	   r	   r
   test_factorize   s    0zTestMethods.test_factorizec                    s   |d\}}t |}t |}||dd }t jdd tt|t|D dd}| || |jd  | d	d }t j fd
dt|D dd}| || d S )N   c                 S   s   | |kS r   r	   x1Zx2r	   r	   r
   r      r   z-TestMethods.test_combine_le.<locals>.<lambda>c                 S   s   g | ]\}}||kqS r	   r	   ).0r"   r!   r	   r	   r
   
<listcomp>   r   z/TestMethods.test_combine_le.<locals>.<listcomp>booleanr   r   c                 S   s   | |kS r   r	   rs   r	   r	   r
   r      r   c                    s   g | ]}| kqS r	   r	   )ru   r"   valr	   r
   rv      r   )r   Seriescombineziplistassert_series_equaliloc)r@   Zdata_repeatedZ
orig_data1Z
orig_data2s1s2rY   rZ   r	   rx   r
   test_combine_le   s    


 zTestMethods.test_combine_lec           	      C   s   t jddgdd}|\}}t|||g}|r:t |}||dksLJ |j|dddksbJ ||dkstJ |j|ddd	ksJ ||ddg}tjddgtjd}t	
|| tddg}|j||d
dksJ d S )NTFrw   r   r   right)siderg   rr   )sorter)r   r   typeZ_from_sequencerz   Zsearchsortedrl   r   rk   rh   rm   )	r@   r   Z	as_seriesr!   r"   ZarrrY   rZ   r   r	   r	   r
   test_searchsorted   s    
zTestMethods.test_searchsortedc                 C   sr   |  dksJ | dks J |g d}|  dks>J | dksNJ |  dks^J | dksnJ d S )Nr   rr   )rr   r   r   rg   rg   rr   rg   )ZargmaxZargminrl   )r@   r   r   r   r	   r	   r
   test_argmin_argmax   s    zTestMethods.test_argmin_argmaxN)
r'   r(   r)   rQ   r_   parametrizerq   r   r   r   r	   r	   r	   r
   rc      s
   
rc   c                   @   s   e Zd ZdS )TestCastingNr&   r	   r	   r	   r
   r      s   r   c                   @   s|   e Zd ZdZdd Zejdddgdd Zd	d
 Z	dd Z
dd Zdd Zdd Zdd Zejdddgdd ZdS )TestGroupbyzw
    Groupby-specific tests are overridden because boolean only has 2
    unique values, base tests uses 3 groups.
    c                 C   s\   t g d|d}|djjd }|djjd }t|j|jj	 t
|j| d S )N)Br   NNAr   r   r   r   r   r   r   )r   	DataFramegroupbyZgrouperZ	groupingsrh   rm   Zgrouping_vectorr   valuesrn   )r@   r$   dfZgr1Zgr2r	   r	   r
   test_grouping_grouper  s    z!TestGroupby.test_grouping_grouperas_indexTFc           	      C   s   t g d|d}|jd|dj }t j|dd\}}|rnt j|dd}t jdd	g|d
d}| || n t |dd	gd}| 	|| d S )Nrg   rg   rr   rr      r   rg   r   r   )r   Tsortname      @      ?r   indexr   )r   r   )
r   r   r   r   meanrj   Indexrz   r~   assert_frame_equal)	r@   r   r$   r   rY   r3   rp   r   rZ   r	   r	   r
   test_groupby_extension_agg  s    z&TestGroupby.test_groupby_extension_aggc                 C   s   t g d|d}|jg d }|d}|dddi}| || |dd}| || |d }| || d S )Nr   r   )r   rr   r   r   r   first)r   r   r   Z	set_indexr   Zaggr   r   )r@   r$   r   rZ   rY   r	   r	   r
   test_groupby_agg_extension  s    
z&TestGroupby.test_groupby_agg_extensionc                 C   sl   t g d|d}|jdddj }t j|dd\}}t j|dd}t jddg|d	d
}| || d S )Nr   r   r   Fr   r   r   r   r   r   )	r   r   r   r   r   rj   r   rz   r~   )r@   r$   r   rY   r3   r   rZ   r	   r	   r
   test_groupby_extension_no_sort.  s    z*TestGroupby.test_groupby_extension_no_sortc                 C   sV   ||    }tg d|d}|djt}tjg ddd}| || d S )N)rg   rg   r   r   rg   r   r   )r   r   rr   rr   r   r   r   )	rU   r   r   r   r   Z	transformlenrz   r~   )r@   r$   Zvalidr   rY   rZ   r	   r	   r
    test_groupby_extension_transform7  s
    z,TestGroupby.test_groupby_extension_transformc                 C   sl   t g d|d}|jddd| |jdddj| |jddd| |jdddj| d S )Nr   r   r   F)Z
group_keysr   )r   r   r   applyr   r   )r@   r$   Zgroupby_apply_opr   r	   r	   r
   test_groupby_extension_apply@  s
    z(TestGroupby.test_groupby_extension_applyc                 C   s   t g d|d}|djdd }t j|jjg d j|jjddg j|jjd	d
g jgt jg ddddd}| 	|| d S )Nr   r   r   c                 S   s   | j S r   )r   )r   r	   r	   r
   r   I  r   z9TestGroupby.test_groupby_apply_identity.<locals>.<lambda>)r   rg      rr   r   r      rg   rr   r   r   r   r   )
r   r   r   r   r   rz   r   r   r   r~   r@   r$   r   rY   rZ   r	   r	   r
   test_groupby_apply_identityG  s    	z'TestGroupby.test_groupby_apply_identityc                 C   s^   t g d|g dd}|d j}|jjrBt ddg}nt dg}t	|| d S )Nr   )rg   rg   rg   rg   rg   rg   rg   )r   r   Cr   r   r   )
r   r   r   sumcolumnsr   Z_is_numericr   rh   Zassert_index_equalr   r	   r	   r
   test_in_numeric_groupbyU  s    z#TestGroupby.test_in_numeric_groupby	min_countr   
   c                 C   s   t g d|d}|dj|d}|dkrjt jdt jg ddd	it jg d
ddd}t|| n>t jdt jt jgd dd	it jg d
ddd}t|| d S )Nr   r   r   )r   r   r   )r   r   r   ZInt64r   r   r   )r   r   )	r   r   r   r   r   r   rh   r   r   )r@   r$   r   r   rY   rZ   r	   r	   r
   test_groupby_sum_mincountf  s    z%TestGroupby.test_groupby_sum_mincountN)r'   r(   r)   __doc__r   rQ   r_   r   r   r   r   r   r   r   r   r   r	   r	   r	   r
   r      s   

		r   c                   @   s   e Zd Zdd ZdS )TestNumericReducec                 C   sX   t |||d}t |d||d}t|r8tj}n|dv rHt|}t|| d S )N)skipnafloat64)minmax)	getattrrS   r   isnanr   r   boolrh   Zassert_almost_equal)r@   rA   rB   r   rY   rZ   r	   r	   r
   check_reducey  s    
zTestNumericReduce.check_reduceN)r'   r(   r)   r   r	   r	   r	   r
   r   x  s   r   c                   @   s   e Zd ZdS )TestBooleanReduceNr&   r	   r	   r	   r
   r     s   r   c                   @   s   e Zd ZdS )TestPrintingNr&   r	   r	   r	   r
   r     s   r   c                   @   s   e Zd ZdS )TestUnaryOpsNr&   r	   r	   r	   r
   r     s   r   c                   @   s   e Zd ZdS )TestParsingNr&   r	   r	   r	   r
   r     s   r   c                   @   s   e Zd ZdS )Test2DCompatNr&   r	   r	   r	   r
   r     s   r   )>r   numpyr   rQ   Zpandasr   Zpandas._testingZ_testingrh   Zpandas.core.arrays.booleanr   Zpandas.tests.extensionr   r   Zfixturer   r   r   r   r   r   r   r    r$   ZBaseDtypeTestsr%   ZBaseInterfaceTestsr*   ZBaseConstructorsTestsr+   ZBaseGetitemTestsr,   ZBaseSetitemTestsr-   ZBaseIndexTestsr.   ZBaseMissingTestsr/   ZBaseArithmeticOpsTestsr0   ZBaseComparisonOpsTestsra   ZBaseReshapingTestsrb   ZBaseMethodsTestsrc   ZBaseCastingTestsr   ZBaseGroupbyTestsr   ZBaseNumericReduceTestsr   ZBaseBooleanReduceTestsr   ZBasePrintingTestsr   ZBaseUnaryOpsTestsr   ZBaseParsingTestsr   ZDim2CompatTestsr   r	   r	   r	   r
   <module>   sX   








<Lx