o
    7?e&                     @   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   Zfloat64pdZ	DataFramer   Zvalue_countsr   )sresultexpectedZdfvc r$   f/home/www/facesmatcher.com/pyenv/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   rangeZ	to_seriesr   r   nan)float_numpy_dtyper    r$   r$   r%   test_get_nant   s   r,   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(   )r+   r    idxr$   r$   r%   test_get_nan_multiple{   s   
r1   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   Zint64r   r   ze;Series.__getitem__ treating keys as positions is deprecatedmatchr'   )r   Zaranger   zipr   FutureWarningtmassert_produces_warning)	Zd0Zd1datar   r    r   r4   msgwarnr$   r$   r%   test_get_with_default   s*   rB   arrr-   r'   r2   )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 )r-   r$   ).0r   r$   r$   r%   
<listcomp>   s    z$test_get_with_ea.<locals>.<listcomp>r   r5   r-      r   r      Zabcdefr3   r   r4   )rI   r-   r   Zr8   r9   )r   r)   lenr   ilocr=   Zassert_series_equalslicer   maxlistr>   r<   )rC   Zserr!   r"   r@   Zser2r$   r$   r%   test_get_with_ea   s<   	
rQ   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 )Nr8      r9   rJ   )r   r   rM   r=   r>   r<   )Zstring_seriesZobject_seriesr@   objr0   r$   r$   r%   test_getitem_get   s   
 "rT   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   objectrP   r   )s1s2r    r!   r$   r$   r%   test_get_none   s   

rY   )numpyr   ZpytestZpandasr   r   r   Zpandas._testingZ_testingr=   r&   r,   r1   rB   markZparametrizerandomZdefault_rngZstandard_normalZmakeDateIndexZtz_localizerQ   rT   rY   r$   r$   r$   r%   <module>   s$    h
(