o
    ŀg                     @   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      dS )N    r   r   r   g/var/www/html/myenv/lib/python3.10/site-packages/pandas/tests/indexing/multiindex/test_indexing_slow.pym      r   c                   C   r   )Nd   r   r   r   r   r   n   r	   r   c                   C   s   g dS )N)jimjoejolie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   
   
abcdefghij20141009periods
ZYXWVUTSRQ)nprandomdefault_rngintegerschoicelistpd
date_rangetolist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      abcdefghijkr   r   ZYXWVUTSRQPc                 S   s   g | ]}|d d qS )Nr   ).0tr   r   r   
<listcomp>6   s    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   r0   c                 C   s   t | | gS )N)r   concat)r0   r   r   r   a@   s   r2   c                 C   s   | j |d d dS )Nr*   )subset)drop_duplicates)r0   r   r   r   r   bE   s   r5   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 r+J |D ]}tjt|t	d}	t
|D ]\}
}|	|jd d |
f |kM }	|	 s_|d |
d  |jvs^J q<|d |
d  |jv slJ ||	 jdd}|
d t|kr|j|d |
d  ddd	}|d u sJ |j||
d d dd
}|d u sJ t|j|d |
d   | q<|j|d d dd
}|d u sJ t|dkrt|d j|jd dgd}t|j|d |
d   | q<t|j|d |
d   | q<q-d S )Nr   F)deep)byr*   )dtype   T)axisinplace)r=   r   )nameindex)getfixturevaluecopysort_values	set_indexr?   _lexsort_depthr   oneslenbool	enumerateilocanydroptmassert_frame_equallocr   valuesassert_series_equal)requestr6   r.   r7   r   framer0   mikeymaskikrightreturn_valuer   r   r   test_multiindex_get_locJ   s>   
   rZ   )numpyr   pytestpandasr   r   r   pandas._testing_testingrL   fixturer   r   r   r&   r.   r0   r2   r5   markfilterwarningsparametrizer   rangerZ   r   r   r   r   <module>   s2    








