U
    f/e?Q                     @   s   d dl Zd dlZd dlmZmZmZmZmZm	Z	m
Z
 d dlmZ ejdd ZG dd dZG dd dZG d	d
 d
ZG dd dZG dd dZG dd dZG dd dZdS )    N)Float64IndexIndex
Int64Index
RangeIndexSeries	TimestampUInt64Indexc                  C   s   dddddg} t | S )N            l   
        l           l           l           )r   )Zlarge r
   N/tmp/pip-unpacked-wheel-tiezk1ph/pandas/tests/indexes/numeric/test_indexing.pyindex_large   s    r   c                   @   s   e Zd Zejdddddgdd Zejdddddgdd	 Zejd
dddgdd Zejddddgdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zejddgd ged!gd"ggejddddgd#d$ ZdS )%
TestGetLocmethodNpadbackfillnearestc                 C   sD   t dddg}|jd|ddks$t|r@|jd|dddks@td S )Nr         r   r   	tolerancer   get_locAssertionErrorselfr   indexr
   r
   r   test_get_loc   s    zTestGetLoc.test_get_locc              	   C   sL   t dddg}|rd}nd}tjt|d |jddg|d W 5 Q R X d S )Nr   r   r   znot supported betweenzinvalid keymatchr   )r   pytestraises	TypeErrorr   )r   r   r   msgr
   r
   r   test_get_loc_raises_bad_label    s    z(TestGetLoc.test_get_loc_raises_bad_labelz
method,locr   r   r   r   r   r   c                 C   s>   t dddg}|d||ks"t|jd|dd|ks:td S )Nr   r   r   皙?r   r   )r   r   locr   r
   r
   r   test_get_loc_tolerance+   s    z!TestGetLoc.test_get_loc_tolerancec              	   C   s<   t dddg}tjtdd |jd|dd W 5 Q R X d S )	Nr   r   r   z1.1r   r(   g?r)   )r   r    r!   KeyErrorr   r   r
   r
   r   %test_get_loc_outside_tolerance_raises3   s    z0TestGetLoc.test_get_loc_outside_tolerance_raisesc              	   C   s<   t dddg}tjtdd |jdddd	 W 5 Q R X d S )
Nr   r   r   must be numericr   r(   r   invalidr)   r   r    r!   
ValueErrorr   r   r   r
   r
   r   !test_get_loc_bad_tolerance_raises9   s    z,TestGetLoc.test_get_loc_bad_tolerance_raisesc              	   C   s:   t dddg}tjtdd |jddd W 5 Q R X d S )Nr   r   r   ztolerance .* valid ifr   r(   r)   r0   r2   r
   r
   r   'test_get_loc_tolerance_no_method_raises>   s    z2TestGetLoc.test_get_loc_tolerance_no_method_raisesc              	   C   s@   t dddg}tjtdd |jddddgd W 5 Q R X d S )	Nr   r   r   tolerance size must matchr   r(   r   r)   r0   r2   r
   r
   r   &test_get_loc_raises_missized_toleranceC   s    z1TestGetLoc.test_get_loc_raises_missized_tolerancec              	   C   s  t dddg}dD ]8}|d|dks*t|d k	r|jd|dddkstqdD ]4\}}|d	||kslt|jd	|d
d|ksPtqPtjtdd |d W 5 Q R X tjtdd |d W 5 Q R X tjtdd |jdddd W 5 Q R X tjtdd |d W 5 Q R X tjtdd |d W 5 Q R X tjtdd |jdddd W 5 Q R X tjtdd |jddtdgd W 5 Q R X tjtdd  |jddtddgd W 5 Q R X d S )N              ?       @)Nr   r   r   r   r   r)   )r%   r&   r'   r(   g?z^'foo'$r   fooz^1\.5$      ?r   皙?r   z^True$Tz^False$Fr.   gffffff?r   zmust contain numeric elementsz+tolerance size must match target index sizer   )	r   r   r   r    r!   r,   r1   nparray)r   idxr   r*   r
   r
   r   test_get_loc_float64H   s6    " zTestGetLoc.test_get_loc_float64c              	   C   s   t tjddg}|ddks"t|tjdks6tt tjdtjg}|ddksZtd}tjt|d |tj W 5 Q R X t tjdtjtjg}|ddkstd}tjt|d |tj W 5 Q R X d S )Nr   r   r   z7'Cannot get left slice bound for non-unique label: nan'r   z6'Cannot get left slice bound for non-unique label: nan)	r   r=   nanr   r   r    r!   r,   
slice_locs)r   r?   r#   r
   r
   r   test_get_loc_nai   s    zTestGetLoc.test_get_loc_nac              	   C   s   t ddg}|ddksttjtdd |d W 5 Q R X tjtdd |tj W 5 Q R X tjtdd |tjg W 5 Q R X d S )	Nr   r   r   z^3$r      z^nan$z'\[nan\]' is an invalid key)	r   r   r   r    r!   r,   r=   rA   r"   )r   r?   r
   r
   r   test_get_loc_missing_nan|   s    z#TestGetLoc.test_get_loc_missing_nanvalsr   r8   z
2019-12-31testc              	   C   s6   t |}tjtdd |jtj|d W 5 Q R X d S )NrA   r   r   )r   r    r!   r,   r   r=   rA   )r   rF   r   r?   r
   r
   r   (test_get_loc_float_index_nan_with_method   s    z3TestGetLoc.test_get_loc_float_index_nan_with_method)__name__
__module____qualname__r    markparametrizer   r$   r+   r-   r3   r4   r6   r@   rC   rE   r   rH   r
   r
   r
   r   r      s(   


 

! r   c                   @   s2  e Zd Zdd Zejdddgejdejdddd	d	gej	d
dfejdddd	d	gej	d
dfejddd	d	dgej	d
dfejddd	d	dgej	d
dfgdd Z
dd Zejddddddgdddgfdddddgdddgfdddddgdddgfdddddgdddgfdddddgdddgfdddddgdddgfdddddgdd	dgfdddddgd	ddgfdddddgdddgfdd	dddgdd	dgfdd	dddgd	ddgfdd	dddgdddgfdddddgdddgfdddddgdddgfdddddgdddgfgdd Zejdeeeejgejd eed!d!d"gdd"d"gd"d#d#ggdddgdddgdddggd$d% Zd&d' Zejd(ddd)dgfdddd	gfddd)dgfgd*d+ Zejd,eeeegejd-d.d/gd0d1 Zejd-dddgd2d3 Zd4d5 Zd6d7 Zd8d9 Zd:d; Zd<d= ZdS )>TestGetIndexerc                 C   sP   t dddddg}t dddg}||}tjdddgtjd}t|| d S )	Nr   r   rD            dtype)r   get_indexerr=   r>   intptmassert_almost_equal)r   index1index2Zr1e1r
   r
   r   test_get_indexer   s
    
zTestGetIndexer.test_get_indexerreverseTFzexpected,methodrR   r   r   rS   r   Zffillr   r   Zbfillc                 C   s^   t dddddg}t dddg}|r@|d d d }|d d d }|j||d}t|| d S )	Nr   r   rD   rO   rP   rQ   rR   r   )r   rU   rW   rX   )r   r]   expectedr   rY   rZ   resultr
   r
   r   test_get_indexer_methods   s    z'TestGetIndexer.test_get_indexer_methodsc              	   C   sj   t td}tjtdd |jddgdd W 5 Q R X tjtdd |jddgdd W 5 Q R X d S )	N
   ztolerance argumentr   r   r   r)   limit argument)limitr   r=   aranger    r!   r1   rU   r2   r
   r
   r   test_get_indexer_invalid   s
    z'TestGetIndexer.test_get_indexer_invalidz$method, tolerance, indexer, expectedNrP   	   r   皙??      !@   c                 C   s:   t td}|j|||d}t|tj|tjd d S )Nra   r   rS   r   r=   re   rU   rW   assert_numpy_array_equalr>   rV   )r   r   r   indexerr^   r   actualr
   r
   r   test_get_indexer_nearest   s    z'TestGetIndexer.test_get_indexer_nearestlisttypeztolerance, expectedg333333?r<         ?c                 C   sD   t td}|jdddgd||d}t|tj|tjd d S )Nra   rh   ri   rj   r   r   rS   rl   )r   r   r^   rq   r   ro   r
   r
   r   +test_get_indexer_nearest_listlike_tolerance   s      z:TestGetIndexer.test_get_indexer_nearest_listlike_tolerancec              	   C   st   t td}tjtdd |jddgddd W 5 Q R X tjtdd  |jddgddd	d
gd W 5 Q R X d S )Nra   rb   r   r   r   r   )r   rc   r5   r   rD   r   rd   r2   r
   r
   r   test_get_indexer_nearest_error   s
    z-TestGetIndexer.test_get_indexer_nearest_errorzmethod,expected   c                 C   sz   t tdd d d }|jdddg|d}t|tjdddgtjd |jd	d
dg|d}t|tj|tjd d S )Nra   rR   r   rP   rg   r   rO   rS   rh   ri   rj   rl   )r   r   r^   r   ro   r
   r
   r   #test_get_indexer_nearest_decreasing   s
    z2TestGetIndexer.test_get_indexer_nearest_decreasing	idx_classr   rU   get_indexer_non_uniquec                 C   s   |t d}tdddg}t|||}tjdddgtjd}|dkrTt|| n0tjdtjd}t|d | t|d	 | d S )
NrO   TFrR   rS   rU   rD   r   r   )	r   r   getattrr=   r>   rV   rW   rm   re   )r   r   rw   Znumeric_indexotherr_   r^   missingr
   r
   r   -test_get_indexer_numeric_index_boolean_target   s    z<TestGetIndexer.test_get_indexer_numeric_index_boolean_targetc              	   C   sn   t dddg}t ddg}tjtdd |j||d W 5 Q R X tjtdd |j||d W 5 Q R X d S )	Nr   r   rD   TFzCannot comparer   r   )r   r    r!   r"   rU   )r   r   leftrightr
   r
   r   ,test_get_indexer_with_method_numeric_vs_bool  s    z;TestGetIndexer.test_get_indexer_with_method_numeric_vs_boolc                 C   s   t dddg}t ddg}||}dtjt|tjd }t|| ||}dtjt|tjd }t|| ||d }dtjt|tjd }t|| ||d }dtjt|tjd }t|| d S )	Nr   r   rD   TFrR   rS   r   )	r   rU   r=   ZoneslenrV   rW   rm   rx   )r   r}   r~   resr^   r
   r
   r    test_get_indexer_numeric_vs_bool  s    

z/TestGetIndexer.test_get_indexer_numeric_vs_boolc                 C   s   t dddg}t||tjdddgtjd dd	d
g}t||dtjdddgtjd t||dtjdddgtjd t||dtjdddgtjd d S )Nr7   r8   r9   r   r   r   rS   grr   r(   r   rR   r   r   )r   rW   rm   rU   r=   r>   rV   )r   r?   targetr
   r
   r   test_get_indexer_float64.  s$     

 
 
 z'TestGetIndexer.test_get_indexer_float64c                 C   s<   t ddtjgtjg}tjdgtjd}t|| d S )Nr   r   rS   )r   r=   rA   rU   r>   rV   rW   rm   )r   r_   r^   r
   r
   r   test_get_indexer_nan?  s    z#TestGetIndexer.test_get_indexer_nanc                 C   s   t tddd}t td}||}tjddddddddddg
tjd	}t|| t td}|j|d
d}tjddddddddddg
tjd	}t|| t td}|j|dd}tjddddddddddg
tjd	}t|| d S )Nr      r   ra   rR   r   rD   rO   rS   r   r   r   rP   )	r   ranger=   re   rU   r>   rV   rW   rm   )r   r   r   rn   r^   r
   r
   r   test_get_indexer_int64E  s    
$$$z%TestGetIndexer.test_get_indexer_int64c                 C   s  t tddd d }||}tjddddd	d
ddddg
tjd}t|| t tddd d }|j|dd}tjddddd	d
d
d
d
d
g
tjd}t|| t tddd d }|j|dd}tjddddd	d
ddddg
tjd}t|| d S )Nra   uint64rP   r	   r   rR   r   r   rD   rO   rS   r   r   r   )	r   r=   re   ZastyperU   r>   rV   rW   rm   )r   r   r   rn   r^   r
   r
   r   test_get_indexer_uint64V  s    
$$$z&TestGetIndexer.test_get_indexer_uint64)rI   rJ   rK   r\   r    rL   rM   r=   r>   rV   r`   rf   rp   listtupler   ziprs   rt   rv   r   r   r   r   r|   r   r   r   r   r   r   r
   r
   r
   r   rN      s|   	

	
&
	 


rN   c                
   @   sp   e Zd Zejdeejddde	e
dddeejdddgejd	eeejegd
d Zdd ZdS )	TestWherer   rP   Zfloat64rS   r   r   r   r   klassc                 C   sr   dgt | }|}|||}dgdgt |d   }t|jg|dd    }|||}t|| d S )NTFr   )r   wherer   Z	_na_valuetolistrW   assert_index_equal)r   r   r   Zcondr^   r_   r
   r
   r   
test_whereh  s    
zTestWhere.test_wherec                 C   st   t dddg}tdddg}tjdgtjd}t dddg}|||}t|| || |}t|| d S )Nr   rQ   r   FTr   rS   )r   r=   r>   Zint64r   rW   r   Zputmask)r   r?   maskrz   r^   r_   r
   r
   r   test_where_uin64{  s    zTestWhere.test_where_uin64N)rI   rJ   rK   r    rL   rM   r   r=   re   r   r   r   r   r   r>   r   r   r   r
   r
   r
   r   r   g  s   
r   c                   @   sJ   e Zd Zejdeeegdd Z	dd Z
ejdeegdd ZdS )	TestTaker   c                 C   s8   |ddddgdd}| dddg}|j|jks4td S )Nr   r   rD   rO   r:   namer   )taker   r   )r   r   r   Ztakenr
   r
   r   test_take_preserve_name  s    z TestTake.test_take_preserve_namec              	   C   s\  t dddgdd}|tdddg}t dddgdd}t|| |jtdddgd	d
}t ddtjgdd}t|| |jtdddgdd	d}t dddgdd}t|| d}tjt	|d  |jtdddgd	d
 W 5 Q R X tjt	|d  |jtdddgd	d
 W 5 Q R X d}tjt
|d |tddg W 5 Q R X d S )Nr8   r9   g      @xxxr   r   r   rR   T
fill_valueFZ
allow_fillr   zJWhen allow_fill=True and fill_value is not None, all indices must be >= -1r   3index -5 is out of bounds for (axis 0 with )?size 3)r   r   r=   r>   rW   r   rA   r    r!   r1   
IndexError)r   r?   r_   r^   r#   r
   r
   r   test_take_fill_value_float64  s&    $$z%TestTake.test_take_fill_value_float64c              	   C   sd  |dddgdd}| tdddg}|dddgdd}t|| |j}d| d	}tjt|d
  |j tdddgdd W 5 Q R X |j tdddgddd}|dddgdd}t|| tjt|d
  |j tdddgdd W 5 Q R X tjt|d
  |j tdddgdd W 5 Q R X d}tjt	|d
 | tddg W 5 Q R X d S )Nr   r   rD   r   r   r   rR   zUnable to fill values because z cannot contain NAr   Tr   Fr   r   r   r   )
r   r=   r>   rW   r   rI   r    r!   r1   r   )r   r   r?   r_   r^   r   r#   r
   r
   r   test_take_fill_value_ints  s$    $$$z"TestTake.test_take_fill_value_intsN)rI   rJ   rK   r    rL   rM   r   r   r   r   r   r   r
   r
   r
   r   r     s
   
r   c                   @   s8   e Zd Zejdeeegdd Z	dd Z
dd ZdS )	TestContainsr   c                 C   s"   |dddddg}d |kst d S )Nr   r   r   rD   rO   )r   )r   r   r   r
   r
   r   test_contains_none  s    zTestContains.test_contains_nonec                 C   s"   t ddtjg}tj|kstd S Nr8   r9   r   r=   rA   r   r2   r
   r
   r   test_contains_float64_nans  s    z'TestContains.test_contains_float64_nansc                 C   s    t ddtjg}d|kstd S r   r   r2   r
   r
   r   test_contains_float64_not_nans  s    z+TestContains.test_contains_float64_not_nansN)rI   rJ   rK   r    rL   rM   r   r   r   r   r   r   r
   r
   r
   r   r     s   
r   c                   @   sj   e Zd Zejdeegdd Zejdeegdd Z	ejdeegdd Z
dd	 Zd
d ZdS )TestSliceLocsrT   c              
   C   s   t tjddddddddg|d	}t|}|jdd
d|fksBt|jdd
d|fksZt|dddksnt|ddd|fkst|jdddkst|jdddkst|d d d }|dddkst|dddkstd S )Nr   r   r   rP   rQ   ru   rg   ra   rS   startrD   rk   )rD   rQ   end)r   rQ   )r   ru   rR   r   rQ   )r   rP   r   r=   r>   r   rB   r   r   rT   r   nrZ   r
   r
   r   test_slice_locs  s    "zTestSliceLocs.test_slice_locsc              
   C   s   t tjddddddddg|d	}t|}|d
dd|fksBt|dddksVt|d d d }|dddksxt|ddd|fkstd S )Nr   r   r   rP   rQ   ru   rg   ra   rS   g      @g      $@rD   g      @g      %@)rD   rk   rR   rj   r;   r   r   r   r
   r
   r   test_slice_locs_float_locs  s    "z(TestSliceLocs.test_slice_locs_float_locsc                 C   s|   t tjddddg|d}|dddks.t|dddksBt|d d d }|dddksdt|dddksxtd S )	Nra         rS   r   rD         rR   )r   r=   r>   rB   r   )r   rT   r   rZ   r
   r
   r   test_slice_locs_dup_numeric  s    z)TestSliceLocs.test_slice_locs_dup_numericc                 C   sd   t tjddg}|ddks"t|tjdks6tt dtjtjddg}|tjdks`td S )Nr   r   r   )r   rD   r   )r   rP   )r   r=   rA   rB   r   r2   r
   r
   r   test_slice_locs_na  s
    z TestSliceLocs.test_slice_locs_nac              	   C   s`   t tjddg}tjtdd |jdd W 5 Q R X tjtdd |jdd W 5 Q R X d S )Nr   r    r   r;   r   r   )r   r=   rA   r    r!   r,   rB   r2   r
   r
   r   test_slice_locs_na_raises  s
    z'TestSliceLocs.test_slice_locs_na_raisesN)rI   rJ   rK   r    rL   rM   intfloatr   r   r   r   r   r
   r
   r
   r   r     s   



	r   c                   @   sz   e Zd Zejddddgejdddgdd	 Zejddddgejd
ddgejdddgdd ZdS )TestGetSliceBoundskindgetitemr*   Nzside, expected)r}   rO   )r~   rP   c                 C   s,   t td}|jd||d}||ks(td S )NrQ   rO   r   sider   r   Zget_slice_boundr   )r   r   r   r^   r   r_   r
   r
   r   test_get_slice_bounds_within  s    z/TestGetSliceBounds.test_get_slice_bounds_withinr   r}   r~   zbound, expected)rR   r   )ra   rQ   c                 C   s,   t td}|j|||d}||ks(td S )NrQ   r   r   )r   r   r   r^   boundr   r_   r
   r
   r   test_get_slice_bounds_outside  s    z0TestGetSliceBounds.test_get_slice_bounds_outside)rI   rJ   rK   r    rL   rM   r   r   r
   r
   r
   r   r     s   r   )Znumpyr=   r    Zpandasr   r   r   r   r   r   r   Zpandas._testingZ_testingrW   Zfixturer   r   rN   r   r   r   r   r   r
   r
   r
   r   <module>   s   $	
z W"C8