o
    7?ey                     @  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)Z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 d/home/www/facesmatcher.com/pyenv/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)comboZ	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   Z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timedelta64Z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)r9   Z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   )r@            rI   r,   N)copyrA   rB   rE   rF   r,   r   loc)r9   Zoriginal_seriesZloc_keyZexpected_indexexpected_dtypetempexpr   r   r    _assert_setitem_index_conversionZ   s   
z4TestSetitemCoercion._assert_setitem_index_conversionzval,exp_dtyper   rI   皙?c              	   C  s   t jg dtdd}|jjtksJ |tu r\| }d}d}tj	||d, t
jt|d d|d< W d    n1 s=w   Y  W d    d S W d    d S 1 sUw   Y  d S t td|g }| |||| d S )Nr@   rJ   rK   rL   abcdrM   z;Series.__setitem__ treating keys as positions is deprecatedz/index 5 is out of bounds for axis 0 with size 4)matchrI   )rA   rB   rC   r,   r   r.   
IndexErrorrN   pytestZraisesrE   Zassert_produces_warningFutureWarningIndexrS   )r9   val	exp_dtypeobjrQ   Zwarn_msg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 )NrU   r   r@   rJ   rK   )rA   rB   r,   r   npr/   r[   rS   )r9   r\   r]   r^   r`   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 )NrU   )rT    @@ffffff@rM   rT   rd   re   rf   )rA   rB   r,   r   rb   r0   r[   rS   )r9   r\   r]   r   r^   r`   r   r   r   test_setitem_index_float64   s   z.TestSetitemCoercion.test_setitem_index_float64Test not implementedreasonc                 C     t r	   r6   r8   r   r   r   test_setitem_series_period      z.TestSetitemCoercion.test_setitem_series_periodc                 C  rk   r	   r6   r8   r   r   r   test_setitem_index_complex128   rm   z1TestSetitemCoercion.test_setitem_index_complex128c                 C  rk   r	   r6   r8   r   r   r   test_setitem_index_bool   rm   z+TestSetitemCoercion.test_setitem_index_boolc                 C  rk   r	   r6   r8   r   r   r   test_setitem_index_datetime64   rm   z1TestSetitemCoercion.test_setitem_index_datetime64c                 C  rk   r	   r6   r8   r   r   r   test_setitem_index_datetime64tz   rm   z3TestSetitemCoercion.test_setitem_index_datetime64tzc                 C  rk   r	   r6   r8   r   r   r   test_setitem_index_timedelta64   rm   z2TestSetitemCoercion.test_setitem_index_timedelta64c                 C  rk   r	   r6   r8   r   r   r   test_setitem_index_period   rm   z-TestSetitemCoercion.test_setitem_index_periodN)r(   r:   r;   r   r!   __annotations__rH   rS   rY   markparametrizer.   rX   ra   rb   r/   r0   rc   rg   xfailrl   rn   ro   rp   rq   rr   rs   r   r   r   r   r=   J   s>   
 	








r=   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 insertr@   N)rN   ry   rE   assert_index_equalr   r9   originalvaluerG   rP   targetresr   r   r   _assert_insert_conversion   s   z1TestInsertIndexCoercion._assert_insert_conversionz"insert, coerced_val, coerced_dtyper@   rT   Fr   c                 C  sD   t td}|jtksJ t d|dddg}| |||| d S )NrV   r?   bcd)rA   r[   rC   r   r.   r   )r9   ry   coerced_valcoerced_dtyper^   rR   r   r   r   test_insert_index_object   s   
z0TestInsertIndexCoercion.test_insert_index_object)r@   r@   Nc                 C  R   |}t jg d|d}|d ur|n|}t jd|dddg|d}| |||| d S )NrU   r   r@   rJ   rK   rL   rA   r[   r   )r9   Zany_int_numpy_dtypery   r   r   r   r^   rR   r   r   r   test_insert_int_index   
   z-TestInsertIndexCoercion.test_insert_int_index)r@         ?Nc                 C  r   )N)r          @      @      @r   r   r   r   r   r   )r9   Zfloat_numpy_dtypery   r   r   r   r^   rR   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]r3   r4   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   r@   r   )rA   ZDatetimeIndexr   r   dater   	Timestampry   astyper.   rE   rz   
tz_convert)
r9   fill_valr]   r   r^   rR   tsresultrG   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   r@   )rA   TimedeltaIndexr   r   	Timedeltar   ry   r   r.   rE   rz   )r9   r^   rR   r   r   rG   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   )rA   PeriodIndexr   Period
isinstancer   strry   r   r.   rE   rz   r   )	r9   ry   r   r   r^   datarR   r   rG   r   r   r   test_insert_index_period8  s(   
z0TestInsertIndexCoercion.test_insert_index_periodrh   ri   c                 C  rk   r	   r6   r8   r   r   r   test_insert_index_complex128`  rm   z4TestInsertIndexCoercion.test_insert_index_complex128c                 C  rk   r	   r6   r8   r   r   r   test_insert_index_boold  rm   z.TestInsertIndexCoercion.test_insert_index_boolN)r(   r:   r;   r!   r   r   rY   ru   rv   r.   r   rb   r0   r   r   rA   r   r   r   r   r   rw   r   r   r   r   r   r   rx      sr    
	

	


	
*
	
rx   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)rN   r   rE   assert_equalr   )r9   r|   condrD   rG   rP   r~   r   r   r   r   _assert_where_conversionm  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)TFTTrL   periodsc                 3  s    | ]}|  V  qd S r	   r   r   r   r   r   r   |  s    z3TestWhereCoercion._construct_exp.<locals>.<genexpr>)rI            r   r@   rJ   rK   r   )r   r   rb   r3   rA   
date_range)r9   r^   r   r   r]   rD   rR   r   r   r   _construct_expv  s   $z TestWhereCoercion._construct_expc                 C  sb   || j }||d ||d |g|d}| ||||| | ||||\}}| ||||| d S )Nr   rJ   r   )_condr   r   )r9   r^   r   r   r]   r   rR   rD   r   r   r   	_run_test  s
   
zTestWhereCoercion._run_testr   r@   rT         ?      ?Tc                 C  s2   |}|t d}|jtksJ | |||| d S )NrV   )rC   r   r.   r   r9   index_or_seriesr   r]   r   r^   r   r   r   test_where_object  s   z#TestWhereCoercion.test_where_objectc                 C  4   |}|g d}|j tjksJ | |||| d S )NrU   )r   rb   r/   r   r9   r   r   r]   r   r   r^   r   r   r   test_where_int64     z"TestWhereCoercion.test_where_int64zfill_val, exp_dtypec                 C  r   )N)rT   皙@ffffff
@皙@)r   rb   r0   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   )rb   r1   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   )rA   rB   r   rb   bool_r   )r9   r   r]   r   r^   r   r   r   test_where_series_bool  s   z(TestWhereCoercion.test_where_series_boolr   r   r   r   r3   r4   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   rL   Dr   r   r   )rA   r   Z
_with_freqr   Zto_pydatetimeZto_datetime64r   )r9   r   r   r]   r   r^   fvZscalarr   r   r   test_where_datetime64  s   	z'TestWhereCoercion.test_where_datetime64rh   ri   c                 C  rk   r	   r6   r8   r   r   r   test_where_index_complex128  rm   z-TestWhereCoercion.test_where_index_complex128c                 C  rk   r	   r6   r8   r   r   r   test_where_index_bool  rm   z'TestWhereCoercion.test_where_index_boolc                 C  rk   r	   r6   r8   r   r   r   test_where_series_timedelta64  rm   z/TestWhereCoercion.test_where_series_timedelta64c                 C  rk   r	   r6   r8   r   r   r   test_where_series_period  rm   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 DayrL   r   )TFFTz4 DaysNaTnsr   rK   r   r@   )rA   Ztimedelta_rangerb   arrayr   r   rE   rz   r3   r[   r.   )r9   r}   Ztdir   rG   r   Z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-01rK   ZQSr   Q)FTF
   r@   r   r   rJ   rL   r   z
2020-04-21r   )rA   r   Z	to_periodrb   r   r   r   r   rE   rz   Zasarrayr.   r   r[   r   )
r9   Zdtipir   r}   rG   r   othertdZperr   r   r   test_where_index_period  s&   
z)TestWhereCoercion.test_where_index_periodN)$r(   r:   r;   r   rb   r   r   r   r   r   rY   ru   rv   r.   r   r/   r0   r1   r   r   r   r   r   rA   r   r   rw   r   r   r   r   r   r   r5   r   r   r   r   r   r   r   i  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fillnarh   ri   c                 C  rk   r	   r6   r8   r   r   r   test_has_comprehensive_tests!  rm   z5TestFillnaSeriesCoercion.test_has_comprehensive_testsc                 C  s0   |  }||}t|| |j|ksJ dS )z!test coercion triggered by fillnaN)rN   r   rE   r   r   r{   r   r   r   _assert_fillna_conversion%  s   
z2TestFillnaSeriesCoercion._assert_fillna_conversionzfill_val, fill_dtyper@   rT   r   Tc                 C  sH   |}|dt jddg}|jtksJ |d|ddg}| |||| d S )Nr?   r   r   )rb   nanr   r.   r   r9   r   r   
fill_dtyper   r^   rR   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 )NrT   r   r   )rb   r   r   r0   r   r   r   r   r   test_fillna_float648  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   )rb   r   r1   r   r   r   r   r   r   test_fillna_complex128D  s
   
z/TestFillnaSeriesCoercion.test_fillna_complex128r   r   r   r   r   )r3   r4   r.   r.   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   )rA   r   r   r   r   r   r   r   r   test_fillna_datetimeU  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   )rA   r   r   r   getattrr   r   )	r9   r   r   r  r   r   r^   r   rR   r   r   r   test_fillna_datetime64tzu  s*   
z1TestFillnaSeriesCoercion.test_fillna_datetime64tzr   rJ   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	  r@   r   rJ   rK   rL   r   )
rA   Zinterval_rangery   rb   r   r   r   ZIntervalDtyper.   r   )r9   r   r   iir^   rR   r  r   r   r   test_fillna_interval     &z-TestFillnaSeriesCoercion.test_fillna_intervalc                 C  rk   r	   r6   r8   r   r   r   test_fillna_series_int64  rm   z1TestFillnaSeriesCoercion.test_fillna_series_int64c                 C  rk   r	   r6   r8   r   r   r   test_fillna_index_int64  rm   z0TestFillnaSeriesCoercion.test_fillna_index_int64c                 C  rk   r	   r6   r8   r   r   r   test_fillna_series_bool  rm   z0TestFillnaSeriesCoercion.test_fillna_series_boolc                 C  rk   r	   r6   r8   r   r   r   test_fillna_index_bool  rm   z/TestFillnaSeriesCoercion.test_fillna_index_boolc                 C  rk   r	   r6   r8   r   r   r   test_fillna_series_timedelta64  rm   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   rL   r   r   r@   r   rJ   rK   r   )	rA   Zperiod_rangery   r   r   r   ZPeriodDtyper.   r   )r9   r   r   r   r^   rR   r  r   r   r   test_fillna_series_period  r  z2TestFillnaSeriesCoercion.test_fillna_series_periodc                 C  rk   r	   r6   r8   r   r   r   test_fillna_index_timedelta64  rm   z6TestFillnaSeriesCoercion.test_fillna_index_timedelta64c                 C  rk   r	   r6   r8   r   r   r   test_fillna_index_period  rm   z1TestFillnaSeriesCoercion.test_fillna_index_periodN)!r(   r:   r;   r   rY   ru   rw   r   r   rv   r.   r  rb   r0   r1   r  r  rA   r   r  r  ZIntervalr   r   r  r  r  r  r  r  r  r  r  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 )=TestReplaceSeriesCoercionr-   replacezdict[str, list]repr?   r   r.   rL   rI   r/   rT   r   r0   r   r   r1   TFr2   r   r   r   )UTCr   zdatetime64[ns, ]r   r   r   r   dict)paramsc                 C     |j S r	   paramr9   r   r   r   r   how  s   zTestReplaceSeriesCoercion.how)	r.   r/   r0   r1   r2   r   datetime64[ns, UTC]r   r   c                 C  r  r	   r   r"  r   r   r   from_key  s   z"TestReplaceSeriesCoercion.from_key)	r.   r/   r0   r1   r2   r3   r4   r4   r5   )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`
        r  r-   rM   )r  zipr  rA   rB   
ValueError)r9   r#  r%  r&  replacerr   r   r   r)  .  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 )NrK   rL   Zxxxr
   yyyr,   r   r   )r   r$  r0   r/   r1   )r/   r0   z32-bit platform buggy: z -> )r,   r   r   )rA   r[   rB   r  r   
startswithr  r   r   rY   skiprE   rF   	r9   r#  r&  r%  r)  r,   r^   r   rR   r   r   r   test_replace_series;  s&   

z-TestReplaceSeriesCoercion.test_replace_seriesr&  )r   r2   r.   r1   r0   r/   )Zindirectr%  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 NrK   rL   Zxyzr
   r*  r+  )rA   r[   rB   r  r   r  rE   rF   r.  r   r   r   test_replace_series_datetime_tzX  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 r0  )rA   r[   rB   r  r   r  r   ZDatetimeTZDtyper   rE   rF   r.  r   r   r   %test_replace_series_datetime_datetimel  s   
z?TestReplaceSeriesCoercion.test_replace_series_datetime_datetimerh   ri   c                 C  rk   r	   r6   r8   r   r   r   test_replace_series_period  rm   z4TestReplaceSeriesCoercion.test_replace_series_periodN)r(   r:   r;   r!   r   r  rt   rA   r   r   keyr   rY   fixturer#  r%  r&  r)  r/  ru   rv   r1  r2  rw   r3  r   r   r   r   r    sh   
 





r  )
__future__r   r   r   r   numpyrb   rY   Zpandas.compatr   r   ZpandasrA   Zpandas._testingZ_testingrE   r5  r*   r+   r=   rx   r   r   r  r   r   r   r   <module>   s&    
g 9 4 N