a
    Pf                     @   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	 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ejdeedejdddgdd ZdS )    N)	DataFrameSeriesc                   C   s   dS )N    r   r   r   l/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/tests/indexing/multiindex/test_indexing_slow.pym   s    r   c                   C   s   dS )Nd   r   r   r   r   r   n   s    r	   c                   C   s   g dS )N)ZjimZjoeZjolieZjolinejoliar   r   r   r   r   cols   s    r   c                 C   s   t jddd| t jdtd| t jdtjddd | t jdtd| t jd	| g}tt
tt| }|S )N   r   
   Z
abcdefghij20141009ZperiodsZ
ZYXWVUTSRQ)nprandomdefault_rngintegerschoicelistpd
date_rangetolistZstandard_normalmaptuplezip)r	   valsr   r   r   r      s    	r   c                 C   s   t jddd|t jdtd|t jdtjddd |t jdtd|g}tt	t
t| }|dd	 |d d | |  D 7 }|S )
Nr   r      Zabcdefghijkr   r   ZZYXWVUTSRQPc                 S   s   g | ]}|d d qS )Nr   ).0tr   r   r   
<listcomp>6       zkeys.<locals>.<listcomp>)r   r   r   r   r   r   r   r   r   r   r   r   )r	   r   r   keysr   r   r   r#   *   s     r#   c                 C   s   t | |dS )N)columns)r   )r   r   r   r   r   df;   s    r%   c                 C   s   t | | gS )N)r   concat)r%   r   r   r   a@   s    r'   c                 C   s   | j |d d dS )Nr   )Zsubset)Zdrop_duplicates)r%   r   r   r   r   bE   s    r(   z(ignore::pandas.errors.PerformanceWarninglexsort_depthr   frame_fixturec              	   C   s  |  |}|dkr |jdd}n|j|d | d}||d d }|jj|k rVJ |D ]}tjt|t	d}	t
|D ]\}
}|	|jd d |
f |kM }	|	 s|d |
d  |jvszJ qz|d |
d  |jv sJ ||	 jdd}|
d t|krp|j|d |
d  ddd	}|d u s(J |j||
d d dd
}|d u sPJ t|j|d |
d   | qz|j|d d dd
}|d u sJ t|dkrt|d j|jd dgd}t|j|d |
d   | qzt|j|d |
d   | qzqZd S )Nr   F)deep)Zbyr   )Zdtype   T)Zaxisinplace)r-   r
   )nameindex)ZgetfixturevaluecopyZsort_valuesZ	set_indexr/   Z_lexsort_depthr   Zoneslenbool	enumerateZilocanyZdroptmZassert_frame_equallocr   valuesZassert_series_equal)requestr)   r#   r*   r   framer%   mikeymaskikrightZreturn_valuer   r   r   test_multiindex_get_locJ   s:    

  r@   )numpyr   ZpytestZpandasr   r   r   Zpandas._testingZ_testingr5   Zfixturer   r	   r   r   r#   r%   r'   r(   markfilterwarningsZparametrizer   ranger@   r   r   r   r   <module>   s0   








