a
    d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autouseZscopec                 #  sv    j }t|j|j|jg} fdd} jj}|js<|j	rDd V  n.|D ]"}||sHt
d|j d| qHd 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 l/var/www/html/stable-diffusion-webui/venv/lib/python3.9/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 | d S 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)r8   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?            rH   r+   N)copyr@   rA   rD   rE   r+   r   loc)r8   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   rH   皙?c              	   C  s   t jg dtdd}|jjtks&J |tu r| }d}d}tj	||dF t
jt|d d|d< W d    n1 sz0    Y  W d    q1 s0    Y  n$t td|g }| |||| d S )Nr?   rI   rJ   rK   abcdrL   z;Series.__setitem__ treating keys as positions is deprecatedz/index 5 is out of bounds for axis 0 with size 4)matchrH   )r@   rA   rB   r+   r   r-   
IndexErrorrM   pytestZraisesrD   Zassert_produces_warningFutureWarningIndexrR   )r8   val	exp_dtypeobjrP   Zwarn_msgmsg	exp_indexr   r   r   test_setitem_index_objectn   s    F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 )NrT   r   r?   rI   rJ   )r@   rA   r+   r   npr.   rZ   rR   )r8   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 )NrT   )rS    @@ffffff@rL   rS   rc   rd   re   )r@   rA   r+   r   ra   r/   rZ   rR   )r8   r[   r\   r   r]   r_   r   r   r   test_setitem_index_float64   s    z.TestSetitemCoercion.test_setitem_index_float64Test not implementedreasonc                 C  s   t d S r   r5   r7   r   r   r   test_setitem_series_period   s    z.TestSetitemCoercion.test_setitem_series_periodc                 C  s   t d S r   r5   r7   r   r   r   test_setitem_index_complex128   s    z1TestSetitemCoercion.test_setitem_index_complex128c                 C  s   t d S r   r5   r7   r   r   r   test_setitem_index_bool   s    z+TestSetitemCoercion.test_setitem_index_boolc                 C  s   t d S r   r5   r7   r   r   r   test_setitem_index_datetime64   s    z1TestSetitemCoercion.test_setitem_index_datetime64c                 C  s   t d S r   r5   r7   r   r   r   test_setitem_index_datetime64tz   s    z3TestSetitemCoercion.test_setitem_index_datetime64tzc                 C  s   t d S r   r5   r7   r   r   r   test_setitem_index_timedelta64   s    z2TestSetitemCoercion.test_setitem_index_timedelta64c                 C  s   t d S r   r5   r7   r   r   r   test_setitem_index_period   s    z-TestSetitemCoercion.test_setitem_index_periodN)r'   r9   r:   r   r    __annotations__rG   rR   rX   markparametrizer-   rW   r`   ra   r.   r/   rb   rf   xfailrj   rk   rl   rm   rn   ro   rp   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)rM   rv   rD   assert_index_equalr   r8   originalvaluerF   rO   targetresr   r   r   _assert_insert_conversion   s    z1TestInsertIndexCoercion._assert_insert_conversionz"insert, coerced_val, coerced_dtyper?   rS   Fr   c                 C  sD   t td}|jtksJ t d|dddg}| |||| d S )NrU   r>   bcd)r@   rZ   rB   r   r-   r}   )r8   rv   coerced_valcoerced_dtyper]   rQ   r   r   r   test_insert_index_object   s    
z0TestInsertIndexCoercion.test_insert_index_object)r?   r?   Nc                 C  sR   |}t jg d|d}|d ur"|n|}t jd|dddg|d}| |||| d S )NrT   r   r?   rI   rJ   rK   r@   rZ   r}   )r8   Zany_int_numpy_dtyperv   r   r   r   r]   rQ   r   r   r   test_insert_int_index   s
    z-TestInsertIndexCoercion.test_insert_int_index)r?         ?Nc                 C  sR   |}t jg d|d}|d ur"|n|}t jd|dddg|d}| |||| d S )N)r          @      @      @r   r   r   r   r   r   )r8   Zfloat_numpy_dtyperv   r   r   r   r]   rQ   r   r   r   test_insert_float_index   s
    z/TestInsertIndexCoercion.test_insert_float_indexfill_val,exp_dtype
2012-01-01datetime64[ns]
US/Easterntzdatetime64[ns, US/Eastern]r2   r3   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rt d}|d|}|t	d|}|jt	ksJ t
|| t jdd	d}|d|}|d||jj}|j|jksJ t
|| nHt jdd	d}|d|}|t	d|}|jt	ks"J t
|| d}	|d|	}|t	d|	}|d |	ksbJ |jt	ksrJ t
|| d S )
N)
2011-01-01
2011-01-02
2011-01-03
2011-01-04r   r   r   r   r   r   r?   r   )r@   ZDatetimeIndexr   r   dater}   	Timestamprv   astyper-   rD   rw   
tz_convert)
r8   fill_valr\   r   r]   rQ   tsresultrF   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	
|| qNd S )N)1 day2 day3 day4 daytimedelta64[ns])r   10 dayr   r   r   r   r   r?   )r@   TimedeltaIndexr   r}   	Timedeltar   rv   r   r-   rD   rw   )r8   r]   rQ   r   r   rF   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rt j|dd}| |||| | |t||| nh|d	|}|t	d	|}t
|| t|t js|d	t|}|t	d	t|}t
|| d S )
N)2011-012011-022011-032011-04r   r   r   r   r   r   r   r   )r@   PeriodIndexr   Period
isinstancer}   strrv   r   r-   rD   rw   r   )	r8   rv   r   r   r]   datarQ   r   rF   r   r   r   test_insert_index_period8  s&    
z0TestInsertIndexCoercion.test_insert_index_periodrg   rh   c                 C  s   t d S r   r5   r7   r   r   r   test_insert_index_complex128`  s    z4TestInsertIndexCoercion.test_insert_index_complex128c                 C  s   t d S r   r5   r7   r   r   r   test_insert_index_boold  s    z.TestInsertIndexCoercion.test_insert_index_boolN)r'   r9   r:   r    r   r}   rX   rr   rs   r-   r   ra   r/   r   r   r@   r   r   r   r   r   rt   r   r   r   r   r   r   ru      sp   	

	


	

*	

ru   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)rM   r   rD   assert_equalr   )r8   ry   condrC   rF   rO   r{   r|   r   r   r   _assert_where_conversionm  s    z*TestWhereCoercion._assert_where_conversionc                   sx    du r|g d}n6t  ttjfr6tj dd}n| fdddD }||d |d	 |d
 |d g|d}||fS )NT)TFTTrK   periodsc                 3  s   | ]}|  V  qd S r   r   r   r   r   r   r   |      z3TestWhereCoercion._construct_exp.<locals>.<genexpr>)rH            r   r?   rI   rJ   r   )r   r   ra   r2   r@   
date_range)r8   r]   r   r   r\   rC   rQ   r   r   r   _construct_expv  s    $z TestWhereCoercion._construct_expc                 C  sb   || j }||d ||d |g|d}| ||||| | ||||\}}| ||||| d S )Nr   rI   r   )_condr   r   )r8   r]   r   r   r\   r   rQ   rC   r   r   r   	_run_test  s
    
zTestWhereCoercion._run_testr   r?   rS         ?      ?Tc                 C  s2   |}|t d}|jtksJ | |||| d S )NrU   )rB   r   r-   r   r8   index_or_seriesr   r\   r   r]   r   r   r   test_where_object  s    z#TestWhereCoercion.test_where_objectc                 C  s4   |}|g d}|j tjks J | |||| d S )NrT   )r   ra   r.   r   r8   r   r   r\   r   r   r]   r   r   r   test_where_int64  s    z"TestWhereCoercion.test_where_int64zfill_val, exp_dtypec                 C  s4   |}|g d}|j tjks J | |||| d S )N)rS   皙@ffffff
@皙@)r   ra   r/   r   r   r   r   r   test_where_float64  s    z$TestWhereCoercion.test_where_float64c                 C  s:   |}|g dt jd}|jt jks&J | |||| d S )N)r          @       @      @      @      @      @r   )ra   r0   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   )r@   rA   r   ra   bool_r   )r8   r   r\   r   r]   r   r   r   test_where_series_bool  s    z(TestWhereCoercion.test_where_series_boolr   r   r   r   r2   r3   r   c                 C  s   |}|t jddddd }|jdks,J |}|dkrb|| | fD ]}| |||| qJn"|| fD ]}| |||| qnd S )Nr   rK   Dr   r   r   )r@   r   Z
_with_freqr   Zto_pydatetimeZto_datetime64r   )r8   r   r   r\   r   r]   fvZscalarr   r   r   test_where_datetime64  s    	z'TestWhereCoercion.test_where_datetime64rg   rh   c                 C  s   t d S r   r5   r7   r   r   r   test_where_index_complex128  s    z-TestWhereCoercion.test_where_index_complex128c                 C  s   t d S r   r5   r7   r   r   r   test_where_index_bool  s    z'TestWhereCoercion.test_where_index_boolc                 C  s   t d S r   r5   r7   r   r   r   test_where_series_timedelta64  s    z/TestWhereCoercion.test_where_series_timedelta64c                 C  s   t d S r   r5   r7   r   r   r   test_where_series_period  s    z*TestWhereCoercion.test_where_series_periodrz   	   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 sJ |||}t|| d S )Nz1 DayrK   r   )TFFTz4 DaysNaTnsr   rJ   r   r?   )r@   Ztimedelta_rangera   arrayr   r   rD   rw   r2   rZ   r-   )r8   rz   Ztdir   rF   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-01rJ   ZQSr   Q)FTF
   r?   r   r   rI   rK   r   z
2020-04-21r   )r@   r   Z	to_periodra   r   r   r   r   rD   rw   Zasarrayr-   r   rZ   r   )
r8   Zdtipir   rz   rF   r   othertdZperr   r   r   test_where_index_period  s&    
z)TestWhereCoercion.test_where_index_periodN)$r'   r9   r:   r   ra   r   r   r   r   r   rX   rr   rs   r-   r   r.   r/   r0   r   r   r   r   r   r@   r   r   rt   r   r   r   r   r   r   r4   r   r   r   r   r   r   r   i  sn   		
 
 
	






 
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fillnarg   rh   c                 C  s   t d S r   r5   r7   r   r   r   test_has_comprehensive_tests!  s    z5TestFillnaSeriesCoercion.test_has_comprehensive_testsc                 C  s0   |  }||}t|| |j|ks,J dS )z!test coercion triggered by fillnaN)rM   r   rD   r   r   rx   r   r   r   _assert_fillna_conversion%  s    
z2TestFillnaSeriesCoercion._assert_fillna_conversionzfill_val, fill_dtyper?   rS   r   Tc                 C  sH   |}|dt jddg}|jtks$J |d|ddg}| |||| d S )Nr>   r   r   )ra   nanr   r-   r   r8   r   r   
fill_dtyper   r]   rQ   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 )NrS   r   r   )ra   r   r   r/   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   )ra   r   r0   r   r   r   r   r   r   test_fillna_complex128D  s
    
z/TestFillnaSeriesCoercion.test_fillna_complex128r   r   r   r   r   )r2   r3   r-   r-   r   c                 C  sl   |}|t dt jt dt dg}|jdks6J |t d|t dt dg}| |||| d S )Nr   r   r   r   )r@   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sFJ 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   )r@   r   r   r   getattrr   r   )	r8   r   r   r   r   r   r]   r   rQ   r   r   r   test_fillna_datetime64tzu  s*    
z1TestFillnaSeriesCoercion.test_fillna_datetime64tzr   rI   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   rI   rJ   rK   r   )
r@   Zinterval_rangerv   ra   r   r   r   ZIntervalDtyper-   r   )r8   r   r   iir]   rQ   r   r   r   r   test_fillna_interval  s    &z-TestFillnaSeriesCoercion.test_fillna_intervalc                 C  s   t d S r   r5   r7   r   r   r   test_fillna_series_int64  s    z1TestFillnaSeriesCoercion.test_fillna_series_int64c                 C  s   t d S r   r5   r7   r   r   r   test_fillna_index_int64  s    z0TestFillnaSeriesCoercion.test_fillna_index_int64c                 C  s   t d S r   r5   r7   r   r   r   test_fillna_series_bool  s    z0TestFillnaSeriesCoercion.test_fillna_series_boolc                 C  s   t d S r   r5   r7   r   r   r   test_fillna_index_bool  s    z/TestFillnaSeriesCoercion.test_fillna_index_boolc                 C  s   t d S r   r5   r7   r   r   r   test_fillna_series_timedelta64  s    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   rK   r   r   r?   r   rI   rJ   r   )	r@   Zperiod_rangerv   r   r   r   ZPeriodDtyper-   r   )r8   r   r   r   r]   rQ   r   r   r   r   test_fillna_series_period  s    &z2TestFillnaSeriesCoercion.test_fillna_series_periodc                 C  s   t d S r   r5   r7   r   r   r   test_fillna_index_timedelta64  s    z6TestFillnaSeriesCoercion.test_fillna_index_timedelta64c                 C  s   t d S r   r5   r7   r   r   r   test_fillna_index_period  s    z1TestFillnaSeriesCoercion.test_fillna_index_periodN)!r'   r9   r:   r   rX   rr   rt   r   r   rs   r-   r   ra   r/   r0   r   r   r@   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 ]0Z	de	 dZ
ejde	dejde	dgee
< qxe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-   rK   rH   r.   rS   r   r/   r   r   r0   TFr1   r   r   r   )UTCr   zdatetime64[ns, ]r   r   r   r   dict)paramsc                 C  s   |j S r   paramr8   r   r   r   r   how  s    zTestReplaceSeriesCoercion.how)	r-   r.   r/   r0   r1   r   datetime64[ns, UTC]r   r   c                 C  s   |j S r   r  r  r   r   r   from_key  s    z"TestReplaceSeriesCoercion.from_key)	r-   r.   r/   r0   r1   r2   r3   r3   r4   )r  r   c                 C  s   |j S r   r  r  r   r   r   to_key  s    z TestReplaceSeriesCoercion.to_keyc                 C  sP   |dkr$t t| j| | j| }n(|dkrHtj| j| | j| d}nt|S )z9
        Object we will pass to `Series.replace`
        r  r,   rL   )r  zipr  r@   rA   
ValueError)r8   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s6J |drN|drNd S |dv rZd S ||}|d	krt|d
v s|dkr|dv rtrt rt	
d| d|  t j| j| |d|d}n$t j| j| |dd}|j|ksJ t|| d S )NrJ   rK   Zxxxr	   yyyr+   r
   r   )r   r  r/   r.   r0   )r.   r/   z32-bit platform buggy: z -> )r+   r
   r   )r@   rZ   rA   r  r   
startswithr  r   r   rX   skiprD   rE   	r8   r  r  r  r!  r+   r]   r   rQ   r   r   r   test_replace_series;  s&    

z-TestReplaceSeriesCoercion.test_replace_seriesr  )r   r1   r-   r0   r/   r.   )Zindirectr  r  r   c           	      C  st   t jddgdd}t j| j| |dd}|j|ks6J ||}t j| j| |dd}|j|ksdJ t|| d S NrJ   rK   Zxyzr	   r"  r#  )r@   rZ   rA   r  r   r  rD   rE   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s6J ||}t j| j| |dd}t|jt jrt|jt jr||j}n|j|ksJ t	
|| d S r(  )r@   rZ   rA   r  r   r  r   ZDatetimeTZDtyper   rD   rE   r&  r   r   r   %test_replace_series_datetime_datetimel  s    
z?TestReplaceSeriesCoercion.test_replace_series_datetime_datetimerg   rh   c                 C  s   t d S r   r5   r7   r   r   r   test_replace_series_period  s    z4TestReplaceSeriesCoercion.test_replace_series_periodN)r'   r9   r:   r    r   r  rq   r@   r   r   keyr   rX   fixturer  r  r  r!  r'  rr   rs   r)  r*  rt   r+  r   r   r   r   r    sf   






r  )
__future__r   r   r   r   numpyra   rX   Zpandas.compatr   r   Zpandasr@   Zpandas._testingZ_testingrD   r-  r)   r*   r<   ru   r   r   r  r   r   r   r   <module>   s$   
g 9 4 N