o
    d[x                     @  s   d dl mZ d dlmZmZ d dlZd dlZd dlZd dlm	Z	m
Z
 d dlZd dlmZ ejddddd	 ZG d
d dZG dd deZG dd deZG dd deZG dd deZG dd deZdS )    )annotations)datetime	timedeltaN)IS64is_platform_windowsTclass)autousescopec                 #  sz     j }t|j|j|jg} fdd} jj}|js|j	r$d V  d S |D ]}||s7t
d|j d| q&d V  d S )Nc                   s.   | \ j jj}t fdd|D S )Nc                 3  s.    | ]}|j v o |j v o|j v V  qd S Nname.0xdtypeklassmethod h/var/www/html/visualizacion-main/env/lib/python3.10/site-packages/pandas/tests/indexing/test_coercion.py	<genexpr>#   s    
z<check_comprehensiveness.<locals>.has_test.<locals>.<genexpr>)nodesessionitemsany)combo	cls_funcsrequestr   r   has_test    s
   

z)check_comprehensiveness.<locals>.has_testztest method is not defined: z, )cls	itertoolsproductklassesdtypesr   configoptionlfkeywordAssertionError__name__)r   r    combosr   optsr   r   r   r   check_comprehensiveness   s   

r-   c                   @  s(   e Zd ZddgZg dZedd ZdS )CoercionBaseindexseries)	objectint64float64
complex128bool
datetime64datetime64tztimedelta64periodc                 C  s   t | r
   NotImplementedErrorselfr   r   r   r   E   s   zCoercionBase.methodN)r*   
__module____qualname__r#   r$   propertyr   r   r   r   r   r.   7   s
    r.   c                   @  sH  e Zd ZU dZg Zded< dd Zdd Zej	
dd	efd
efdefgdd Zej	
dd
ejfdejfd	efgdd Zej	
dd
ejfdejfd	efgdd Zej	jdddd Zej	jdddd Zej	jdddd Zej	jdddd Zej	jdddd Zej	jdddd  Zej	jddd!d" Zd#S )$TestSetitemCoercionsetitemz	list[str]r#   c                 C  s>   t ddg}t|j|d d < t ddg}t|| d S )Na   )pdSerieslistvaluestmassert_series_equal)r=   serexpectedr   r   r   0test_setitem_series_no_coercion_from_values_listQ   s   zDTestSetitemCoercion.test_setitem_series_no_coercion_from_values_listc                 C  s   |  }d||< tjg d|d}t|| |jj|ksJ |  }d|j|< tjg d|d}t|| |jj|ks?J dS )z-test index's coercion triggered by assign key   )rD            rN   r/   N)copyrE   rF   rI   rJ   r/   r   loc)r=   original_seriesloc_keyexpected_indexexpected_dtypetempexpr   r   r    _assert_setitem_index_conversionZ   s   
z4TestSetitemCoercion._assert_setitem_index_conversionzval,exp_dtyper   rN   皙?c                 C  s   t jg dtdd}|jjtksJ |tu r;| }d}tj	||d d|d< W d    d S 1 s4w   Y  d S t 
td|g }| |||| d S )NrD   rO   rP   rQ   abcdrR   z/index 5 is out of bounds for axis 0 with size 4)matchrN   )rE   rF   rG   r/   r   r1   
IndexErrorrS   pytestraisesIndexr[   )r=   val	exp_dtypeobjrY   msg	exp_indexr   r   r   test_setitem_index_objectn   s   
"z-TestSetitemCoercion.test_setitem_index_objectc                 C  sH   t g d}|jjtjksJ t dddd|g}| |||| d S )Nr]   r   rD   rO   rP   )rE   rF   r/   r   npr2   rc   r[   )r=   rd   re   rf   rh   r   r   r   test_setitem_index_int64~   s   z,TestSetitemCoercion.test_setitem_index_int64gffffff@c                 C  sP   t jg dg dd}|jjtjksJ t dddd|g}| |||| d S )Nr]   )r\    @@ffffff@rR   r\   rl   rm   rn   )rE   rF   r/   r   rj   r3   rc   r[   )r=   rd   re   r   rf   rh   r   r   r   test_setitem_index_float64   s   z.TestSetitemCoercion.test_setitem_index_float64Test not implementedreasonc                 C     t r
   r:   r<   r   r   r   test_setitem_series_period      z.TestSetitemCoercion.test_setitem_series_periodc                 C  rs   r
   r:   r<   r   r   r   test_setitem_index_complex128   ru   z1TestSetitemCoercion.test_setitem_index_complex128c                 C  rs   r
   r:   r<   r   r   r   test_setitem_index_bool   ru   z+TestSetitemCoercion.test_setitem_index_boolc                 C  rs   r
   r:   r<   r   r   r   test_setitem_index_datetime64   ru   z1TestSetitemCoercion.test_setitem_index_datetime64c                 C  rs   r
   r:   r<   r   r   r   test_setitem_index_datetime64tz   ru   z3TestSetitemCoercion.test_setitem_index_datetime64tzc                 C  rs   r
   r:   r<   r   r   r   test_setitem_index_timedelta64   ru   z2TestSetitemCoercion.test_setitem_index_timedelta64c                 C  rs   r
   r:   r<   r   r   r   test_setitem_index_period   ru   z-TestSetitemCoercion.test_setitem_index_periodN)r*   r>   r?   r   r#   __annotations__rM   r[   ra   markparametrizer1   r`   ri   rj   r2   r3   rk   ro   xfailrt   rv   rw   rx   ry   rz   r{   r   r   r   r   rA   J   s>   
 	








rA   c                	   @  s  e Zd ZdgZdZdd Zejddde	fdde	fdde	fd	d	e	fgd
d Z
ejddddejfdde	fd	d	e	fgdd Zejddddejfdde	fd	d	e	fgdd Zejjdeddfejddddfgddgdejdedejddddgdd Zdd  Zejdejd!d"d#d!d$fedede	fdde	fd	d	e	fgd%d& Zejjd'd(d)d* Zejjd'd(d+d, Zd-S ).TestInsertIndexCoercionr/   insertc                 C  s2   |  }|d|}t|| |j|ksJ dS )z!test coercion triggered by insertrD   N)rS   r   rI   assert_index_equalr   r=   originalvaluerL   rX   targetresr   r   r   _assert_insert_conversion   s   z1TestInsertIndexCoercion._assert_insert_conversionz"insert, coerced_val, coerced_dtyperD   r\   Fr   c                 C  sD   t td}|jtksJ t d|dddg}| |||| d S )Nr^   rC   bcd)rE   rc   rG   r   r1   r   )r=   r   coerced_valcoerced_dtyperf   rZ   r   r   r   test_insert_index_object   s   
z0TestInsertIndexCoercion.test_insert_index_object)rD   rD   Nc                 C  R   |}t jg d|d}|d ur|n|}t jd|dddg|d}| |||| d S )Nr]   r   rD   rO   rP   rQ   rE   rc   r   )r=   any_int_numpy_dtyper   r   r   r   rf   rZ   r   r   r   test_insert_int_index   
   z-TestInsertIndexCoercion.test_insert_int_index)rD         ?Nc                 C  r   )N)r          @      @      @r   r   r   r   r   r   )r=   float_numpy_dtyper   r   r   r   rf   rZ   r   r   r   test_insert_float_index   r   z/TestInsertIndexCoercion.test_insert_float_indexfill_val,exp_dtype
2012-01-01datetime64[ns]
US/Easterntzdatetime64[ns, US/Eastern]r6   r7   idsinsert_value
Asia/Tokyoc           
      C  s|  t jg d|jd}|j|ksJ t jd| dddg|jd}| |||| |jrst d}|d|}|t	d|}|jt	ksFJ t
|| t jdd	d}|d|}|d||jj}|j|jkslJ t
|| n#t jdd	d}|d|}|t	d|}|jt	ksJ t
|| d}	|d|	}|t	d|	}|d |	ksJ |jt	ksJ t
|| d S )
N)
2011-01-01
2011-01-02
2011-01-03
2011-01-04r   r   r   r   r   r   rD   r   )rE   DatetimeIndexr   r   dater   	Timestampr   astyper1   rI   r   
tz_convert)
r=   fill_valre   r   rf   rZ   tsresultrL   itemr   r   r   test_insert_index_datetimes   s>   

z3TestInsertIndexCoercion.test_insert_index_datetimesc                 C  s   t g d}|jdksJ t g d}| |t d|d t ddfD ]}|d|}|td|}|jtks?J t	
|| q'd S )N)1 day2 day3 day4 daytimedelta64[ns])r   10 dayr   r   r   r   r   rD   )rE   TimedeltaIndexr   r   	Timedeltar   r   r   r1   rI   r   )r=   rf   rZ   r   r   rL   r   r   r   test_insert_index_timedelta64&  s   z5TestInsertIndexCoercion.test_insert_index_timedelta64z2012-01Mfreq	period[M]c           	      C  s  t jg ddd}|jdksJ t jddd|t jdddt jdddt jdddg}t|t jrLt j|dd}| |||| | |t||| d S |d	|}|t	d	|}t
|| t|t js|d	t|}|t	d	t|}t
|| d S d S )
N)2011-012011-022011-032011-04r   r   r   r   r   r   r   r   )rE   PeriodIndexr   Period
isinstancer   strr   r   r1   rI   r   r   )	r=   r   r   r   rf   datarZ   r   rL   r   r   r   test_insert_index_period6  s(   
z0TestInsertIndexCoercion.test_insert_index_periodrp   rq   c                 C  rs   r
   r:   r<   r   r   r   test_insert_index_complex128^  ru   z4TestInsertIndexCoercion.test_insert_index_complex128c                 C  rs   r
   r:   r<   r   r   r   test_insert_index_boolb  ru   z.TestInsertIndexCoercion.test_insert_index_boolN)r*   r>   r?   r#   r   r   ra   r}   r~   r1   r   rj   r3   r   r   rE   r   r   r   r   r   r   r   r   r   r   r   r   r      sr    
	

	


	
*
	
r   c                	   @  s  e Zd ZdZeg dZdd Zdd Zdd Z	e
jd	d
efdefdefdefgdd Ze
jd	d
ejfdejfdejfdefgdd Ze
jdd
ejfdejfdejfdefgdd Ze
jd	d
ejfdejfdejfdefgdd Ze
jd	d
efdefdefdejfgdd Ze
jjd	eddfejdddefgddgdd d! Ze
jjd"d#d$d% Ze
jjd"d#d&d' Ze
jjd"d#d(d) Ze
jjd"d#d*d+ Ze
jd,ejd-d.e d-d.e!d-d/gd0d1 Z"d2d3 Z#d4S )5TestWhereCoercionwhereTFTFc                 C  s2   |  }|||}t|| |j|ksJ dS )z test coercion triggered by whereN)rS   r   rI   assert_equalr   )r=   r   condrH   rL   rX   r   r   r   r   r   _assert_where_conversionk  s   z*TestWhereCoercion._assert_where_conversionc                   sx    du r|g d}nt  ttjfrtj dd}n| fdddD }||d |d	 |d
 |d g|d}||fS )NT)TFTTrQ   periodsc                 3  s    | ]}|  V  qd S r
   r   r   r   r   r   r   z  s    z3TestWhereCoercion._construct_exp.<locals>.<genexpr>)rN            r   rD   rO   rP   r   )r   r   rj   r6   rE   
date_range)r=   rf   r   r   re   rH   rZ   r   r   r   _construct_expt  s   $z TestWhereCoercion._construct_expc                 C  sb   || j }||d ||d |g|d}| ||||| | ||||\}}| ||||| d S )Nr   rO   r   )_condr   r   )r=   rf   r   r   re   r   rZ   rH   r   r   r   	_run_test  s
   
zTestWhereCoercion._run_testr   rD   r\         ?      ?Tc                 C  s2   |}|t d}|jtksJ | |||| d S )Nr^   )rG   r   r1   r   r=   index_or_seriesr   re   r   rf   r   r   r   test_where_object  s   z#TestWhereCoercion.test_where_objectc                 C  4   |}|g d}|j tjksJ | |||| d S )Nr]   )r   rj   r2   r   r=   r   r   re   r   r   rf   r   r   r   test_where_int64     z"TestWhereCoercion.test_where_int64zfill_val, exp_dtypec                 C  r   )N)r\   皙@ffffff
@皙@)r   rj   r3   r   r   r   r   r   test_where_float64  r   z$TestWhereCoercion.test_where_float64c                 C  s:   |}|g dt jd}|jt jksJ | |||| d S )N)r          @       @      @      @      @      @r   )rj   r4   r   r   r   r   r   r   test_where_complex128  s   
z'TestWhereCoercion.test_where_complex128c                 C  s6   t j}|g d}|jtjksJ | |||| d S )Nr   )rE   rF   r   rj   bool_r   )r=   r   re   r   rf   r   r   r   test_where_series_bool  s   z(TestWhereCoercion.test_where_series_boolr   r   r   r   r6   r7   r   c                 C  s   |}|t jddddd }|jdksJ |}|dkr2|| | fD ]
}| |||| q%d S || fD ]
}| |||| q8d S )Nr   rQ   Dr   r   r   )rE   r   
_with_freqr   to_pydatetimeto_datetime64r   )r=   r   r   re   r   rf   fvscalarr   r   r   test_where_datetime64  s   	z'TestWhereCoercion.test_where_datetime64rp   rq   c                 C  rs   r
   r:   r<   r   r   r   test_where_index_complex128  ru   z-TestWhereCoercion.test_where_index_complex128c                 C  rs   r
   r:   r<   r   r   r   test_where_index_bool  ru   z'TestWhereCoercion.test_where_index_boolc                 C  rs   r
   r:   r<   r   r   r   test_where_series_timedelta64  ru   z/TestWhereCoercion.test_where_series_timedelta64c                 C  rs   r
   r:   r<   r   r   r   test_where_series_period  ru   z*TestWhereCoercion.test_where_series_periodr   	   daysr   c                 C  s   t jddd}tg d}t d||dg}|||}t|| tdd}t j	|d |||d	 gt
d
}|d |u s@J |||}t|| d S )Nz1 DayrQ   r   )TFFTz4 DaysNaTnsr   rP   r   rD   )rE   timedelta_rangerj   arrayr   r   rI   r   r6   rc   r1   )r=   r   tdir   rL   r   dtnatr   r   r   test_where_index_timedelta64  s   z.TestWhereCoercion.test_where_index_timedelta64c           
      C  s.  t jdddd}|d}tg d}|d |jd  }t ||d	 |g}|||}t	|| tj
||jd  td
}|||}t |d |d	 |d g}t	|| t jdd}t j||d	 |gtd
}|||}t	|| t dd}	t j|	|d	 |	gtd
}|||	}t	|| d S )N
2016-01-01rP   QSr   Q)FTF
   rD   r   r   rO   rQ   r   z
2020-04-21r   )rE   r   	to_periodrj   r  r   r   r   rI   r   asarrayr1   r   rc   r   )
r=   dtipir   r   rL   r   othertdperr   r   r   test_where_index_period  s&   
z)TestWhereCoercion.test_where_index_periodN)$r*   r>   r?   r   rj   r  r   r   r   r   ra   r}   r~   r1   r   r2   r3   r4   r   r   r   r   r   rE   r   r   r   r   r   r   r   r   r   r8   r  r  r   r   r   r   r   g  sp    		
 
 

	





 
r   c                   @  s  e Zd ZdZejjdddd Zdd Zej	dd	e
fd
e
fde
fde
fgdd Zej	dd	ejfd
ejfdejfde
fgdd Zej	dd	ejfd
ejfdejfde
fgdd Zejj	deddfejddde
fd	e
fde
fgg dddd Zej	dejddddfede
fejddddfd	e
fde
fgdd  Zej	d!d	d
ddejd	d"d#d$ejdddedejd	d%ed&d'g	d(d) Zejjddd*d+ Zejjddd,d- Zejjddd.d/ Zejjddd0d1 Zejjddd2d3 Zej	d!d	d
ddejd	d"d#d$ejdddedejd	d%ed&d4g	d5d6 Zejjddd7d8 Zejjddd9d: Z d;S )<TestFillnaSeriesCoercionfillnarp   rq   c                 C  rs   r
   r:   r<   r   r   r   test_has_comprehensive_tests  ru   z5TestFillnaSeriesCoercion.test_has_comprehensive_testsc                 C  s0   |  }||}t|| |j|ksJ dS )z!test coercion triggered by fillnaN)rS   r  rI   r   r   r   r   r   r   _assert_fillna_conversion#  s   
z2TestFillnaSeriesCoercion._assert_fillna_conversionzfill_val, fill_dtyperD   r\   r   Tc                 C  sH   |}|dt jddg}|jtksJ |d|ddg}| |||| d S )NrC   r   r   )rj   nanr   r1   r  r=   r   r   
fill_dtyper   rf   rZ   r   r   r   test_fillna_object*  s
   z+TestFillnaSeriesCoercion.test_fillna_objectzfill_val,fill_dtypec                 C  sJ   |}|dt jddg}|jt jksJ |d|ddg}| |||| d S )Nr\   r   r   )rj   r  r   r3   r  r  r   r   r   test_fillna_float646  s
   z,TestFillnaSeriesCoercion.test_fillna_float64c                 C  sP   |}|dt jddgt jd}|jt jksJ |d|ddg}| |||| d S )Nr   r   r   r   )rj   r  r4   r   r  r  r   r   r   test_fillna_complex128B  s
   
z/TestFillnaSeriesCoercion.test_fillna_complex128r   r   r   r   r   )r6   r7   r1   r1   r   c                 C  sl   |}|t dt jt dt dg}|jdksJ |t d|t dt dg}| |||| d S )Nr   r   r   r   )rE   r   r   r   r  r  r   r   r   test_fillna_datetimeS  s"   z-TestFillnaSeriesCoercion.test_fillna_datetimer   r   c           	      C  s   |}d}|t jd|dt jt jd|dt jd|dg}|jdks#J t|dd d u r.|}n||}|t jd|d|t jd|dt jd|dg}| |||| d S )Nr   r   r   r   r   r   r   )rE   r   r   r   getattrr   r  )	r=   r   r   r  r   r   rf   r   rZ   r   r   r   test_fillna_datetime64tzs  s*   
z1TestFillnaSeriesCoercion.test_fillna_datetime64tzr   rO   leftclosedr   r  r   c                 C  sr   t jdddddtj}t|jt jsJ ||}||d ||d |d |d	 gtd
}t}| 	|||| d S )Nr   g      @rightr   rD   r   rO   rP   rQ   r   )
rE   interval_ranger   rj   r  r   r   IntervalDtyper1   r  )r=   r   r   iirf   rZ   r  r   r   r   test_fillna_interval     &z-TestFillnaSeriesCoercion.test_fillna_intervalc                 C  rs   r
   r:   r<   r   r   r   test_fillna_series_int64  ru   z1TestFillnaSeriesCoercion.test_fillna_series_int64c                 C  rs   r
   r:   r<   r   r   r   test_fillna_index_int64  ru   z0TestFillnaSeriesCoercion.test_fillna_index_int64c                 C  rs   r
   r:   r<   r   r   r   test_fillna_series_bool  ru   z0TestFillnaSeriesCoercion.test_fillna_series_boolc                 C  rs   r
   r:   r<   r   r   r   test_fillna_index_bool  ru   z/TestFillnaSeriesCoercion.test_fillna_index_boolc                 C  rs   r
   r:   r<   r   r   r   test_fillna_series_timedelta64  ru   z7TestFillnaSeriesCoercion.test_fillna_series_timedelta64Wc                 C  sr   t jdddddt j}t|jt jsJ ||}||d ||d |d |d gtd	}t}| |||| d S )
Nr  rQ   r   r   rD   r   rO   rP   r   )	rE   period_ranger   r   r   r   PeriodDtyper1   r  )r=   r   r   r  rf   rZ   r  r   r   r   test_fillna_series_period  r'  z2TestFillnaSeriesCoercion.test_fillna_series_periodc                 C  rs   r
   r:   r<   r   r   r   test_fillna_index_timedelta64  ru   z6TestFillnaSeriesCoercion.test_fillna_index_timedelta64c                 C  rs   r
   r:   r<   r   r   r   test_fillna_index_period  ru   z1TestFillnaSeriesCoercion.test_fillna_index_periodN)!r*   r>   r?   r   ra   r}   r   r  r  r~   r1   r  rj   r3   r4   r  r  rE   r   r  r  Intervalr   r   r&  r(  r)  r*  r+  r,  r0  r1  r2  r   r   r   r   r    s    

 

	
















r  c                   @  s  e Zd ZU dgZdZi Zded< ddged< dd	ged
< ddged< ddged< ddged< ededged< dD ]Z	de	 dZ
ejde	dejde	dgee
< q<ededged< ejddgdd d! Zejg d"dd#d$ Zejg d"g d%d&d'd( Zejd)d* Zd+d, Zejjd-g d.dd/ejjd0d1d2gdd/d3d4 Zejjd-g d5dd/ejjd0g d5dd/d6d7 Zejjd8d9d:d; Zd<S )=TestReplaceSeriesCoercionr0   replacezdict[str, list]reprC   r   r1   rQ   rN   r2   r\   r   r3   r   r   r4   TFr5   r   r   r   )UTCr   zdatetime64[ns, ]r   r   r   r   dict)paramsc                 C     |j S r
   paramr=   r   r   r   r   how  s   zTestReplaceSeriesCoercion.how)	r1   r2   r3   r4   r5   r   datetime64[ns, UTC]r   r   c                 C  r;  r
   r<  r>  r   r   r   from_key  s   z"TestReplaceSeriesCoercion.from_key)	r1   r2   r3   r4   r5   r6   r7   r7   r8   )r:  r   c                 C  r;  r
   r<  r>  r   r   r   to_key  s   z TestReplaceSeriesCoercion.to_keyc                 C  sP   |dkrt t| j| | j| }|S |dkr&tj| j| | j| d}|S t)z9
        Object we will pass to `Series.replace`
        r9  r0   rR   )r9  zipr6  rE   rF   
ValueError)r=   r?  rA  rB  replacerr   r   r   rE  ,  s   z"TestReplaceSeriesCoercion.replacerc           	      C  s   t jddgdd}t j| j| |dd}|j|ksJ |dr'|dr'd S |dv r-d S ||}|d	kr:|d
v sB|dkr_|dv r_trGt rRt	
d| d|  t j| j| |d|d}nt j| j| |dd}|j|ksqJ t|| d S )NrP   rQ   xxxr   yyyr/   r   r   )r   r@  r3   r2   r4   )r2   r3   z32-bit platform buggy: z -> )r/   r   r   )rE   rc   rF   r6  r   
startswithr5  r   r   ra   skiprI   rJ   	r=   r?  rB  rA  rE  r/   rf   r   rZ   r   r   r   test_replace_series9  s&   

z-TestReplaceSeriesCoercion.test_replace_seriesrB  )r   r5   r1   r4   r3   r2   )indirectrA  r@  r   c           	      C  st   t jddgdd}t j| j| |dd}|j|ksJ ||}t j| j| |dd}|j|ks2J t|| d S NrP   rQ   xyzr   rG  rH  )rE   rc   rF   r6  r   r5  rI   rJ   rK  r   r   r   test_replace_series_datetime_tzV  s   	
z9TestReplaceSeriesCoercion.test_replace_series_datetime_tz)r   r@  r   c           	      C  s   t jddgdd}t j| j| |dd}|j|ksJ ||}t j| j| |dd}t|jt jr@t|jt jr@||j}n|j|ksGJ t	
|| d S rN  )rE   rc   rF   r6  r   r5  r   DatetimeTZDtyper   rI   rJ   rK  r   r   r   %test_replace_series_datetime_datetimej  s   
z?TestReplaceSeriesCoercion.test_replace_series_datetime_datetimerp   rq   c                 C  rs   r
   r:   r<   r   r   r   test_replace_series_period  ru   z4TestReplaceSeriesCoercion.test_replace_series_periodN)r*   r>   r?   r#   r   r6  r|   rE   r   r   keyr   ra   fixturer?  rA  rB  rE  rL  r}   r~   rP  rR  r   rS  r   r   r   r   r4    sh   
 





r4  )
__future__r   r   r   r!   numpyrj   ra   pandas.compatr   r   pandasrE   pandas._testing_testingrI   rU  r-   r.   rA   r   r   r  r4  r   r   r   r   <module>   s&    
e 9 4 N