
    o[we'                         d dl Z d dlZd dlZd dlZd dlZd dlmZmZm	Z	m
Z
 d dlmZ  G d d          Zej                            deeeg          d             ZdS )    N)	DataFrameIndexSeries
date_rangec                      e Zd Zd Zd Zd Zd Zd Zej	        
                    d ej        d           ej        d           eg dd	z             ej        d           ej        d           ej        d           ej        d           ej        d           ej        d           ej        d           ed
dg           ed  ed          D                       ej                             edd                    ej                             eg dd                    g          d             Zej	        
                    d ee           edg          g          d             Zd Zd Zd Z d Z!d Z"d Z#d Z$ej	        
                    ddeg          d             Z%ej	        
                    dd d!g          d"             Z&d# Z'd$ Z(d% Z)ej	        
                    d&g d'          d(             Z*d) Z+d* Z,ej	        
                    d+g d,          ej	        
                    de-e.e/eg          d-                         Z0d S ).TestSeriesMiscc                     t          t          d                    }dt          |          v sJ dt          |          vsJ dt          |          vsJ d S )Nabcdstrdtcatr   listdirselfss     <lib/python3.11/site-packages/pandas/tests/series/test_api.pytest_tab_completionz"TestSeriesMisc.test_tab_completion   s\    4<<  A3q66!!!!CFF""""""    c                     t          t          dd                    }dt          |          v sJ dt          |          vsJ dt          |          vsJ d S )N1/1/2015   periodsr   r   r   )r   r   r   r   s     r   test_tab_completion_dtz%TestSeriesMisc.test_tab_completion_dt   sc    :j!44455s1vv~~~~CFF""""CFF""""""r   c                     t          t          d          d          }dt          |          v sJ dt          |          v sJ dt          |          vsJ d S )Nabbcdcategorydtyper   r   r   r   r   s     r   test_tab_completion_catz&TestSeriesMisc.test_tab_completion_cat!   s_     4==
333AA3q66!!!!!!r   c                     t          t          dd                                        d          }dt          |          v sJ dt          |          vsJ dt          |          v sJ d S )Nr   r   r   r   r   r   r   )r   r   astyper   r   s     r   test_tab_completion_cat_strz*TestSeriesMisc.test_tab_completion_cat_str)   sk    :j!44455<<ZHHACFF""""s1vv~~~~~~r   c                 *   g d}t          t          d                                        d          }t          d |j                                        D                       }t          j        |t          t          |                               d S )N)
categoriescodesorderedset_categoriesadd_categoriesremove_categoriesrename_categoriesreorder_categoriesremove_unused_categories
as_orderedas_unorderedaabbcder   c                 <    h | ]}|                     d           |S )_)
startswith).0rs     r   	<setcomp>zFTestSeriesMisc.test_tab_completion_with_categorical.<locals>.<setcomp>A   s)    NNNALL<M<MN!NNNr   )	r   r   r$   sortedr   __dir__tmassert_almost_equalset)r   
ok_for_catr   resultss       r   $test_tab_completion_with_categoricalz3TestSeriesMisc.test_tab_completion_with_categorical0   s    
 
 

 4	??##**:66NNQU]]__NNNOO
ws:(?(?@@@@@r   index
   )foobarbaz   TFc                     g | ]}d | S )a )r6   is     r   
<listcomp>zTestSeriesMisc.<listcomp>R   s    ///q7q77///r   e   ABCDEFGH)r      rF      c                 $   t          |t                    }t          |          }t          |j                            d                    D ]A\  }}|dk     r0t          |t                    r|                                r||v sJ ;||vsJ Bd S )NrA   r!   r   )leveld   )	r   objectr   	enumeraterA   unique
isinstancer   isidentifier)r   rA   r   dir_srJ   xs         r   test_index_tab_completionz(TestSeriesMisc.test_index_tab_completionD   s    * f---Aagnn1n5566 	& 	&DAq3ww%a--SQ^^5E5ESe~~~~~		& 	&r   serr    rO   c                     d}t          j        t          |          5  t          |           d d d            d S # 1 swxY w Y   d S )Nzunhashable type: 'Series'match)pytestraises	TypeErrorhashr   r]   msgs      r   test_not_hashablez TestSeriesMisc.test_not_hashablea   s    )]9C000 	 	III	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s   ;??c                 :    t          j        |j        |           d S N)r;   assert_contains_allrA   )r   datetime_seriess     r   test_containszTestSeriesMisc.test_containsg   s    
4oFFFFFr   c                 ~   t          ddt          j        g          }t          j        |                    d          |                    d                     |                                                    d          dk    sJ |                    d          dk    sJ |                    d          dk    sJ d S )NrO   rF   rows)axisrA   rP   r   )	r   npnanr;   assert_series_equaldropnasum_get_axis_number_get_axis_namer   s     r   test_axis_aliaszTestSeriesMisc.test_axis_aliasj   s    Aq"&>""
qxxVx44ahhGh6L6LMMMxxzz~~f%%****!!&))Q....''7222222r   c                 F    t          j        t          j                  sJ d S ri   )pydocgetdocr   rA   )r   s    r   test_class_axiszTestSeriesMisc.test_class_axisq   s"     |FL)))))))r   c                 .   t          t          j                            d                              d          g dt          dd                    }d }|                    |          }|                                }t          j	        ||           d S )	NrF   )  rP   )ABCz1/1/2000r}   r   )columnsrA   c                 6    | |                                           S ri   )idxmax)r[   s    r   fz-TestSeriesMisc.test_ndarray_compat.<locals>.f~   s    QXXZZ= r   )
r   rp   randomdefault_rngstandard_normalr   applymaxr;   rr   )r   tsdfr   resultexpecteds        r   test_ndarray_compatz"TestSeriesMisc.test_ndarray_compatv   s    I!!!$$44Y??#OOZ666
 
 
	! 	! 	! A88::
vx00000r   c                 .   t          t          j                            d                              d                    }t          t          j        |                    }t          dt          d          d          }t          j        ||           d S )NrF   rB   rO   float64rR   )	r   rp   r   r   r   	ones_likeranger;   rr   )r   r   r   r   s       r   test_ndarray_compat_like_funcz,TestSeriesMisc.test_ndarray_compat_like_func   su    29((++;;B??@@Q((!599I>>>
vx00000r   c                    t          t          j                            d                              d                    }t          j        |                    d          |j                            d                     d S )NrF   rB   F)order)	r   rp   r   r   r   r;   r<   ravelvaluesr   s     r   test_ndarray_compat_ravelz(TestSeriesMisc.test_ndarray_compat_ravel   sd    29((++;;B??@@
qwwSw1118>>>3L3LMMMMMr   c                 B    t          t                    }|j        sJ d S )Nr    )r   rU   empty)r   s_emptys     r   test_empty_methodz TestSeriesMisc.test_empty_method   s&    v&&&}r   r!   int64c                 <    t          dg|          }|j        rJ d S )NrO   rR   )r   r   )r   r!   full_seriess      r   test_empty_method_full_seriesz,TestSeriesMisc.test_empty_method_full_series   s+    A3e444$$$$$$r   NInt64c                 \    t          t          d          |          }|j        dk    sJ d S )N	   r    )r   r   size)r   r!   r   s      r   test_integer_series_sizez'TestSeriesMisc.test_integer_series_size   s/     5885)))v{{{{{{r   c                     t          ddgd          }|j        i k    sJ d|j        d<   |dz   }|j        ddik    sJ d S )Nr   rO   abc)nameversion)r   attrs)r   r   r   s      r   
test_attrszTestSeriesMisc.test_attrs   sY    Aq6&&&w"}}}}	Q|	1~------r   c                     t          j        d           t          t                    }d}t	          j        t          |d          5  t          j        |           d d d            d S # 1 swxY w Y   d S )Njinja2r    zSeries._data is deprecatedF)r`   check_stacklevel)	ra   importorskipr   rU   r;   assert_produces_warningDeprecationWarninginspect
getmembersre   s      r   test_inspect_getmembersz&TestSeriesMisc.test_inspect_getmembers   s    H%%%6"""*'cE
 
 
 	$ 	$ s###	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$ 	$s   A**A.1A.c                 H   t          j        ddd          }t          t          j                            d                              d          |          }d|j        vsJ d	}t          j	        t          |
          5  |j         d d d            d S # 1 swxY w Y   d S )Nr   rB   1s)startr   freqrF   )r   )rA   rC   z&'Series' object has no attribute 'foo'r_   )pdtimedelta_ranger   rp   r   r   normal__dict__ra   rb   AttributeErrorrC   )r   tdir]   rf   s       r   test_unknown_attributez%TestSeriesMisc.test_unknown_attribute   s     q"4@@@RY**1--44"4==SIIICL((((6]>555 	 	GG	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s   BBBop)yeardaysecondweekdayc                     d| d}t          j        t          |          5  t          ||           d d d            d S # 1 swxY w Y   d S )Nz"'Series' object has no attribute ''r_   )ra   rb   r   getattr)r   r   rk   rf   s       r   &test_datetime_series_no_datelike_attrsz5TestSeriesMisc.test_datetime_series_no_datelike_attrs   s     92888]>555 	) 	)OR(((	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	) 	)s   A  AAc                 z    t          dddd          }|j        dk    sJ |j        dk    sJ |j        dk    sJ d S )N  rO   rB   r   monthr   )r   r   r   r   )r   r]   s     r   )test_series_datetimelike_attribute_accessz8TestSeriesMisc.test_series_datetimelike_attribute_access   sN    dQr::;;x4yA~~~~w"}}}}}}r   c                     t          dddd          }d}t          j        t          |          5  |j         d d d            d S # 1 swxY w Y   d S )Nr   rO   rB   r   z*'Series' object has no attribute 'weekday'r_   )r   ra   rb   r   r   re   s      r   1test_series_datetimelike_attribute_access_invalidz@TestSeriesMisc.test_series_datetimelike_attribute_access_invalid   s    dQr::;;:]>555 	 	KK	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	 	s   AA
A
zkernel, has_numeric_only)$)skewT)varT)allF)prodT)anyF)idxminF)quantileF)r   F)minT)semT)meanT)nuniqueF)r   T)rt   T)countF)medianT)stdT)backfillF)rankT)
pct_changeF)cummaxF)shiftF)diffF)cumsumF)cumminF)cumprodF)fillnaF)ffillF)padF)bfillF)sampleF)tailF)takeF)headF)covF)corrFc                    t          g d|          }|dk    r|f}nE|dk    r|f}n;|dk    r|f}n1|dk    rd}n(|dk    rd	}n|dk    rd
}n|dk    rdgf}n|dk    rd}nd}t          ||          }|s@d}t          j        t          |          5   ||ddi d d d            d S # 1 swxY w Y   d S |t
          u rDd| d}t          j        t          |          5   ||ddi d d d            d S # 1 swxY w Y   d S  ||ddi} ||ddi}	t          |	t                     rt          j        ||	           d S ||	k    sJ d S )N)r   rO   rO   r    corrwithr   r   nth)r   r   )T)r   r   r   r   )g      ?rI   zP(got an unexpected keyword argument 'numeric_only'|too many arguments passed in)r_   numeric_onlyTzSeries.z2 does not allow numeric_only=True with non-numericF)	r   r   ra   rb   rc   rU   rX   r;   rr   )
r   kernelhas_numeric_onlyr!   r]   argsmethodrf   r   r   s
             r   test_numeric_onlyz TestSeriesMisc.test_numeric_only   s|   X YYYe,,,Z6DDv6DDu__6DDu__DDxDDxDDvC6DDz!!DDDf%% 	*1  y444 1 1040001 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1f__VFVVVCy444 1 1040001 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 VT555Fvt8%88H(F++ *&vx88888 ))))))s$   	B((B,/B,	C55C9<C9)1__name__
__module____qualname__r   r   r"   r%   r@   ra   markparametrizer;   makeStringIndexmakeCategoricalIndexr   makeDateIndexmakePeriodIndexmakeTimedeltaIndexmakeIntIndexmakeUIntIndexmakeFloatIndexr   r   
MultiIndexfrom_tupleszipr\   r   rU   rg   rl   rw   r{   r   r   r   r   r   r   r   r   r   r   r   r   boolintfloatr   rI   r   r   r   r      s       # # ## # #" " "  A A A( [Br""#B#B''E'''!+,,BR  Br""!B!"%%BOBBR  BOBBb!!E4-  E//EE#JJ///00M%%cc&&&9&9::M%%cc,,,&?&?@@	
 && &' && [UVV&%9%9%9661#;;$GHH  IH
G G G3 3 3* * *
1 1 11 1 1N N N
   [Ww&788% % 98% [WtWo66  76
. . .$ $ $   [T#G#G#GHH) ) IH)     ["%	
 %	
 %	
( (R [WtS%&@AA)* )* BAS( (T)* )* )*r   r   	converterc                     t          j        t                    5   | t          dg                     | d          k    sJ 	 d d d            d S # 1 swxY w Y   d S )NrO   )r;   r   FutureWarningr   )r  s    r   test_float_int_deprecatedr  $  s     
	#M	2	2 6 6y%%1555556 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s   )AAA)r   ry   numpyrp   ra   pandasr   r   r   r   r   pandas._testing_testingr;   r   r   r   r  r  complexr  rI   r   r   <module>r     s                               P* P* P* P* P* P* P* P*f sE7&;<<6 6 =<6 6 6r   