o
    ҷh&v                     @   s   d dl Zd dlZd dlm  mZ d dlmZm	Z	m
Z
mZmZmZmZmZmZmZ d dlmZ d dlmZ G dd dZdd Zdd	 Zejd
ddgdd ZdS )    N)
Categorical	DataFrameDatetimeIndexNaTPeriodIndexSeriesTimedeltaIndex	Timestamp
date_rangeto_datetime)_check_mixed_floatc                   @   s  e Zd Zdd Zejdd Zdd Zdd Zd	d
 Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd  Zejd!g d"d#d#ggd$d% Zd&d' Zejd(d)d*ejd+gd,d- Zd.d/ Zd0d1 Zd2d3 Zd4d5 Z d6d7 Z!d8d9 Z"ejd:e#e$gd;d< Z%ejd:e#e$gd=d> Z&d?d@ Z'dAdB Z(dCdD Z)dEdF Z*dGdH Z+dIdJ Z,dKdL Z-dMdN Z.dOdP Z/dQdR Z0dSdT Z1dUdV Z2ej3ejd(dWdWdWdXgdYdZ Z4d[d\ Z5d]d^ Z6d_S )`
TestFillNAc                 C   s   t tjgd ttdtgtjddgd}g d|_|d d  }|jddidd	 t d
gd dtddgg dd}g d|_t|| |s^t	|j
d d df |j
d d df s^J t	|j
d d df |j
d d df rtJ |st	|j
d d df |j
d d df sJ d S d S )N      foo   ABC)r   r   r   r   Tinplace       @)r   r   r   r   )r   npnanr   r	   columnsfillnatmassert_frame_equalshares_memoryiloc)selfusing_copy_on_writedforigexpected r&   Y/var/www/html/venv/lib/python3.10/site-packages/pandas/tests/frame/methods/test_fillna.py*test_fillna_dict_inplace_nonunique_columns   s$   $

,,,z5TestFillNA.test_fillna_dict_inplace_nonunique_columnsc                 C   s   t dt j}t|dd}|r<t  |d jddd W d    n1 s'w   Y  t |d d df  s;J n|d jddd |d d df dk sSJ t	|j
jdks]J t |j|sfJ d S )	N)(   2   F)copyr   Tr   r   )r   fullr   r   r   raises_chained_assignment_errorr   isnanalllen_mgrarraysr   values)r!   r"   arrr#   r&   r&   r'   test_fillna_on_column_view.   s   
 z%TestFillNA.test_fillna_on_column_viewc                 C   s~  |}t j|j|jd d df< t j|j|jdd  df< |d}|j|jd d df dk s3J d}tjt|d |jdd}W d    n1 sMw   Y  t 	|j|jd d df  seJ |j|jdd  df |j|jd df k sJ d	}t
jt|d |  W d    n1 sw   Y  d
}t
jt|d |jddd W d    d S 1 sw   Y  d S )N   r   r   ,DataFrame.fillna with 'method' is deprecatedmatchpadmethodz'Must specify a fill 'value' or 'method'z(Cannot specify both 'value' and 'method'ffill)r   r   locindexr   r0   r   assert_produces_warningFutureWarningr/   pytestraises
ValueError)r!   datetime_frametfzero_filledmsgpaddedr&   r&   r'   test_fillna_datetime@   s*   
$&*
"zTestFillNA.test_fillna_datetimec                 C   s   |}t j|j|jdd df< t j|j|jdd  df< |jdd d}tjt|d	 |jd
d W d    d S 1 s=w   Y  d S )Nr7      r   r   r   valuer9   r:   r<   r=   )r   r   r@   rA   r   r   rB   rC   )r!   float_string_framemfrJ   r&   r&   r'   test_fillna_mixed_typeW   s   "z!TestFillNA.test_fillna_mixed_typec                 C   s   |j g dd}tj|j|jdd  df< |jdd}t|dd id d	}tjt	|d
 |jdd}W d    n1 s=w   Y  t|dd id d S )N)r   r   Dr   rN   r   r   rO   r   dtyper9   r:   r<   r=   )
reindexr   r   r@   rA   r   r   r   rB   rC   )r!   mixed_float_framerR   resultrJ   r&   r&   r'   test_fillna_mixed_floata   s   z"TestFillNA.test_fillna_mixed_floatc              	   C   sx   |rt d tdgd}dD ]*}d}tjt|d |jj|dd |jj|d	 W d    n1 s4w   Y  qd S )
Nz9condition is unnecessary complex and is deprecated anywayxrU   )r<   backfillz)Series.fillna with 'method' is deprecatedr:   Tr>   r   r=   )rD   skipr   r   rB   rC   r\   r   )r!   r"   r#   mrJ   r&   r&   r'   test_fillna_emptym   s   
zTestFillNA.test_fillna_emptyc                 C   s   t ddtjdgddtjdgddtjdgg}|ddi}t g dg dg dg}t|| |jddid	d
}t|| |d u sEJ d S )Nabcr   r   )rb   rb   r   rb   )rc   rc   r   rc   )rd   rd   r   rd   Tr   r   r   r   r   r   r   )r!   r#   rZ   r%   return_valuer&   r&   r'   test_fillna_different_dtypex   s   &z&TestFillNA.test_fillna_different_dtypec                 C   sx   t tjdd}tj|jdddf< tj|jdddf< | }d|jd< d|jd	< |jdd
d}t	
|| d S )Nr   )
   r      r   r   r7   i  )r   r   )r   r   r   limit)r   r   randomdefault_rngstandard_normalr   r    r+   r   r   r   r!   r#   r%   rZ   r&   r&   r'   test_fillna_limit_and_value   s   

z&TestFillNA.test_fillna_limit_and_valuec                 C   sl   t ttdgtdtgd}| }|d |j|jd df |d< |jd|d id}t|| d S )Nz2014-1-1z2013-1-1)DateDate2rq   r   rr   rO   )	r   r   r	   r+   r   r@   rA   r   r   ro   r&   r&   r'   test_fillna_datelike   s   

"zTestFillNA.test_fillna_datelikec                 C   s   t dtdtgi}t dtdtdgi}d}tjt|d |jdd}W d    n1 s0w   Y  t|| t dttdgi}t dtdtdgi}d}tjt|d |jdd}W d    n1 skw   Y  t|| d S )Nr   z2012-11-11 00:00:00+01:00r9   r:   r<   r=   bfill)r   r	   r   r   rB   rC   r   r   )r!   r#   exprJ   resr&   r&   r'   test_fillna_tzaware   s4   zTestFillNA.test_fillna_tzawarec                 C   s   t tddddddtjtjgd}d}tjt|d	 |jd
d}W d    n1 s,w   Y  t tddddg dd}t|| d S )N20130101   z
US/Eastern)periodstzr   r   r   r   r9   r:   r<   r=   )      ?r   r   r   )	r   r
   r   r   r   rB   rC   r   r   r!   r#   rJ   rZ   r%   r&   r&   r'   $test_fillna_tzaware_different_column   s   z/TestFillNA.test_fillna_tzaware_different_columnc                 C   s  t dddtjgg dd}ddtjdg}t||d	}t g d
g dd}g d}t||d	}t g dg dd}ddtjg}t||d	}	t ddgg dd}
ddg}t|
|d	}|jddd	d}t|| d}tjt	|d |jddd	d W d    n1 sw   Y  d}tj
t|d |jdd}W d    n1 sw   Y  t|| |jdgd}t||	 | }t|| t tjdtjgddgd}t|g dd	}t g dddgd}t|g dd	}|d}t|| d S )Nr   r   r   r   r   r   )
categoriesrb   rc   dcatsvals)r   r   r   r   )rb   rc   rc   r   rO   z3Cannot setitem on a Categorical with a new categoryr:   ry   rd   r9   r<   r=   r   )subset)rb   rc   rb   )r   r   r   r   r   r   r   rD   rE   	TypeErrorrB   rC   dropna)r!   catr   r#   cat2vals2df_exp_fillcat3vals3df_exp_drop_catscat4vals4df_exp_drop_allrv   rJ   rd   cat_expdf_expr&   r&   r'   test_na_actions_categorical   sB   
z&TestFillNA.test_na_actions_categoricalc           
      C   sb  t tjdtjg}t tjtjtjg}t||d}tdtjd}||}tjtjtjg}tg d|ddd}t|| |jtj}t	||j |j
tj}t	||j
 tdddttg}	td	t |	i}t|jtd
| tdddttgdd}	td	t |	i}t|jtd
| tdddttg}	td	t |	i}t|jtd
| d S )Nr   r   r   )r   r   r   categoryrV   z2011-01-01 09:00z2016-01-01 23:45rb   rO   z2011-01M)freqz1 daysz2 days)r   r   r   r   r   r   r   r   r   assert_series_equalr   r   r   r   r   )
r!   r   valr#   medianrv   v_expr   rZ   idxr&   r&   r'   test_fillna_categorical_nan  s.   
z&TestFillNA.test_fillna_categorical_nanc                 C   s   t ddtjgi}d}tjt|d |jddd}W d    n1 s$w   Y  t dddgi}t|| t ddtjgi}tjt|d |jddidd}W d    n1 s[w   Y  t dddgi}t|| d S )	Nrb   r}   z.The 'downcast' keyword in fillna is deprecatedr:   r   inferdowncastr   r   r   r   r   rB   rC   r   r   r~   r&   r&   r'   test_fillna_downcast&  s   zTestFillNA.test_fillna_downcastc                 C   s`   |g ddd}d}t jt|d |jddd}W d    n1 s#w   Y  t || d S )	Nr   objectrV    The 'downcast' keyword in fillnar:    Fr   )r   rB   rC   r   assert_equal)r!   frame_or_seriesobjrJ   rZ   r&   r&   r'   test_fillna_downcast_false7  s   z%TestFillNA.test_fillna_downcast_falsec           
      C   s  |g dt jd}d}tjt|d |jdt t jd}W d    n1 s(w   Y  |t j}t	|| |t j
}tjt|d |jddd}W d    n1 sXw   Y  |}t	|| tjt|d |jdt t jd}	W d    n1 sw   Y  t	|	| d S )Nr   rV   r   r:   r   r   r   )r   int64r   rB   rC   r   rW   int32astyper   float64)
r!   r   r   rJ   rv   r%   obj2res2	expected2res3r&   r&   r'   test_fillna_downcast_noop?  s"   z$TestFillNA.test_fillna_downcast_noopr   )r   r   r   r   c                 C   sl   t tjddg|d}td  |ddi}W d    n1 s!w   Y  | }d|d< t|| d S )Nr   r   rA   r   r           )r   r   r   r   rB   r   r+   r   )r!   r   r#   rZ   r%   r&   r&   r'   -test_fillna_dictlike_value_duplicate_colnamesY  s   z8TestFillNA.test_fillna_dictlike_value_duplicate_colnamesc                 C   s   t g dg dd}|j}ttdgd g dd}t|| |d}t dg dg dd}t|| t t	dd	d
gdd}|d}t dt	dd	d
gd}t|| d S )Nr   )r   r   r   ry   r7   r   r   r7   rA   r   r   r   r   r   )rA   r   rW   r   )
r   dtypesr   r   rW   r   r   r   r   ranger!   r#   rZ   r%   r&   r&   r'   test_fillna_dtype_conversiond  s   

z'TestFillNA.test_fillna_dtype_conversionr   r   r   r}   c                 C   s@   t dtjgddgd}|tj|}||}t|| d S )Nr   r}   r   r|   )r   r   r   replacer   r   r   )r!   r   r#   r%   rZ   r&   r&   r'   *test_fillna_dtype_conversion_equiv_replaceu  s   
z5TestFillNA.test_fillna_dtype_conversion_equiv_replacec                 C   s  t ddtjgtdddg dg ddtd	ddd
}|d}t g dtdddg dg ddtd	ddd
}t|| t ddtjgtdtdtgg dg ddtd	ddd
}|d}t g dtdtddgg dg ddtd	ddd
}t|| d S )Nr,   rx   r   )rz   )r   barN)foo2bar2Nr   r   r   rT   20130110r   ?)r,   r   r   )r   r   r   )r   r   r   z
2013-01-01z
2013-01-02)	r   r   r   r
   r   r   r   r	   r   r   r&   r&   r'   test_fillna_datetime_columns|  sH   



	

	


	
	z'TestFillNA.test_fillna_datetime_columnsc                 C      t j|j|jd d df< t j|j|jdd  df< d}tjt|d |jdd}W d    n1 s4w   Y  t|	 | d S )Nr7   r   r8   r9   r:   r?   r=   )
r   r   r@   rA   r   rB   rC   r   r   r?   r!   rG   rJ   altr&   r&   r'   
test_ffill  s   zTestFillNA.test_ffillc                 C   r   )Nr7   r   r8   r9   r:   rt   r=   )
r   r   r@   rA   r   rB   rC   r   r   rt   r   r&   r&   r'   
test_bfill  s   zTestFillNA.test_bfillc                 C   s$  t d}tt jdd|d}|d d j|ddd}d}tjt	|d	 |d d |j
dd
}W d    n1 s?w   Y  t j|jdd < t|| |dd  j|ddd}tjt	|d	 |dd  |j
dd
}W d    n1 s}w   Y  t j|jd d< t|| d S )Nrh   r   rh   ry   r   r<   r7   r>   rk   r9   r:   r=   r   r]   r   r   aranger   rl   rm   rn   rX   r   rB   rC   r   r   r    r   r!   rA   r#   rZ   rJ   r%   r&   r&   r'   test_frame_pad_backfill_limit  s   
z(TestFillNA.test_frame_pad_backfill_limitc                 C   s  t d}tt jdd|d}|d d |}d}tjt	|d |j
ddd	}W d    n1 s6w   Y  tjt	|d |d d |j
dd
}W d    n1 sZw   Y  t j|jdd < t|| |dd  |}tjt	|d |j
ddd	}W d    n1 sw   Y  tjt	|d |dd  |j
dd
}W d    n1 sw   Y  t j|jd d< t|| d S )Nrh   r   r   r   r9   r:   r<   r7   r   r=   r   r   r]   r   r   r   r&   r&   r'   test_frame_fillna_limit  s*   
z"TestFillNA.test_frame_fillna_limitc                 C   s,   t tjddt}|tj d S )Nr   r   )	r   r   rl   rm   rn   r   intr   r   )r!   r#   r&   r&   r'   test_fillna_skip_certain_blocks  s   z*TestFillNA.test_fillna_skip_certain_blockstypec                 C   b   t tjdd|}d}tjt|d |j	ddd W d    d S 1 s*w   Y  d S )Nr   r   zLimit must be greater than 0r:   r   r8   rj   
r   r   rl   rm   rn   r   rD   rE   rF   r   r!   r   r#   rJ   r&   r&   r'   test_fillna_positive_limit  
   "z%TestFillNA.test_fillna_positive_limitc                 C   r   )Nr   r   zLimit must be an integerr:   r   g      ?rj   r   r   r&   r&   r'   test_fillna_integer_limit  r   z$TestFillNA.test_fillna_integer_limitc                 C   sL  t tjdd}tj|jd ddf< tj|jdd df< |jdd}||us+J |jdd	d
 t	|| |jddid	d
}|d u sGJ tj|jd ddf< tj|jdd df< d}tj
t|d |jdd}W d    n1 suw   Y  ||usJ tj
t|d |jdd	d W d    n1 sw   Y  t	|| d S )Nr   r   ry   r   r   r   rO   TrP   r   r9   r:   r?   r=   r^   )r   r   rl   rm   rn   r   r@   r   r   r   rB   rC   )r!   r#   r%   rJ   r&   r&   r'   test_fillna_inplace  s(   zTestFillNA.test_fillna_inplacec                 C   s  t tjddtjtjgdddtjtjgtjddddgd}|ddd}| }|d	 d|d	< |d
 d|d
< t|| |dddd}|| }||  }t|| t	j
tdd |j|ddd W d    d S 1 s}w   Y  d S )Nr   r   r   ry   rb   rc   rd   r   r7   )rb   rc   rb   rc   ri   )rb   rc   r   zcolumn by columnr:   )axis)r   r   r   r   r+   r   r   maxto_dictrD   rE   NotImplementedErrorr   r&   r&   r'   test_fillna_dict_series  s$   "z"TestFillNA.test_fillna_dict_seriesc                 C   s   t tjddtjtjgdddtjtjgtjddddgdtdd}t tjdd	d
dgg ddgd dtdd}||}t tjddtjdgdddtjdgtjddddgdtdd}t|| d S )Nr   r   r   ry   r   VWXYZr   rh   rM      r)   )r*   <   F   P   Z   r   r7   )rb   rc   r   VWXuZr   r   r   r   listr   r   r   )r!   r#   df2rZ   r%   r&   r&   r'   test_fillna_dataframe2  s.   

		z TestFillNA.test_fillna_dataframec                 C   sX  t jdd}t j|d d d d df< t|}d}tjt|d |j	ddd}W d    n1 s4w   Y  tjt|d |j
j	dd	j
}W d    n1 sSw   Y  t|| |d
dd tjt|d |j	ddd}W d    n1 s~w   Y  tjt|d |tj	ddd}W d    n1 sw   Y  t|| d S )Nr   )rh   rh   r9   r:   r?   r   )r>   r   r<   r=      r   r7   )r   rl   rm   rn   r   r   r   rB   rC   r   Tr   insertr   float)r!   r5   r#   rJ   rZ   r%   r&   r&   r'   test_fillna_columnsU  s&   zTestFillNA.test_fillna_columnsc                 C   s@   t jtdd |jdd W d    d S 1 sw   Y  d S )Nffilr:   r=   )rD   rE   rF   r   )r!   float_framer&   r&   r'   test_fillna_invalid_methodh  s   "z%TestFillNA.test_fillna_invalid_methodc                 C   s   d}t jt|dd |ddg W d    n1 sw   Y  t jt|dd |d W d    n1 s=w   Y  d}t jt|d |jd d d	f | W d    d S 1 scw   Y  d S )
NzA"value" parameter must be a scalar or dict, but you passed a "{}"r   r:   r   r   tuple)r   r   zP"value" parameter must be a scalar, dict or Series, but you passed a "DataFrame"r   )rD   rE   r   formatr   r    )r!   r   rJ   r&   r&   r'   test_fillna_invalid_valuel  s   "z$TestFillNA.test_fillna_invalid_valuec                 C   s   dd t dddD }tjdd}tt d||d	}d
}tjt|d |jdd}W d    n1 s7w   Y  |j	
 |j	
 ksHJ d S )Nc                 S   s   g | ]}d t | qS )zCOL.)str).0ir&   r&   r'   
<listcomp>}  s    z9TestFillNA.test_fillna_col_reordering.<locals>.<listcomp>r7   r   r,   r   )rM   r7   rM   )rA   r   datar9   r:   r?   r=   )r   r   rl   rm   r   r   rB   rC   r   r   tolist)r!   colsr   r#   rJ   filledr&   r&   r'   test_fillna_col_reordering|  s   z%TestFillNA.test_fillna_col_reorderingc                 C   s   |}t j|j|jdd df< t j|j|jdd  df< |jdd}|j|jdd df dk s4J |d= |jg djdd d S )	Nr7   rM   r   rN   r   r   rO   rU   )r   r   r@   rA   r   r0   rX   )r!   r   rQ   rR   r  r&   r&   r'   test_fill_corner  s   $zTestFillNA.test_fill_cornerc                 C   sz   t ddtjgi}d}tjt|d |jddiddid}W d    n1 s(w   Y  t dddgi}t|| d S )Ncol1r   r   r:   r   r   r   r   r~   r&   r&   r'   test_fillna_downcast_dict  s   z$TestFillNA.test_fillna_downcast_dictc                 C   s  t tjdtjdgddtjdgtjtjtjdgtjdtjdggtdd}|jdd	dd
}|jdd	dd
}t tg dddddtjdgtjd	tjtjgtg ddddg dd}t tg dddtg dddd	d	tjd	gtg ddddg dd}t|| t|| d S )Nr   r   r   ry   r   r7   ABCDrU   d   r   rP   rk   )r  r   r  r  r   rV   )r   r   r7   ry   r   )r   r   r   r   r   )r   ry   r  r   )r   r   r   r   r   r   r   r   )r!   r#   rZ   result2r%   r   r&   r&   r'   "test_fillna_with_columns_and_limit  s8   		
z-TestFillNA.test_fillna_with_columns_and_limitc                 C   sF   t tdd gtdd gd}| }|jtjdd t|| d S )Nz
2018-05-30z
2018-09-30)date1date2Tr   )r   r   r+   r   r   r   r   r   r!   r#   r%   r&   r&   r'   test_fillna_datetime_inplace  s   

z'TestFillNA.test_fillna_datetime_inplacec                 C   s   t tjdtjdgddtjdgtjtjtjdgtjdtjdggtdd}|jdd	dd
}||us2J |jdd	ddd t|| d S )Nr   r   r   ry   r   r7   r  rU   r  r	  T)r   rP   rk   r   r   r  r&   r&   r'   0test_fillna_inplace_with_columns_limit_and_value  s   
z;TestFillNA.test_fillna_inplace_with_columns_limit_and_valuer,   r\   yc                 C   s   t tjdgtjdgd}| }|d d  }|j|dd t ddgddgd}t|| |r9t|| d S t|| d S )Nr   r  Tr   r,   r         )r   r   r   r+   r   r   r   )r!   r   r"   r#   df_origresult_viewr%   r&   r&   r'   test_inplace_dict_update_view  s   z(TestFillNA.test_inplace_dict_update_viewc              	   C   s   t ddtjdtjgdtjtjddgdtjdddgtjddtjd	gd
}|jdddd}t g dddtjdgdtjddgg dg dgg d
d}t|| d S )Nr7   r   rh   ri   r      r   r      )r  col2col3col4r*   )rk   r   )      @g      @g      (@      I@r   r  r}   )g      $@r  r   r  )r        @r   g      2@rU   re   r   r&   r&   r'   )test_single_block_df_with_horizontal_axis  s$   
z4TestFillNA.test_single_block_df_with_horizontal_axisc                 C   s   t tjddgddtjgddtjgd}t g dg dddtjgd}t|ddi| t|dd	d
| t g dg dddtjgd}t|d	dd| d S )Nr   r  r}   ))r\   rb   r\   rc   )r  rd   )r  r   r  )r}   r   r  r\   r,   r   )r\   r   )r}   r   g       )r   r\   )r   r   r   r   r   r   )r!   pdfr%   r&   r&   r'   "test_fillna_with_multi_index_frame  s*   




z-TestFillNA.test_fillna_with_multi_index_frameN)7__name__
__module____qualname__r(   td&skip_array_manager_not_yet_implementedr6   rL   rS   r[   ra   rg   rp   rs   rw   r   r   r   r   r   r   rD   markparametrizer   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  r  r  r  r  r  skip_array_manager_invalid_testr  r  r"  r&   r&   r&   r'   r      sd    

,$


,	


#	
$r   c                  C   sV   t g dg dg dgg dd} | jddd}|d	}|   d	ks)J d S )
N)r   r   r   r}   )r   r   r   r   )r   r   r   r  )i1i2i3f1rU   r+  r,  r   r   )r   pivotr   isnasum)r#   df_nonconsolrZ   r&   r&   r'   !test_fillna_nonconsolidated_frame  s   
r3  c                  C   s   t d d gd d ggddgd} td | jddddd	 W d    n1 s(w   Y  t ddgddggddgd}t| | d S )
Nr   r   rU   Fr   r   r|   Tr   )r   r   rB   r   r   )r#   r%   r&   r&   r'   test_fillna_nones_inplace.  s   r4  funcr<   r]   c                 C   sL   t dg di}tt t||   W d    d S 1 sw   Y  d S )Nrb   r   )r   r   rB   rC   getattr)r5  r#   r&   r&   r'   test_pad_backfill_deprecated;  s   "r7  )numpyr   rD   pandas.util._test_decoratorsutil_test_decoratorsr&  pandasr   r   r   r   r   r   r   r	   r
   r   pandas._testing_testingr   pandas.tests.frame.commonr   r   r3  r4  r(  r)  r7  r&   r&   r&   r'   <module>   s"    0      