
    d`W                         d dl 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lZd dlmZ d dlmZ d dlmZ  G d de          ZdS )	    N)Dtype)is_bool_dtype)na_value_for_dtype)nargsort)BaseExtensionTestsc                      e Zd ZU dZd Zej                            dddg          d             Zd Z	d Z
d	 Zd
 Zd Zd Zd Zd Zej                            dddg          d             Zej                            dddg          d             Zej                            dddddddej        fddej        fddg          d             Zd Zej                            dd  ej        g d! ej        d"          #          fd$ ej        g d% ej        d"          #          fg          d&             Zej                            d'ddg          d(             Zej                            d'ddg          d)             Zej                            d'ddg          d*             Zej                            d+ej        d, g          ej                            dd- ej        g          d.                         Z d/ Z!d0 Z"d1 Z#d2 Z$d3 Z%d4 Z& ej        e'          Z(e)e*d5<   d6 Z+d7 Z,d8 Z-ej                            d9ddg          ej                            d:d;g d<fd=g d>fd?g d@fg          dA                         Z.dB Z/ej                            dCdDd;g          dE             Z0ej                            d:dFdGdGggdGdDdGggd=d=dDggdDdGd=ggdHdGdGggg          dI             Z1ej                            dCg dJ          dK             Z2dL Z3dM Z4dN Z5dO Z6dP Z7dQ Z8ej                            dRd=dDd?g dSg          dT             Z9ej                            dUd?dVdDie:dVfdGi e:dWfdDd?gi e:dXfd?dYdZie;d[fg          d\             Z<d] Z=d^ Z>d_ Z?d` Z@ej                            d+ej        ej        ejA        g          da             ZBdbS )cBaseMethodsTestsz%Various Series and DataFrame methods.c                     t          |d          s$t          j        dt          |                      t	          j        |j                  }|j        d         }|j        du sJ d S )Nvalue_countsz$value_counts is not implemented for dropnaT)	hasattrpytestskiptypeinspect	signaturer   
parametersdefault)selfdatasigkwargs       _/var/www/html/t/fyr/venv311/lib/python3.11/site-packages/pandas/tests/extension/base/methods.py test_value_counts_default_dropnaz1BaseMethodsTests.test_value_counts_default_dropna   sl    t^,, 	MKKtDzzKKLLL 122x(}$$$$$$    r   TFc                 n   |d d         }|r||                                           }n|}t          j        |                              |                                          }t          j        |                              |                                          }|                     ||           d S )N
   )r   )isnapdSeriesr   
sort_indexassert_series_equal)r   all_datar   otherresultexpecteds         r   test_value_countsz"BaseMethodsTests.test_value_counts   s    CRC= 	hmmoo-.EEE8$$111@@KKMM9U##000??JJLL  22222r   c                    |d d                                          }t          j        ||                                                    }t	          j        ||j                  }|                    d                                          }t          |t          j
                  s>t	          j        dt          |          z  gt          |          z  |j        d          }n5t	          j        d|j        d          }dt          |          z  ||d	k    <   t          |j                  t          j        u r|                    d
          }|                     ||           d S )Nr   dtypeT)	normalize   
proportion)indexnameg        r   Float64)uniquenparrayr   r   r    r*   r   r!   
isinstanceCategoricallenr.   r   NAastyper"   )r   r   valuesserr%   r&   s         r    test_value_counts_with_normalizez1BaseMethodsTests.test_value_counts_with_normalize*   s4   CRCy!!$		|,--iDJ///!!D!11<<>>$// 	3yS[[!CKK/v|,  HH yFL|LLLH#$s6{{?HVaZ dj))RU22y11H  22222r   c                     t          j        d|i          }|                    d          }t          j        ddg          }|                     ||           d S )NAcolumnsaxisr   r,   )r   	DataFramecountr    r"   )r   data_missingdfr%   r&   s        r   
test_countzBaseMethodsTests.test_count?   sX    \3-..y))9aV$$  22222r   c                 j    t          j        |          }|                                }d}||k    sJ d S )Nr,   )r   r    rB   )r   rC   r:   r%   r&   s        r   test_series_countz"BaseMethodsTests.test_series_countE   s:    i%%!!!!!!r   c                     t          j        |                              t                    }t	          |t           j                  sJ d S N)r   r    applyidr4   r   r   r%   s      r   test_apply_simple_seriesz)BaseMethodsTests.test_apply_simple_seriesL   s;    4&&r**&"),,,,,,,r   c                     t          j        |                                          }t          j        t          j        g dt          j                            }|                     ||           d S N   r   r,   r)   r   r    argsortr2   r3   intpr"   )r   data_for_sortingr%   r&   s       r   test_argsortzBaseMethodsTests.test_argsortP   s^    +,,44669RXiiirw???@@  22222r   c                     |                                 }t          j        g dt          j                  }t	          j        ||           d S rO   )rS   r2   r3   rT   tmassert_numpy_array_equalr   data_missing_for_sortingr%   r&   s       r   test_argsort_missing_arrayz+BaseMethodsTests.test_argsort_missing_arrayV   sF    )11338IIIRW555
#FH55555r   c                     t          j        |                                          }t          j        t          j        g dt          j                            }|                     ||           d S )N)r,   r   r)   rR   rZ   s       r   test_argsort_missingz%BaseMethodsTests.test_argsort_missing\   s^    344<<>>9RXjjj@@@AA  22222r   c                 l   |                                 dk    sJ |                                dk    sJ |                    g d          }|                                 dk    sJ |                                dk    sJ |                                 dk    sJ |                                dk    sJ d S )Nr,   rQ   )rQ   r   r   r,   r,   rQ      r   )argmaxargmintake)r   rU   r[   na_valuer   s        r   test_argmin_argmaxz#BaseMethodsTests.test_argmin_argmaxa   s      &&((A----&&((A----  $$%7%7%788{{}}!!!!{{}}!!!! (..00A5555'..00A555555r   methodrb   rc   c                     d}t          j        t          |          5   t          |d d         |                       d d d            d S # 1 swxY w Y   d S )Nattempt to getmatchr   )r   raises
ValueErrorgetattr)r   rg   r   err_msgs       r   test_argmin_argmax_empty_arrayz/BaseMethodsTests.test_argmin_argmax_empty_arrayr   s     #]:W555 	( 	(%GD!Hf%%'''	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	( 	(s   !AAAc                     d}t          |                              ||g|j                  }t          j        t
          |          5   t          ||                       d d d            d S # 1 swxY w Y   d S )Nri   r)   rj   )r   _from_sequencer*   r   rl   rm   rn   )r   rg   r   re   ro   data_nas         r   test_argmin_argmax_all_naz*BaseMethodsTests.test_argmin_argmax_all_nay   s     #t**++Xx,@
+SS]:W555 	' 	'$GGV$$&&&	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	' 	's   	A//A36A3zop_name, skipna, expected)idxmaxTr   )idxminTrQ   )rb   Tr   )rc   TrQ   ru   rv   )rb   Fr^   )rc   Fr^   c                     t          j        |          } t          ||          |          }t          j        ||           d S )Nskipna)r   r    rn   rX   assert_almost_equal)r   r[   op_namery   r&   r:   r%   s          r   test_argreduce_seriesz&BaseMethodsTests.test_argreduce_series   sI    " i011&g&&f555
vx00000r   c                 0   |}t          j        t          d          5  |                    d           d d d            n# 1 swxY w Y   t          j        t          d          5  |                    d           d d d            d S # 1 swxY w Y   d S )N rj   Frx   )r   rl   NotImplementedErrorrc   rb   )r   r[   r   s      r   +test_argmax_argmin_no_skipna_notimplementedz<BaseMethodsTests.test_argmax_argmin_no_skipna_notimplemented   s   '].b999 	& 	&KKuK%%%	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& ].b999 	& 	&KKuK%%%	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	&s#   AAA'BBBzna_position, expectedlastrP   rT   r)   firstr,   rQ   r   c                 R    t          ||          }t          j        ||           d S )N)na_position)r   rX   rY   )r   r[   r   r&   r%   s        r   test_nargsortzBaseMethodsTests.test_nargsort   s0     2LLL
#FH55555r   	ascendingc                    t          j        |          }|                    ||          }|j        g d         }|s7|                                dk    r|j        g d         }n|j        g d         }|                     ||           d S )Nr   keyrP   rQ   )r   r,   rQ   )r,   r   rQ   )r   r    sort_valuesilocnuniquer"   )r   rU   r   sort_by_keyr:   r%   r&   s          r   test_sort_valuesz!BaseMethodsTests.test_sort_values   s    i())9+FF8III& 	/{{}}!!8III.8III.  22222r   c                     t          j        |          }|                    ||          }|r|j        g d         }n|j        g d         }|                     ||           d S )Nr   rP   )r   rQ   r,   )r   r    r   r   r"   )r   r[   r   r   r:   r%   r&   s          r   test_sort_values_missingz)BaseMethodsTests.test_sort_values_missing   so     i0119+FF 	+x			*HHx			*H  22222r   c                     t          j        g d|d          }|                    ddg          }t          j        g d|                    g d          dg d          }|                     ||           d S )N)r,   rQ   r,   r=   Br=   r   )r,   r,   rQ   rP   r.   )r   rA   r   rd   assert_frame_equal)r   rU   r   rD   r%   r&   s         r   test_sort_values_framez'BaseMethodsTests.test_sort_values_frame   s    \			0@AABBc
++<))"2"7"7			"B"BCC999
 
 
 	11111r   boxc                     | S rI    xs    r   <lambda>zBaseMethodsTests.<lambda>   s    ! r   c                 *    |                                  S rI   )r1   r   s    r   r   zBaseMethodsTests.<lambda>   s    !((** r   c                     ||                     |d         |d         g                    } ||          }t          |          dk    sJ t          |t          |                    sJ |d         |d         k    sJ d S Nr   r,   )rr   r6   r4   r   )r   r   r   rg   
duplicatedr%   s         r   test_uniquezBaseMethodsTests.test_unique   s     S,,d1gtAw-?@@AA

##6{{a&$t**-----ayJqM))))))r   c                     t          j        |d          \  }}t          j        g dt          j                  }|                    g d          }t          j        ||           |                     ||           d S )NTuse_na_sentinel)r   r   r^   r^   r,   r,   r   rQ   r)   )r         )	r   	factorizer2   r3   rT   rd   rX   rY   assert_extension_array_equal)r   data_for_groupingcodesuniquesexpected_codesexpected_uniquess         r   test_factorizezBaseMethodsTests.test_factorize   s    &7NNNw"<"<"<BGLLL,11)))<<
#E>:::))'3CDDDDDr   c                 L   t          j        |d          \  }}|                    d          \  }}t          j        ||           |                     ||           t          |          t          t          j        |                    k    sJ |j        |j        k    sJ d S )NTr   )r   r   rX   rY   r   r6   r1   r*   )r   r   codes_1	uniques_1codes_2	uniques_2s         r   test_factorize_equivalencez+BaseMethodsTests.test_factorize_equivalence   s    \*;TRRR.888NN
#GW555)))Y???9~~RYy%9%9!:!:::::"3"9999999r   c                 <   t          j        |d d                   \  }}t          j        g t          j                  }t          |                              g |d d         j                  }t          j	        ||           | 
                    ||           d S )Nr   r)   )r   r   r2   r3   rT   r   rr   r*   rX   rY   r   )r   r   r   r   r   r   s         r   test_factorize_emptyz%BaseMethodsTests.test_factorize_empty   s    d2A2h//w"BG444::44RtBQBx~4NN
#E>:::))'3CDDDDDr   c                    |                     ddg          }t          j        d|i          }|                                }|j        d         }|                    |          }||j        d<   |                     ||           d S )Nr,   r=   )r   r   )rd   r   rA   copyr   fillnar   )r   rC   arrrD   df_orig
filled_valr%   s          r   test_fillna_copy_framez'BaseMethodsTests.test_fillna_copy_frame   s}    A''\3*%%''))WT]
:&&&DG,,,,,r   c                    |                     ddg          }t          j        |d          }|                                }|d         }|                    |          }||j        d<   |                     ||           d S )Nr,   F)r   r   )rd   r   r    r   r   r   r"   )r   rC   r   r:   ser_origr   r%   s          r   test_fillna_copy_seriesz(BaseMethodsTests.test_fillna_copy_series   s{    A''i%(((88::V
J''#A  h/////r   c                     d}t          j        t          |          5  |                    |                    dg                     d d d            d S # 1 swxY w Y   d S )Nz!Length of 'value' does not match.rj   r,   )r   rl   rm   r   rd   )r   rC   msgs      r   test_fillna_length_mismatchz,BaseMethodsTests.test_fillna_length_mismatch
  s    1]:S111 	8 	8 1 11# 6 6777	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8 	8s   *AAA_combine_le_expected_dtypec           	      D    |d          \  }}t          j        |          }t          j        |          }|                    |d           }t          j        d t          t	          |          t	          |                    D             | j                  }|                     ||           |j        d         |                    d           }t          j        fdt	          |          D             | j                  }|                     ||           d S )NrQ   c                     | |k    S rI   r   x1x2s     r   r   z2BaseMethodsTests.test_combine_le.<locals>.<lambda>  s
    rRx r   c                      g | ]\  }}||k    S r   r   .0abs      r   
<listcomp>z4BaseMethodsTests.test_combine_le.<locals>.<listcomp>  s     JJJAQ!VJJJr   r)   r   c                     | |k    S rI   r   r   s     r   r   z2BaseMethodsTests.test_combine_le.<locals>.<lambda>   s
    b r   c                     g | ]}|k    	S r   r   r   r   vals     r   r   z4BaseMethodsTests.test_combine_le.<locals>.<listcomp>"  s    000!Q#X000r   )r   r    combineziplistr   r"   r   	r   data_repeated
orig_data1
orig_data2s1s2r%   r&   r   s	           @r   test_combine_lez BaseMethodsTests.test_combine_le  s(    "/q!1!1
JYz""Yz""B 7 7889JJ#d:&6&6Z8H8H"I"IJJJ1
 
 
 	  222gajC!8!89990000tJ//0001
 
 
 	  22222r   c                     |d          \  }}t          j        |          }t          j        |          }|                    |d           }t          j        d          5  t          j        |                    d t          t          |          t          |                    D                                 }d d d            n# 1 swxY w Y   |                     ||           |j	        d         |                    d           }t          j        |                    fdt          |          D                                 }|                     ||           d S )	NrQ   c                     | |z   S rI   r   r   s     r   r   z3BaseMethodsTests.test_combine_add.<locals>.<lambda>,  s
    rBw r   ignore)overc                     g | ]
\  }}||z   S r   r   r   s      r   r   z5BaseMethodsTests.test_combine_add.<locals>.<listcomp>0  s     QQQv1QUQQQr   r   c                     | |z   S rI   r   r   s     r   r   z3BaseMethodsTests.test_combine_add.<locals>.<lambda>6  s
    R r   c                     g | ]}|z   S r   r   r   s     r   r   z5BaseMethodsTests.test_combine_add.<locals>.<listcomp>8  s    &I&I&I1q3w&I&I&Ir   )
r   r    r   r2   errstaterr   r   r   r"   r   r   s	           @r   test_combine_addz!BaseMethodsTests.test_combine_add'  s   !.q!1!1
JYz""Yz""B 6 677[h''' 	 	y))QQT*-=-=tJ?O?O)P)PQQQ  H	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	  222gajC!7!7889%%&I&I&I&IZ8H8H&I&I&IJJ
 
 	  22222s   $AC

CCc                    t          j        |d d                   }t          j        |dd         g d          }|                    |          }t          j        |d d                   }|                     ||           d S )Nra   rQ      )rQ   ra   r   r   )r   r    combine_firstr"   )r   r   r   r   r%   r&   s         r   test_combine_firstz#BaseMethodsTests.test_combine_first<  s}    Id2A2hId1Q3iyyy111##9T"1"X&&  22222r   framezperiods, indices)rQ   ra   r   r^   r^   r   )r   r,   rQ   ra   r   rQ   )r^   r^   r   r,   rQ   c                     |d d         }t          j        |d          }t          j        |                    |d          d          }|r|                    d                              d                              |          }t          j        |t          j        dgdz  d                              |          gd	          }| j        }n|                    |          }| j        } |||           d S )
Nr   r=   r/   T
allow_fillr,   )r   r   r?   )	r   r    rd   to_frameassignshiftconcatr   r"   )	r   r   r   periodsindicessubsetr&   r%   compares	            r   test_container_shiftz%BaseMethodsTests.test_container_shiftD  s    bqbyc***9V[[T[BBMMM 	/]]],,33a388>>wGGFy29aS1W3777==gFFGa  H -GGZZ((F.G!!!!!r   c                     |                     d          }|d         |d         k    sJ |d         |d<   |d         |d         k    sJ d S r   r   rL   s      r   test_shift_0_periodsz%BaseMethodsTests.test_shift_0_periods[  sV    AAw$q'!!!!q'QayF1I%%%%%%r   r   r,   c                    |d d         }t          |j                  rt          j        }nt          j        }	  |||           n4# t
          $ r' t          j        t          |           d           Y nw xY wt          j
        |          }|                    |          }t          j
         |||                    |                              }|                     ||           t          j        |dgdz  d          }|                    |          }|dk    rt          j        ddddg}ndddt          j        t          j        g}t          j        ||d          }|                     ||           d S )Nr   z does not support diffg      ?r   r,   r   )r   r*   operatorxorsub	Exceptionr   r   r   r   r    diffr   r"   rA   r2   nanr   )	r   r   r   opsr%   r&   rD   r   s	            r   	test_diffzBaseMethodsTests.test_diffb  sr   BQBx$$ 	BBB	?BtTNNNN 	? 	? 	?K4::===>>>>>	?IdOO9RRdjj&9&9::;;  222\C5195566!!a<<Aq!$AAAq"&"&)A<hQ 7 78811111s   A .A76A7r^   r   c                     |d d         }|                     |          }|                    |d          }|                     ||           d S )NrQ   Tr   r   rd   r   )r   r   r   r   r   r%   r&   s          r   test_shift_non_empty_arrayz+BaseMethodsTests.test_shift_non_empty_array|  sS     bqbg&&;;w4;88))&(;;;;;r   )r  r^   r   r,   r   c                 t    |d d         }|                     |          }|}|                     ||           d S )Nr   )r   r   )r   r   r   emptyr%   r&   s         r   test_shift_empty_arrayz'BaseMethodsTests.test_shift_empty_array  sC     RaRW%%))&(;;;;;r   c                     |                     d          }||usJ |d d                              d          }||usJ d S )Nr   rQ   r   rL   s      r   test_shift_zero_copiesz'BaseMethodsTests.test_shift_zero_copies  sR    AT!!!!bqb""T!!!!!!r   c                 :   |d d         }|d         }|                     d|          }|                    g d          }|                     ||           |                     d|          }|                    g d          }|                     ||           d S )Nr   r   r,   )
fill_value)r   r   r,   rQ   r   )rQ   ra   r   r   r
  )r   r   r   r  r%   r&   s         r   test_shift_fill_valuez&BaseMethodsTests.test_shift_fill_value  s    2A2h!W
14499\\\**))&(;;;2*5599\\\**))&(;;;;;r   c                     t          j        t          d          5  t          |           d d d            d S # 1 swxY w Y   d S )Nzunhashable typerj   )r   rl   	TypeErrorhash)r   r   s     r   test_not_hashablez"BaseMethodsTests.test_not_hashable  s    ]9,=>>> 	 	JJJ	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s   9= =c                    t          j        |          }|r|                                }t           j                            |          }t           j                            |          }|                     ||           d S rI   )r   r    r   utilhash_pandas_objectassert_equal)r   r   as_framer   r   s        r   test_hash_pandas_object_worksz.BaseMethodsTests.test_hash_pandas_object_works  sj    y 	#==??DG&&t,,G&&t,,!Qr   c                    |\  }}}|                     g d          }|rt          j        |          }|                    |          dk    sJ |                    |d          dk    sJ |                    |          dk    sJ |                    |d          dk    sJ |                    |          dk    sJ |                    |d          dk    sJ |                    |                     ddg                    }t	          j        ddgt          j                  }t          j        ||           t	          j        g d	          }	|                    ||	
          dk    sJ d S )NrP   r   right)sider,   rQ   ra   r)   r   )sorter)	rd   r   r    searchsortedr2   r3   rT   rX   rY   )
r   rU   	as_seriesr   cr   r   r%   r&   r!  s
             r   test_searchsortedz"BaseMethodsTests.test_searchsorted  s   "1a##III.. 	!)C..C""a''''00A5555""a''''00A5555""a''''00A5555!!#((Aq6"2"2338QF"'222
#FH555 )))$$,,Qv,>>!CCCCCCr   c                 *   |d         |d         k    sJ t          |          }|d d         \  }}t          j        |                    ||||g|j                            }|                                }t          j        g d          }	|r,|                    d          }|		                    dd          }	|
                    |	          }
t          j        |                    ||||g|j                            }|r|                    d          }|                     |
|           |                    |	 d	
           |                     ||           |                                }|r|                    d          }t          j        g d          }	|                    ||||g|j                  }|r,t          j        d|i          }t          j        d|	i          }	|
                    |	|          }
t          j        |                    ||||g|j                            }|r|                    d          }|                     |
|           |                    |	 |d	
           |                     ||           d S )Nr   r,   rQ   r)   )TTFFr   r   r^   T)inplace)TFTT)r   r   r    rr   r*   r   r2   r3   r   reshapewherer  maskrA   )r   r   re   r  clsr   r   origr:   condr%   r&   r$   s                r   test_where_seriesz"BaseMethodsTests.test_where_series  s   Aw$q'!!!!4jjBQBx1y++Q1aL
+KKLLiikkx22233 	',,C,((C<<A&&D491h9LL
 
  	3((c(22H&(+++$%%%#x((( iikk 	),,C,((Cx11122""Aq!Q<tz"BB 	-L#u..E<d,,D4''9S//Aq!DJ/OOPP 	3((c(22H&(+++$t,,,#x(((((r   repeats)r,   rQ   ra   c                 8   t          |                              |d d         |j                  }|rt          j        |          }|rt          j        ||          n|                    |          }t          |t                    r|gdz  n|}d t          ||          D             }t          |                              ||j                  }|r.t          j        ||j
                            |                    }|                     ||           d S )Nra   r)   c                 <    g | ]\  }}t          |          D ]}|S r   )range)r   r   n_s       r   r   z0BaseMethodsTests.test_repeat.<locals>.<listcomp>  s/    FFF$!QU1XXFFAFFFFr   r   )r   rr   r*   r   r    r2   repeatr4   intr   r.   r  )r   r   r/  r#  	use_numpyr   r%   r&   s           r   test_repeatzBaseMethodsTests.test_repeat  s
   4jj''RaR
'CC 	!)C..C,5N3(((3::g;N;N#-gs#;#;H7)a--FF#c7"3"3FFF::,,XTZ,HH 	Ly1A1A'1J1JKKKH&(+++++r   zrepeats, kwargs, error, msgr@   negativeshapefoobarz'foo'c                     t          j        ||          5  |rt          j        ||fi | n |j        |fi | d d d            d S # 1 swxY w Y   d S )Nrj   )r   rl   r2   r5  )r   r   r/  kwargserrorr   r7  s          r   test_repeat_raisesz#BaseMethodsTests.test_repeat_raises  s     ]5,,, 	/ 	/ /	$2262222G..v...		/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/ 	/s   %A		AAc                 &   |                     d          }|dd          }|                     ||           |                     ddg          }|                    |dg         |dg         |dd          g          }|                     ||           d S )Nr   r,   ra   rQ   r   )deleter   _concat_same_type)r   r   r%   r&   s       r   test_deletezBaseMethodsTests.test_delete  s    Q8))&(;;;aV$$))49dA3iabb*JKK))&(;;;;;r   c                 B   |dd                               d|d                   }|                     ||           |dd                               t          |dd                     |d                   }|                     ||           |d d                              d|d                   }t          j        t          |                    }|dd         |dd <   t          |          dz
  |d<   |                    |          }|                     ||           d S )Nr,   r   r^   r   r   )insertr   r6   r2   arangerd   )r   r   r%   takerr&   s        r   test_insertzBaseMethodsTests.test_insert  s	   abbDG,,))&$777abb#d122h--a99))&$777 crc!!!T"X..	#d))$$!B$Kabb	t99q=a99U##))&(;;;;;r   c                     |}t          j        t          t          f          5  |                    d|           d d d            n# 1 swxY w Y   t          j        t          t          f          5  |                    d|           d d d            n# 1 swxY w Y   t          j        t          t          f          5  |                    t          |          dz
  |           d d d            d S # 1 swxY w Y   d S )Nr   r   r,   )r   rl   r  rm   rF  r6   )r   r   invalid_scalaritems       r   test_insert_invalidz$BaseMethodsTests.test_insert_invalid-  s   ]Iz233 	! 	!KK4   	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! ]Iz233 	! 	!KK4   	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! 	! ]Iz233 	- 	-KKD		At,,,	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	- 	-s5   AA
A
1BBB?'C33C7:C7c                    t          |          }t          j        t                    5  |                    |dz   |d                    d d d            n# 1 swxY w Y   t          j        t                    5  |                    | dz
  |d                    d d d            n# 1 swxY w Y   t          j        t
                    5  |                    d|d                    d d d            d S # 1 swxY w Y   d S )Nr,   r   g      ?)r6   r   rl   
IndexErrorrF  r  )r   r   ubs      r   test_insert_invalid_locz(BaseMethodsTests.test_insert_invalid_loc9  s   YY]:&& 	) 	)KKQQ(((	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) ]:&& 	* 	*KKaa)))	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* 	* ]9%% 	& 	&KKT!W%%%	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	& 	&s5    AAA9!B&&B*-B*
C44C8;C8c                    t          |                              |d         gt          |          z  |j                  }t          |                              |gt          |          z  |j                  }t	          j        ||d          }t	          j        ||d          }t	          j        ||d          }|                    |          du sJ |                    |                                          du sJ |                    |          du sJ |                    |          du sJ |d d                             |d d                   du sJ |d d                             |d d                   du sJ |                    d           du sJ |dg                             |d                   du sJ d S )Nr   r)   F)	transposeTrQ   ra   )r   rr   r6   r*   rX   box_expectedequalsr   )r   r   re   r#  r   data2rs   s          r   test_equalszBaseMethodsTests.test_equalsF  s   T

))47)c$ii*?tz)RRt**++XJT,B$*+UUtSE:::se<<</'3%@@@
 {{4  D(((({{499;;''4//// {{5!!U****{{7##u,,,, BQBxtBQBx((E1111 BQBxtBQBx((D0000 {{4  E))))QCyQ((E111111r   N)C__name__
__module____qualname____doc__r   r   markparametrizer'   r;   rE   rG   rM   rV   r\   r_   rf   rp   rt   r2   r  r|   r   r3   r*   r   r   r   r   r   r    r1   r   r   r   r   r   r   r   boolr   r   __annotations__r   r   r   r   r   r  r  r  r  r  r  r  r%  r.  r8  rm   r  r@  rD  rI  rM  rQ  rA   rW  r   r   r   r	   r	      s        //% % % [Xe}55
3 
3 65
33 3 3*3 3 3" " "- - -3 3 36 6 63 3 3
6 6 6" [X(';<<( ( =<( [X(';<<' ' =<' [#ubf%ubf%!!		
 1 1 1& & & [XRXiiixrx/?/?@@@Ahbhyyy0@0@AAAB	
 6 6 6
 [[4-883 3 983 [[4-88	3 	3 98	3 [[4-882 2 982 [URY$<==[X(<(<bi'HII* * JI >=*E E E: : :E E E
- 
- 
-	0 	0 	08 8 8 )16663 3 3*3 3 3*3 3 3 [WtUm44[
	 1ooo"6<M<M<M8NO " "	  54
"$& & & [YB002 2 1022 [
r2h"q"gQFa"a\ABx=Q < <	 < [Y(9(9(9::< < ;:<" " "	< 	< 	<  
     D D D0)) )) ))V [YAq)))(<==, , >=, [%Z0Z,VRW-	73		
 / / /< < << < <"
- 
- 
-& & & [URXry",$GHH2 2 IH2 2 2r   r	   )r   r   numpyr2   r   pandas._typingr   pandas.core.dtypes.commonr   pandas.core.dtypes.missingr   pandasr   pandas._testing_testingrX   pandas.core.sortingr    pandas.tests.extension.base.baser   r	   r   r   r   <module>ri     s                       3 3 3 3 3 3 9 9 9 9 9 9           ( ( ( ( ( ( ? ? ? ? ? ?O	2 O	2 O	2 O	2 O	2) O	2 O	2 O	2 O	2 O	2r   