o
    ҷh&                     @   s   d dl Zd dlZd dlZd dlmZmZ d dlmZ	 dd Z
dd Zdd Zd	d
 Zejdejdde	jdddjddgdd Zdd Zdd ZdS )    N)IndexSeriesc                  C   s   t tg d} | dd}d}||ksJ t tg dtg dtjdd} | dd}d}||ks7J tdgd d	gd d
}|j	 }|jddd}|dksWJ |j
	 }|jd	dd}|dksiJ |jddd}|dksvJ d S )N)+   0   <   r   2   3   r   -   9   r   8   r	   r   '   7   r   6   4   r   r      r   )g      9@g      B@g     H@g      P@g     @T@g      Y@g     @^@g      b@g      e@g     h@g     $@g     @@g     d@g     @g     ė@g      @g     D@g     @g     @g     @@dtypeindexr      F)ibc   MissingdefaultT)r   nparraygetr   float64pd	DataFramer   value_countsr   )sresultexpecteddfvc r(   X/var/www/html/venv/lib/python3.10/site-packages/pandas/tests/series/indexing/test_get.pytest_get   s:   4

r*   c                 C   sD   t td| d }|tjd u sJ |jtjdddks J d S )N
   r   r   r   r   range	to_seriesr   r   nan)float_numpy_dtyper#   r(   r(   r)   test_get_nant   s   r1   c                 C   s   t td| d }ddg}||d u sJ dtjg}||d u s%J ddg}||d u s2J tjtjg}||d u sAJ d S )Nr+   r            r,   )r0   r#   idxr(   r(   r)   test_get_nan_multiple{   s   
r6   c            	   
   C   s   g d} t jddd}| |f|| ffD ]z\}}t||d}t||D ]j\}}|||ks/J ||||ks9J ||d|ksCJ |dddksMJ |dddksWJ d}d }|| u rat}tj||d	 |d
ddkssJ |d
d
d
ks}J W d    n1 sw   Y  q"qd S )N)ar   cd   int64r   r   ze;Series.__getitem__ treating keys as positions is deprecatedmatchr+   )r   aranger   zipr   FutureWarningtmassert_produces_warning)	d0d1datar   r#   r   r9   msgwarnr(   r(   r)   test_get_with_default   s*   rK   arrr2   r+   r7   )namez
US/Eastern)tzc                 C   s&  t | dd tt| D d}|d|jd ksJ |ddg}|jddg }t|| |td}|jdd	g }t|| |d
d u sLJ ||j	 d	 d u sZJ t | d d t
dd}|d|jd ksrJ |tdd}|jg d }t|| |d}|d u sJ d}tjt|d |d|jd ksJ W d    n1 sw   Y  tjt|d |d
|jd
 ksJ W d    n1 sw   Y  tjt|d |t|d u sJ W d    n1 sw   Y  t | }|d d d }|d	d u sJ d S )Nc                 S   s   g | ]}d | qS )r2   r(   ).0r   r(   r(   r)   
<listcomp>   s    z$test_get_with_ea.<locals>.<listcomp>r   r:   r2      r   r      abcdefr8   r   r9   )rR   r2   r   Zr>   r?   )r   r-   lenr   ilocrD   assert_series_equalslicer   maxlistrE   rC   )rL   serr$   r%   rI   ser2r(   r(   r)   test_get_with_ea   s<   	
r^   c                 C   s   d}| |fD ]}|j d }|| ||ksJ || |jd ks#J qtjt|d | d| | j d ks;J W d    n1 sEw   Y  | jd | | j d ksYJ d S )Nr>      r?   rS   )r   r   rW   rD   rE   rC   )string_seriesobject_seriesrI   objr5   r(   r(   r)   test_getitem_get   s   
 "rc   c                  C   sB   t td} t ttdd}| |fD ]}|d }|d u sJ qd S )Nr   abc)r   r   )r   objectr[   r   )s1s2r#   r$   r(   r(   r)   test_get_none   s   

rh   )numpyr   pytestpandasr    r   r   pandas._testing_testingrD   r*   r1   r6   rK   markparametrizerandomdefault_rngstandard_normalmakeDateIndextz_localizer^   rc   rh   r(   r(   r(   r)   <module>   s$    h
(